dsh-plugin-image-workbench 1.1.5 → 1.1.7

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (173) hide show
  1. package/LICENSE +21 -21
  2. package/README.en.md +5 -3
  3. package/README.md +7 -3
  4. package/client.js +241 -241
  5. package/cordis.patch.yml +14 -14
  6. package/host.js +51 -0
  7. package/install.mjs +333 -387
  8. package/package.json +3 -2
  9. package/scripts/fetch-style-images.mjs +104 -0
  10. package/scripts/vendor-snapshot.py +154 -154
  11. package/server/access.py +26 -26
  12. package/server/app.py +1371 -1329
  13. package/server/catalog.py +147 -147
  14. package/server/composer.py +391 -391
  15. package/server/drafts.py +158 -158
  16. package/server/history.py +575 -575
  17. package/server/imaging.py +172 -172
  18. package/server/pool.py +1335 -1335
  19. package/server/pool_admin.py +921 -921
  20. package/server/pool_service.py +1073 -1073
  21. package/server/requirements.txt +6 -6
  22. package/server/smoke_check.py +402 -402
  23. package/server/static/access.html +202 -202
  24. package/server/static/index.html +15 -0
  25. package/vendor/chatgpt2api/source/.env.example +82 -82
  26. package/vendor/chatgpt2api/source/.python-version +1 -1
  27. package/vendor/chatgpt2api/source/LICENSE +21 -21
  28. package/vendor/chatgpt2api/source/README.md +391 -391
  29. package/vendor/chatgpt2api/source/VERSION +1 -1
  30. package/vendor/chatgpt2api/source/api/__init__.py +2 -2
  31. package/vendor/chatgpt2api/source/api/accounts.py +537 -537
  32. package/vendor/chatgpt2api/source/api/ai.py +217 -217
  33. package/vendor/chatgpt2api/source/api/app.py +62 -62
  34. package/vendor/chatgpt2api/source/api/errors.py +46 -46
  35. package/vendor/chatgpt2api/source/api/image_inputs.py +309 -309
  36. package/vendor/chatgpt2api/source/api/image_tasks.py +120 -120
  37. package/vendor/chatgpt2api/source/api/support.py +134 -134
  38. package/vendor/chatgpt2api/source/api/system.py +365 -365
  39. package/vendor/chatgpt2api/source/assets/kunrouai.svg +50 -50
  40. package/vendor/chatgpt2api/source/config.default.json +108 -108
  41. package/vendor/chatgpt2api/source/main.py +9 -9
  42. package/vendor/chatgpt2api/source/pyproject.toml +27 -27
  43. package/vendor/chatgpt2api/source/scripts/init_proxy_config.py +186 -186
  44. package/vendor/chatgpt2api/source/scripts/migrate_storage.py +163 -163
  45. package/vendor/chatgpt2api/source/scripts/privoxy-warp.conf +13 -13
  46. package/vendor/chatgpt2api/source/scripts/test_outlook_token_mailbox.py +385 -385
  47. package/vendor/chatgpt2api/source/scripts/test_storage.py +129 -129
  48. package/vendor/chatgpt2api/source/scripts/verify_oauth_refresh.py +92 -92
  49. package/vendor/chatgpt2api/source/services/account_service.py +1757 -1757
  50. package/vendor/chatgpt2api/source/services/auth_service.py +253 -253
  51. package/vendor/chatgpt2api/source/services/backup_service.py +671 -671
  52. package/vendor/chatgpt2api/source/services/config.py +667 -667
  53. package/vendor/chatgpt2api/source/services/content_filter.py +242 -242
  54. package/vendor/chatgpt2api/source/services/cpa_service.py +315 -315
  55. package/vendor/chatgpt2api/source/services/editable_file_task_service.py +260 -260
  56. package/vendor/chatgpt2api/source/services/image_service.py +376 -376
  57. package/vendor/chatgpt2api/source/services/image_storage_service.py +405 -405
  58. package/vendor/chatgpt2api/source/services/image_tags_service.py +76 -76
  59. package/vendor/chatgpt2api/source/services/image_task_service.py +550 -550
  60. package/vendor/chatgpt2api/source/services/log_service.py +337 -337
  61. package/vendor/chatgpt2api/source/services/model_service.py +182 -182
  62. package/vendor/chatgpt2api/source/services/oauth_login_service.py +267 -267
  63. package/vendor/chatgpt2api/source/services/openai_backend_api.py +2763 -2763
  64. package/vendor/chatgpt2api/source/services/openai_oauth.py +32 -32
  65. package/vendor/chatgpt2api/source/services/protocol/__init__.py +2 -2
  66. package/vendor/chatgpt2api/source/services/protocol/anthropic_v1_messages.py +307 -307
  67. package/vendor/chatgpt2api/source/services/protocol/chat_completion_cache.py +238 -238
  68. package/vendor/chatgpt2api/source/services/protocol/conversation.py +1644 -1644
  69. package/vendor/chatgpt2api/source/services/protocol/error_response.py +124 -124
  70. package/vendor/chatgpt2api/source/services/protocol/openai_search.py +19 -19
  71. package/vendor/chatgpt2api/source/services/protocol/openai_v1_chat_complete.py +316 -316
  72. package/vendor/chatgpt2api/source/services/protocol/openai_v1_image_edit.py +86 -86
  73. package/vendor/chatgpt2api/source/services/protocol/openai_v1_image_generations.py +42 -42
  74. package/vendor/chatgpt2api/source/services/protocol/openai_v1_models.py +53 -53
  75. package/vendor/chatgpt2api/source/services/protocol/openai_v1_response.py +453 -453
  76. package/vendor/chatgpt2api/source/services/protocol/web_search_tool.py +160 -160
  77. package/vendor/chatgpt2api/source/services/proxy_service.py +648 -648
  78. package/vendor/chatgpt2api/source/services/storage/__init__.py +5 -5
  79. package/vendor/chatgpt2api/source/services/storage/base.py +38 -38
  80. package/vendor/chatgpt2api/source/services/storage/database_storage.py +194 -194
  81. package/vendor/chatgpt2api/source/services/storage/factory.py +104 -104
  82. package/vendor/chatgpt2api/source/services/storage/git_storage.py +183 -183
  83. package/vendor/chatgpt2api/source/services/storage/json_storage.py +95 -95
  84. package/vendor/chatgpt2api/source/services/sub2api_service.py +531 -531
  85. package/vendor/chatgpt2api/source/utils/helper.py +469 -469
  86. package/vendor/chatgpt2api/source/utils/image_tokens.py +346 -346
  87. package/vendor/chatgpt2api/source/utils/log.py +110 -110
  88. package/vendor/chatgpt2api/source/utils/pkce.py +19 -19
  89. package/vendor/chatgpt2api/source/utils/pow.py +207 -207
  90. package/vendor/chatgpt2api/source/utils/sentinel.py +159 -159
  91. package/vendor/chatgpt2api/source/utils/turnstile.py +184 -184
  92. package/vendor/chatgpt2api/source/uv.lock +904 -904
  93. package/vendor/chatgpt2api/source/web/bun.lock +1501 -1501
  94. package/vendor/chatgpt2api/source/web/components.json +24 -24
  95. package/vendor/chatgpt2api/source/web/eslint.config.mjs +17 -17
  96. package/vendor/chatgpt2api/source/web/next.config.ts +40 -40
  97. package/vendor/chatgpt2api/source/web/package.json +58 -58
  98. package/vendor/chatgpt2api/source/web/postcss.config.mjs +5 -5
  99. package/vendor/chatgpt2api/source/web/src/app/accounts/components/account-import-dialog.tsx +925 -925
  100. package/vendor/chatgpt2api/source/web/src/app/accounts/page.tsx +1307 -1307
  101. package/vendor/chatgpt2api/source/web/src/app/debug/components/chat-panel.tsx +224 -224
  102. package/vendor/chatgpt2api/source/web/src/app/debug/components/editable-file-panel.tsx +444 -444
  103. package/vendor/chatgpt2api/source/web/src/app/debug/components/ppt-panel.tsx +9 -9
  104. package/vendor/chatgpt2api/source/web/src/app/debug/components/psd-panel.tsx +9 -9
  105. package/vendor/chatgpt2api/source/web/src/app/debug/components/search-panel.tsx +172 -172
  106. package/vendor/chatgpt2api/source/web/src/app/debug/components/skill-panel.tsx +169 -169
  107. package/vendor/chatgpt2api/source/web/src/app/debug/components/types.ts +39 -39
  108. package/vendor/chatgpt2api/source/web/src/app/debug/page.tsx +59 -59
  109. package/vendor/chatgpt2api/source/web/src/app/globals.css +230 -230
  110. package/vendor/chatgpt2api/source/web/src/app/image/components/image-composer.tsx +542 -542
  111. package/vendor/chatgpt2api/source/web/src/app/image/components/image-results.tsx +554 -554
  112. package/vendor/chatgpt2api/source/web/src/app/image/components/image-sidebar.tsx +176 -176
  113. package/vendor/chatgpt2api/source/web/src/app/image/page.tsx +1799 -1799
  114. package/vendor/chatgpt2api/source/web/src/app/image-manager/page.tsx +732 -732
  115. package/vendor/chatgpt2api/source/web/src/app/layout.tsx +50 -50
  116. package/vendor/chatgpt2api/source/web/src/app/login/page.tsx +101 -101
  117. package/vendor/chatgpt2api/source/web/src/app/logs/page.tsx +356 -356
  118. package/vendor/chatgpt2api/source/web/src/app/page.tsx +30 -30
  119. package/vendor/chatgpt2api/source/web/src/app/settings/components/api-docs-card.tsx +341 -341
  120. package/vendor/chatgpt2api/source/web/src/app/settings/components/backup-settings-card.tsx +447 -447
  121. package/vendor/chatgpt2api/source/web/src/app/settings/components/base-url-card.tsx +71 -71
  122. package/vendor/chatgpt2api/source/web/src/app/settings/components/config-card.tsx +491 -491
  123. package/vendor/chatgpt2api/source/web/src/app/settings/components/cpa-pool-dialog.tsx +108 -108
  124. package/vendor/chatgpt2api/source/web/src/app/settings/components/cpa-pools-card.tsx +169 -169
  125. package/vendor/chatgpt2api/source/web/src/app/settings/components/import-browser-dialog.tsx +185 -185
  126. package/vendor/chatgpt2api/source/web/src/app/settings/components/proxy-runtime-card.tsx +379 -379
  127. package/vendor/chatgpt2api/source/web/src/app/settings/components/proxy-settings-card.tsx +127 -127
  128. package/vendor/chatgpt2api/source/web/src/app/settings/components/proxy-settings.tsx +200 -200
  129. package/vendor/chatgpt2api/source/web/src/app/settings/components/settings-header.tsx +12 -12
  130. package/vendor/chatgpt2api/source/web/src/app/settings/components/sub2api-connections.tsx +886 -886
  131. package/vendor/chatgpt2api/source/web/src/app/settings/components/third-party-apps-card.tsx +81 -81
  132. package/vendor/chatgpt2api/source/web/src/app/settings/components/user-keys-card.tsx +422 -422
  133. package/vendor/chatgpt2api/source/web/src/app/settings/page.tsx +135 -135
  134. package/vendor/chatgpt2api/source/web/src/app/settings/store.ts +1069 -1069
  135. package/vendor/chatgpt2api/source/web/src/components/date-range-filter.tsx +49 -49
  136. package/vendor/chatgpt2api/source/web/src/components/header-actions.tsx +24 -24
  137. package/vendor/chatgpt2api/source/web/src/components/image-lightbox.tsx +434 -434
  138. package/vendor/chatgpt2api/source/web/src/components/image-thumbnail.tsx +46 -46
  139. package/vendor/chatgpt2api/source/web/src/components/theme-script.tsx +18 -18
  140. package/vendor/chatgpt2api/source/web/src/components/theme-toggle.tsx +14 -14
  141. package/vendor/chatgpt2api/source/web/src/components/top-nav.tsx +263 -263
  142. package/vendor/chatgpt2api/source/web/src/components/ui/animated-theme-toggler.tsx +256 -256
  143. package/vendor/chatgpt2api/source/web/src/components/ui/badge.tsx +41 -41
  144. package/vendor/chatgpt2api/source/web/src/components/ui/button.tsx +59 -59
  145. package/vendor/chatgpt2api/source/web/src/components/ui/calendar.tsx +46 -46
  146. package/vendor/chatgpt2api/source/web/src/components/ui/card.tsx +61 -61
  147. package/vendor/chatgpt2api/source/web/src/components/ui/checkbox.tsx +27 -27
  148. package/vendor/chatgpt2api/source/web/src/components/ui/dia-text-reveal.tsx +267 -267
  149. package/vendor/chatgpt2api/source/web/src/components/ui/dialog.tsx +132 -132
  150. package/vendor/chatgpt2api/source/web/src/components/ui/field.tsx +13 -13
  151. package/vendor/chatgpt2api/source/web/src/components/ui/input.tsx +19 -19
  152. package/vendor/chatgpt2api/source/web/src/components/ui/label.tsx +24 -24
  153. package/vendor/chatgpt2api/source/web/src/components/ui/morphing-text.tsx +149 -149
  154. package/vendor/chatgpt2api/source/web/src/components/ui/popover.tsx +38 -38
  155. package/vendor/chatgpt2api/source/web/src/components/ui/select.tsx +175 -175
  156. package/vendor/chatgpt2api/source/web/src/components/ui/sheet.tsx +143 -143
  157. package/vendor/chatgpt2api/source/web/src/components/ui/table.tsx +29 -29
  158. package/vendor/chatgpt2api/source/web/src/components/ui/tabs.tsx +91 -91
  159. package/vendor/chatgpt2api/source/web/src/components/ui/text-3d-flip.tsx +269 -269
  160. package/vendor/chatgpt2api/source/web/src/components/ui/textarea.tsx +18 -18
  161. package/vendor/chatgpt2api/source/web/src/components/version-release-dialog.tsx +127 -127
  162. package/vendor/chatgpt2api/source/web/src/constants/common-env.ts +6 -6
  163. package/vendor/chatgpt2api/source/web/src/hooks/use-version-check.ts +85 -85
  164. package/vendor/chatgpt2api/source/web/src/lib/api.ts +926 -926
  165. package/vendor/chatgpt2api/source/web/src/lib/auth-session.ts +26 -26
  166. package/vendor/chatgpt2api/source/web/src/lib/release.ts +25 -25
  167. package/vendor/chatgpt2api/source/web/src/lib/request.ts +93 -93
  168. package/vendor/chatgpt2api/source/web/src/lib/use-auth-guard.ts +89 -89
  169. package/vendor/chatgpt2api/source/web/src/lib/utils.ts +6 -6
  170. package/vendor/chatgpt2api/source/web/src/store/auth.ts +112 -112
  171. package/vendor/chatgpt2api/source/web/src/store/editable-file-history.ts +33 -33
  172. package/vendor/chatgpt2api/source/web/src/store/image-conversations.ts +318 -318
  173. package/vendor/chatgpt2api/source/web/tsconfig.json +42 -42
@@ -1,163 +1,163 @@
1
- #!/usr/bin/env python3
2
- """
3
- 存储后端数据迁移脚本
4
-
5
- 用法:
6
- python scripts/migrate_storage.py --from json --to postgres
7
- python scripts/migrate_storage.py --from postgres --to git
8
- python scripts/migrate_storage.py --export accounts.json
9
- python scripts/migrate_storage.py --import accounts.json
10
- """
11
-
12
- import argparse
13
- import json
14
- import os
15
- import sys
16
- from pathlib import Path
17
-
18
- # 添加项目根目录到 Python 路径
19
- sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
20
-
21
- DATA_DIR = Path(__file__).resolve().parents[1] / "data"
22
-
23
- from services.storage.factory import create_storage_backend
24
-
25
-
26
- def export_to_json(output_file: str):
27
- """导出当前存储后端的数据到 JSON 文件"""
28
- print(f"[migrate] Exporting data to {output_file}")
29
- DATA_DIR.mkdir(parents=True, exist_ok=True)
30
- storage = create_storage_backend(DATA_DIR)
31
- accounts = storage.load_accounts()
32
-
33
- output_path = Path(output_file)
34
- output_path.parent.mkdir(parents=True, exist_ok=True)
35
- output_path.write_text(
36
- json.dumps(accounts, ensure_ascii=False, indent=2) + "\n",
37
- encoding="utf-8",
38
- )
39
-
40
- print(f"[migrate] Exported {len(accounts)} accounts to {output_file}")
41
-
42
-
43
- def import_from_json(input_file: str):
44
- """从 JSON 文件导入数据到当前存储后端"""
45
- print(f"[migrate] Importing data from {input_file}")
46
- DATA_DIR.mkdir(parents=True, exist_ok=True)
47
- input_path = Path(input_file)
48
- if not input_path.exists():
49
- print(f"[migrate] Error: File not found: {input_file}")
50
- sys.exit(1)
51
-
52
- try:
53
- accounts = json.loads(input_path.read_text(encoding="utf-8"))
54
- if not isinstance(accounts, list):
55
- print(f"[migrate] Error: Invalid JSON format, expected array")
56
- sys.exit(1)
57
- except json.JSONDecodeError as e:
58
- print(f"[migrate] Error: Invalid JSON: {e}")
59
- sys.exit(1)
60
-
61
- storage = create_storage_backend(DATA_DIR)
62
- storage.save_accounts(accounts)
63
-
64
- print(f"[migrate] Imported {len(accounts)} accounts")
65
-
66
-
67
- def migrate_data(from_backend: str, to_backend: str):
68
- """从一个存储后端迁移到另一个"""
69
- print(f"[migrate] Migrating from {from_backend} to {to_backend}")
70
- DATA_DIR.mkdir(parents=True, exist_ok=True)
71
- # 保存原始环境变量
72
- original_backend = os.environ.get("STORAGE_BACKEND")
73
-
74
- try:
75
- # 从源后端读取数据
76
- os.environ["STORAGE_BACKEND"] = from_backend
77
- from_storage = create_storage_backend(DATA_DIR)
78
- accounts = from_storage.load_accounts()
79
- print(f"[migrate] Loaded {len(accounts)} accounts from {from_backend}")
80
-
81
- # 写入目标后端
82
- os.environ["STORAGE_BACKEND"] = to_backend
83
- to_storage = create_storage_backend(DATA_DIR)
84
- to_storage.save_accounts(accounts)
85
- print(f"[migrate] Saved {len(accounts)} accounts to {to_backend}")
86
-
87
- print(f"[migrate] Migration completed successfully!")
88
-
89
- finally:
90
- # 恢复原始环境变量
91
- if original_backend:
92
- os.environ["STORAGE_BACKEND"] = original_backend
93
- elif "STORAGE_BACKEND" in os.environ:
94
- del os.environ["STORAGE_BACKEND"]
95
-
96
-
97
- def main():
98
- parser = argparse.ArgumentParser(
99
- description="ChatGPT2API 存储后端数据迁移工具",
100
- formatter_class=argparse.RawDescriptionHelpFormatter,
101
- epilog="""
102
- 示例:
103
- # 从 JSON 迁移到 PostgreSQL
104
- python scripts/migrate_storage.py --from json --to postgres
105
-
106
- # 从 PostgreSQL 迁移到 Git
107
- python scripts/migrate_storage.py --from postgres --to git
108
-
109
- # 导出当前数据到 JSON 文件
110
- python scripts/migrate_storage.py --export backup.json
111
-
112
- # 从 JSON 文件导入数据
113
- python scripts/migrate_storage.py --import backup.json
114
-
115
- 环境变量:
116
- STORAGE_BACKEND - 存储后端类型 (json, sqlite, postgres, git)
117
- DATABASE_URL - 数据库连接字符串
118
- GIT_REPO_URL - Git 仓库地址
119
- GIT_TOKEN - Git 访问令牌
120
- """
121
- )
122
-
123
- parser.add_argument(
124
- "--from",
125
- dest="from_backend",
126
- choices=["json", "sqlite", "postgres", "git"],
127
- help="源存储后端",
128
- )
129
- parser.add_argument(
130
- "--to",
131
- dest="to_backend",
132
- choices=["json", "sqlite", "postgres", "git"],
133
- help="目标存储后端",
134
- )
135
- parser.add_argument(
136
- "--export",
137
- dest="export_file",
138
- metavar="FILE",
139
- help="导出数据到 JSON 文件",
140
- )
141
- parser.add_argument(
142
- "--import",
143
- dest="import_file",
144
- metavar="FILE",
145
- help="从 JSON 文件导入数据",
146
- )
147
-
148
- args = parser.parse_args()
149
-
150
- # 检查参数
151
- if args.from_backend and args.to_backend:
152
- migrate_data(args.from_backend, args.to_backend)
153
- elif args.export_file:
154
- export_to_json(args.export_file)
155
- elif args.import_file:
156
- import_from_json(args.import_file)
157
- else:
158
- parser.print_help()
159
- sys.exit(1)
160
-
161
-
162
- if __name__ == "__main__":
163
- main()
1
+ #!/usr/bin/env python3
2
+ """
3
+ 存储后端数据迁移脚本
4
+
5
+ 用法:
6
+ python scripts/migrate_storage.py --from json --to postgres
7
+ python scripts/migrate_storage.py --from postgres --to git
8
+ python scripts/migrate_storage.py --export accounts.json
9
+ python scripts/migrate_storage.py --import accounts.json
10
+ """
11
+
12
+ import argparse
13
+ import json
14
+ import os
15
+ import sys
16
+ from pathlib import Path
17
+
18
+ # 添加项目根目录到 Python 路径
19
+ sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
20
+
21
+ DATA_DIR = Path(__file__).resolve().parents[1] / "data"
22
+
23
+ from services.storage.factory import create_storage_backend
24
+
25
+
26
+ def export_to_json(output_file: str):
27
+ """导出当前存储后端的数据到 JSON 文件"""
28
+ print(f"[migrate] Exporting data to {output_file}")
29
+ DATA_DIR.mkdir(parents=True, exist_ok=True)
30
+ storage = create_storage_backend(DATA_DIR)
31
+ accounts = storage.load_accounts()
32
+
33
+ output_path = Path(output_file)
34
+ output_path.parent.mkdir(parents=True, exist_ok=True)
35
+ output_path.write_text(
36
+ json.dumps(accounts, ensure_ascii=False, indent=2) + "\n",
37
+ encoding="utf-8",
38
+ )
39
+
40
+ print(f"[migrate] Exported {len(accounts)} accounts to {output_file}")
41
+
42
+
43
+ def import_from_json(input_file: str):
44
+ """从 JSON 文件导入数据到当前存储后端"""
45
+ print(f"[migrate] Importing data from {input_file}")
46
+ DATA_DIR.mkdir(parents=True, exist_ok=True)
47
+ input_path = Path(input_file)
48
+ if not input_path.exists():
49
+ print(f"[migrate] Error: File not found: {input_file}")
50
+ sys.exit(1)
51
+
52
+ try:
53
+ accounts = json.loads(input_path.read_text(encoding="utf-8"))
54
+ if not isinstance(accounts, list):
55
+ print(f"[migrate] Error: Invalid JSON format, expected array")
56
+ sys.exit(1)
57
+ except json.JSONDecodeError as e:
58
+ print(f"[migrate] Error: Invalid JSON: {e}")
59
+ sys.exit(1)
60
+
61
+ storage = create_storage_backend(DATA_DIR)
62
+ storage.save_accounts(accounts)
63
+
64
+ print(f"[migrate] Imported {len(accounts)} accounts")
65
+
66
+
67
+ def migrate_data(from_backend: str, to_backend: str):
68
+ """从一个存储后端迁移到另一个"""
69
+ print(f"[migrate] Migrating from {from_backend} to {to_backend}")
70
+ DATA_DIR.mkdir(parents=True, exist_ok=True)
71
+ # 保存原始环境变量
72
+ original_backend = os.environ.get("STORAGE_BACKEND")
73
+
74
+ try:
75
+ # 从源后端读取数据
76
+ os.environ["STORAGE_BACKEND"] = from_backend
77
+ from_storage = create_storage_backend(DATA_DIR)
78
+ accounts = from_storage.load_accounts()
79
+ print(f"[migrate] Loaded {len(accounts)} accounts from {from_backend}")
80
+
81
+ # 写入目标后端
82
+ os.environ["STORAGE_BACKEND"] = to_backend
83
+ to_storage = create_storage_backend(DATA_DIR)
84
+ to_storage.save_accounts(accounts)
85
+ print(f"[migrate] Saved {len(accounts)} accounts to {to_backend}")
86
+
87
+ print(f"[migrate] Migration completed successfully!")
88
+
89
+ finally:
90
+ # 恢复原始环境变量
91
+ if original_backend:
92
+ os.environ["STORAGE_BACKEND"] = original_backend
93
+ elif "STORAGE_BACKEND" in os.environ:
94
+ del os.environ["STORAGE_BACKEND"]
95
+
96
+
97
+ def main():
98
+ parser = argparse.ArgumentParser(
99
+ description="ChatGPT2API 存储后端数据迁移工具",
100
+ formatter_class=argparse.RawDescriptionHelpFormatter,
101
+ epilog="""
102
+ 示例:
103
+ # 从 JSON 迁移到 PostgreSQL
104
+ python scripts/migrate_storage.py --from json --to postgres
105
+
106
+ # 从 PostgreSQL 迁移到 Git
107
+ python scripts/migrate_storage.py --from postgres --to git
108
+
109
+ # 导出当前数据到 JSON 文件
110
+ python scripts/migrate_storage.py --export backup.json
111
+
112
+ # 从 JSON 文件导入数据
113
+ python scripts/migrate_storage.py --import backup.json
114
+
115
+ 环境变量:
116
+ STORAGE_BACKEND - 存储后端类型 (json, sqlite, postgres, git)
117
+ DATABASE_URL - 数据库连接字符串
118
+ GIT_REPO_URL - Git 仓库地址
119
+ GIT_TOKEN - Git 访问令牌
120
+ """
121
+ )
122
+
123
+ parser.add_argument(
124
+ "--from",
125
+ dest="from_backend",
126
+ choices=["json", "sqlite", "postgres", "git"],
127
+ help="源存储后端",
128
+ )
129
+ parser.add_argument(
130
+ "--to",
131
+ dest="to_backend",
132
+ choices=["json", "sqlite", "postgres", "git"],
133
+ help="目标存储后端",
134
+ )
135
+ parser.add_argument(
136
+ "--export",
137
+ dest="export_file",
138
+ metavar="FILE",
139
+ help="导出数据到 JSON 文件",
140
+ )
141
+ parser.add_argument(
142
+ "--import",
143
+ dest="import_file",
144
+ metavar="FILE",
145
+ help="从 JSON 文件导入数据",
146
+ )
147
+
148
+ args = parser.parse_args()
149
+
150
+ # 检查参数
151
+ if args.from_backend and args.to_backend:
152
+ migrate_data(args.from_backend, args.to_backend)
153
+ elif args.export_file:
154
+ export_to_json(args.export_file)
155
+ elif args.import_file:
156
+ import_from_json(args.import_file)
157
+ else:
158
+ parser.print_help()
159
+ sys.exit(1)
160
+
161
+
162
+ if __name__ == "__main__":
163
+ main()
@@ -1,13 +1,13 @@
1
- listen-address 0.0.0.0:8118
2
- toggle 1
3
- enable-remote-toggle 0
4
- enable-remote-http-toggle 0
5
- enable-edit-actions 0
6
- enforce-blocks 0
7
- buffer-limit 4096
8
- keep-alive-timeout 300
9
- socket-timeout 300
10
-
11
- # Forward every HTTP/HTTPS request through the WARP SOCKS5 proxy.
12
- # The trailing dot is required by Privoxy syntax.
13
- forward-socks5t / warp-proxy:1080 .
1
+ listen-address 0.0.0.0:8118
2
+ toggle 1
3
+ enable-remote-toggle 0
4
+ enable-remote-http-toggle 0
5
+ enable-edit-actions 0
6
+ enforce-blocks 0
7
+ buffer-limit 4096
8
+ keep-alive-timeout 300
9
+ socket-timeout 300
10
+
11
+ # Forward every HTTP/HTTPS request through the WARP SOCKS5 proxy.
12
+ # The trailing dot is required by Privoxy syntax.
13
+ forward-socks5t / warp-proxy:1080 .