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.
- package/LICENSE +21 -21
- package/README.en.md +5 -3
- package/README.md +7 -3
- package/client.js +241 -241
- package/cordis.patch.yml +14 -14
- package/host.js +51 -0
- package/install.mjs +333 -387
- package/package.json +3 -2
- package/scripts/fetch-style-images.mjs +104 -0
- package/scripts/vendor-snapshot.py +154 -154
- package/server/access.py +26 -26
- package/server/app.py +1371 -1329
- package/server/catalog.py +147 -147
- package/server/composer.py +391 -391
- package/server/drafts.py +158 -158
- package/server/history.py +575 -575
- package/server/imaging.py +172 -172
- package/server/pool.py +1335 -1335
- package/server/pool_admin.py +921 -921
- package/server/pool_service.py +1073 -1073
- package/server/requirements.txt +6 -6
- package/server/smoke_check.py +402 -402
- package/server/static/access.html +202 -202
- package/server/static/index.html +15 -0
- package/vendor/chatgpt2api/source/.env.example +82 -82
- package/vendor/chatgpt2api/source/.python-version +1 -1
- package/vendor/chatgpt2api/source/LICENSE +21 -21
- package/vendor/chatgpt2api/source/README.md +391 -391
- package/vendor/chatgpt2api/source/VERSION +1 -1
- package/vendor/chatgpt2api/source/api/__init__.py +2 -2
- package/vendor/chatgpt2api/source/api/accounts.py +537 -537
- package/vendor/chatgpt2api/source/api/ai.py +217 -217
- package/vendor/chatgpt2api/source/api/app.py +62 -62
- package/vendor/chatgpt2api/source/api/errors.py +46 -46
- package/vendor/chatgpt2api/source/api/image_inputs.py +309 -309
- package/vendor/chatgpt2api/source/api/image_tasks.py +120 -120
- package/vendor/chatgpt2api/source/api/support.py +134 -134
- package/vendor/chatgpt2api/source/api/system.py +365 -365
- package/vendor/chatgpt2api/source/assets/kunrouai.svg +50 -50
- package/vendor/chatgpt2api/source/config.default.json +108 -108
- package/vendor/chatgpt2api/source/main.py +9 -9
- package/vendor/chatgpt2api/source/pyproject.toml +27 -27
- package/vendor/chatgpt2api/source/scripts/init_proxy_config.py +186 -186
- package/vendor/chatgpt2api/source/scripts/migrate_storage.py +163 -163
- package/vendor/chatgpt2api/source/scripts/privoxy-warp.conf +13 -13
- package/vendor/chatgpt2api/source/scripts/test_outlook_token_mailbox.py +385 -385
- package/vendor/chatgpt2api/source/scripts/test_storage.py +129 -129
- package/vendor/chatgpt2api/source/scripts/verify_oauth_refresh.py +92 -92
- package/vendor/chatgpt2api/source/services/account_service.py +1757 -1757
- package/vendor/chatgpt2api/source/services/auth_service.py +253 -253
- package/vendor/chatgpt2api/source/services/backup_service.py +671 -671
- package/vendor/chatgpt2api/source/services/config.py +667 -667
- package/vendor/chatgpt2api/source/services/content_filter.py +242 -242
- package/vendor/chatgpt2api/source/services/cpa_service.py +315 -315
- package/vendor/chatgpt2api/source/services/editable_file_task_service.py +260 -260
- package/vendor/chatgpt2api/source/services/image_service.py +376 -376
- package/vendor/chatgpt2api/source/services/image_storage_service.py +405 -405
- package/vendor/chatgpt2api/source/services/image_tags_service.py +76 -76
- package/vendor/chatgpt2api/source/services/image_task_service.py +550 -550
- package/vendor/chatgpt2api/source/services/log_service.py +337 -337
- package/vendor/chatgpt2api/source/services/model_service.py +182 -182
- package/vendor/chatgpt2api/source/services/oauth_login_service.py +267 -267
- package/vendor/chatgpt2api/source/services/openai_backend_api.py +2763 -2763
- package/vendor/chatgpt2api/source/services/openai_oauth.py +32 -32
- package/vendor/chatgpt2api/source/services/protocol/__init__.py +2 -2
- package/vendor/chatgpt2api/source/services/protocol/anthropic_v1_messages.py +307 -307
- package/vendor/chatgpt2api/source/services/protocol/chat_completion_cache.py +238 -238
- package/vendor/chatgpt2api/source/services/protocol/conversation.py +1644 -1644
- package/vendor/chatgpt2api/source/services/protocol/error_response.py +124 -124
- package/vendor/chatgpt2api/source/services/protocol/openai_search.py +19 -19
- package/vendor/chatgpt2api/source/services/protocol/openai_v1_chat_complete.py +316 -316
- package/vendor/chatgpt2api/source/services/protocol/openai_v1_image_edit.py +86 -86
- package/vendor/chatgpt2api/source/services/protocol/openai_v1_image_generations.py +42 -42
- package/vendor/chatgpt2api/source/services/protocol/openai_v1_models.py +53 -53
- package/vendor/chatgpt2api/source/services/protocol/openai_v1_response.py +453 -453
- package/vendor/chatgpt2api/source/services/protocol/web_search_tool.py +160 -160
- package/vendor/chatgpt2api/source/services/proxy_service.py +648 -648
- package/vendor/chatgpt2api/source/services/storage/__init__.py +5 -5
- package/vendor/chatgpt2api/source/services/storage/base.py +38 -38
- package/vendor/chatgpt2api/source/services/storage/database_storage.py +194 -194
- package/vendor/chatgpt2api/source/services/storage/factory.py +104 -104
- package/vendor/chatgpt2api/source/services/storage/git_storage.py +183 -183
- package/vendor/chatgpt2api/source/services/storage/json_storage.py +95 -95
- package/vendor/chatgpt2api/source/services/sub2api_service.py +531 -531
- package/vendor/chatgpt2api/source/utils/helper.py +469 -469
- package/vendor/chatgpt2api/source/utils/image_tokens.py +346 -346
- package/vendor/chatgpt2api/source/utils/log.py +110 -110
- package/vendor/chatgpt2api/source/utils/pkce.py +19 -19
- package/vendor/chatgpt2api/source/utils/pow.py +207 -207
- package/vendor/chatgpt2api/source/utils/sentinel.py +159 -159
- package/vendor/chatgpt2api/source/utils/turnstile.py +184 -184
- package/vendor/chatgpt2api/source/uv.lock +904 -904
- package/vendor/chatgpt2api/source/web/bun.lock +1501 -1501
- package/vendor/chatgpt2api/source/web/components.json +24 -24
- package/vendor/chatgpt2api/source/web/eslint.config.mjs +17 -17
- package/vendor/chatgpt2api/source/web/next.config.ts +40 -40
- package/vendor/chatgpt2api/source/web/package.json +58 -58
- package/vendor/chatgpt2api/source/web/postcss.config.mjs +5 -5
- package/vendor/chatgpt2api/source/web/src/app/accounts/components/account-import-dialog.tsx +925 -925
- package/vendor/chatgpt2api/source/web/src/app/accounts/page.tsx +1307 -1307
- package/vendor/chatgpt2api/source/web/src/app/debug/components/chat-panel.tsx +224 -224
- package/vendor/chatgpt2api/source/web/src/app/debug/components/editable-file-panel.tsx +444 -444
- package/vendor/chatgpt2api/source/web/src/app/debug/components/ppt-panel.tsx +9 -9
- package/vendor/chatgpt2api/source/web/src/app/debug/components/psd-panel.tsx +9 -9
- package/vendor/chatgpt2api/source/web/src/app/debug/components/search-panel.tsx +172 -172
- package/vendor/chatgpt2api/source/web/src/app/debug/components/skill-panel.tsx +169 -169
- package/vendor/chatgpt2api/source/web/src/app/debug/components/types.ts +39 -39
- package/vendor/chatgpt2api/source/web/src/app/debug/page.tsx +59 -59
- package/vendor/chatgpt2api/source/web/src/app/globals.css +230 -230
- package/vendor/chatgpt2api/source/web/src/app/image/components/image-composer.tsx +542 -542
- package/vendor/chatgpt2api/source/web/src/app/image/components/image-results.tsx +554 -554
- package/vendor/chatgpt2api/source/web/src/app/image/components/image-sidebar.tsx +176 -176
- package/vendor/chatgpt2api/source/web/src/app/image/page.tsx +1799 -1799
- package/vendor/chatgpt2api/source/web/src/app/image-manager/page.tsx +732 -732
- package/vendor/chatgpt2api/source/web/src/app/layout.tsx +50 -50
- package/vendor/chatgpt2api/source/web/src/app/login/page.tsx +101 -101
- package/vendor/chatgpt2api/source/web/src/app/logs/page.tsx +356 -356
- package/vendor/chatgpt2api/source/web/src/app/page.tsx +30 -30
- package/vendor/chatgpt2api/source/web/src/app/settings/components/api-docs-card.tsx +341 -341
- package/vendor/chatgpt2api/source/web/src/app/settings/components/backup-settings-card.tsx +447 -447
- package/vendor/chatgpt2api/source/web/src/app/settings/components/base-url-card.tsx +71 -71
- package/vendor/chatgpt2api/source/web/src/app/settings/components/config-card.tsx +491 -491
- package/vendor/chatgpt2api/source/web/src/app/settings/components/cpa-pool-dialog.tsx +108 -108
- package/vendor/chatgpt2api/source/web/src/app/settings/components/cpa-pools-card.tsx +169 -169
- package/vendor/chatgpt2api/source/web/src/app/settings/components/import-browser-dialog.tsx +185 -185
- package/vendor/chatgpt2api/source/web/src/app/settings/components/proxy-runtime-card.tsx +379 -379
- package/vendor/chatgpt2api/source/web/src/app/settings/components/proxy-settings-card.tsx +127 -127
- package/vendor/chatgpt2api/source/web/src/app/settings/components/proxy-settings.tsx +200 -200
- package/vendor/chatgpt2api/source/web/src/app/settings/components/settings-header.tsx +12 -12
- package/vendor/chatgpt2api/source/web/src/app/settings/components/sub2api-connections.tsx +886 -886
- package/vendor/chatgpt2api/source/web/src/app/settings/components/third-party-apps-card.tsx +81 -81
- package/vendor/chatgpt2api/source/web/src/app/settings/components/user-keys-card.tsx +422 -422
- package/vendor/chatgpt2api/source/web/src/app/settings/page.tsx +135 -135
- package/vendor/chatgpt2api/source/web/src/app/settings/store.ts +1069 -1069
- package/vendor/chatgpt2api/source/web/src/components/date-range-filter.tsx +49 -49
- package/vendor/chatgpt2api/source/web/src/components/header-actions.tsx +24 -24
- package/vendor/chatgpt2api/source/web/src/components/image-lightbox.tsx +434 -434
- package/vendor/chatgpt2api/source/web/src/components/image-thumbnail.tsx +46 -46
- package/vendor/chatgpt2api/source/web/src/components/theme-script.tsx +18 -18
- package/vendor/chatgpt2api/source/web/src/components/theme-toggle.tsx +14 -14
- package/vendor/chatgpt2api/source/web/src/components/top-nav.tsx +263 -263
- package/vendor/chatgpt2api/source/web/src/components/ui/animated-theme-toggler.tsx +256 -256
- package/vendor/chatgpt2api/source/web/src/components/ui/badge.tsx +41 -41
- package/vendor/chatgpt2api/source/web/src/components/ui/button.tsx +59 -59
- package/vendor/chatgpt2api/source/web/src/components/ui/calendar.tsx +46 -46
- package/vendor/chatgpt2api/source/web/src/components/ui/card.tsx +61 -61
- package/vendor/chatgpt2api/source/web/src/components/ui/checkbox.tsx +27 -27
- package/vendor/chatgpt2api/source/web/src/components/ui/dia-text-reveal.tsx +267 -267
- package/vendor/chatgpt2api/source/web/src/components/ui/dialog.tsx +132 -132
- package/vendor/chatgpt2api/source/web/src/components/ui/field.tsx +13 -13
- package/vendor/chatgpt2api/source/web/src/components/ui/input.tsx +19 -19
- package/vendor/chatgpt2api/source/web/src/components/ui/label.tsx +24 -24
- package/vendor/chatgpt2api/source/web/src/components/ui/morphing-text.tsx +149 -149
- package/vendor/chatgpt2api/source/web/src/components/ui/popover.tsx +38 -38
- package/vendor/chatgpt2api/source/web/src/components/ui/select.tsx +175 -175
- package/vendor/chatgpt2api/source/web/src/components/ui/sheet.tsx +143 -143
- package/vendor/chatgpt2api/source/web/src/components/ui/table.tsx +29 -29
- package/vendor/chatgpt2api/source/web/src/components/ui/tabs.tsx +91 -91
- package/vendor/chatgpt2api/source/web/src/components/ui/text-3d-flip.tsx +269 -269
- package/vendor/chatgpt2api/source/web/src/components/ui/textarea.tsx +18 -18
- package/vendor/chatgpt2api/source/web/src/components/version-release-dialog.tsx +127 -127
- package/vendor/chatgpt2api/source/web/src/constants/common-env.ts +6 -6
- package/vendor/chatgpt2api/source/web/src/hooks/use-version-check.ts +85 -85
- package/vendor/chatgpt2api/source/web/src/lib/api.ts +926 -926
- package/vendor/chatgpt2api/source/web/src/lib/auth-session.ts +26 -26
- package/vendor/chatgpt2api/source/web/src/lib/release.ts +25 -25
- package/vendor/chatgpt2api/source/web/src/lib/request.ts +93 -93
- package/vendor/chatgpt2api/source/web/src/lib/use-auth-guard.ts +89 -89
- package/vendor/chatgpt2api/source/web/src/lib/utils.ts +6 -6
- package/vendor/chatgpt2api/source/web/src/store/auth.ts +112 -112
- package/vendor/chatgpt2api/source/web/src/store/editable-file-history.ts +33 -33
- package/vendor/chatgpt2api/source/web/src/store/image-conversations.ts +318 -318
- 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 .
|