dsh-plugin-image-workbench 1.1.5 → 1.1.6

package/vendor/chatgpt2api/source/scripts/test_storage.py

@@ -1,129 +1,129 @@
-#!/usr/bin/env python3
-"""
-存储后端测试脚本
-
-用法：
-  python scripts/test_storage.py
-"""
-
-import os
-import sys
-from pathlib import Path
-
-# 添加项目根目录到 Python 路径
-sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
-
-DATA_DIR = Path(__file__).resolve().parents[1] / "data"
-
-from services.storage.factory import create_storage_backend
-
-
-def test_storage():
-    """测试当前配置的存储后端"""
-    DATA_DIR.mkdir(parents=True, exist_ok=True)
-    print("=" * 60)
-    print("ChatGPT2API 存储后端测试")
-    print("=" * 60)
-    
-    # 显示当前配置
-    backend_type = os.getenv("STORAGE_BACKEND", "json")
-    print(f"\n当前存储后端: {backend_type}")
-    
-    if backend_type in ("sqlite", "postgres", "postgresql", "mysql", "database"):
-        database_url = os.getenv("DATABASE_URL", "")
-        if database_url:
-            # 隐藏密码
-            if "://" in database_url and "@" in database_url:
-                protocol, rest = database_url.split("://", 1)
-                if "@" in rest:
-                    credentials, host = rest.split("@", 1)
-                    if ":" in credentials:
-                        username, _ = credentials.split(":", 1)
-                        database_url = f"{protocol}://{username}:****@{host}"
-            print(f"数据库连接: {database_url}")
-        else:
-            print(f"数据库连接: 本地 SQLite (data/accounts.db)")
-    
-    elif backend_type == "git":
-        repo_url = os.getenv("GIT_REPO_URL", "")
-        branch = os.getenv("GIT_BRANCH", "main")
-        file_path = os.getenv("GIT_FILE_PATH", "accounts.json")
-        print(f"Git 仓库: {repo_url}")
-        print(f"Git 分支: {branch}")
-        print(f"文件路径: {file_path}")
-    
-    print("\n" + "=" * 60)
-    
-    try:
-        # 创建存储后端
-        print("\n[1/5] 创建存储后端...")
-        storage = create_storage_backend(DATA_DIR)
-        print("✅ 存储后端创建成功")
-        
-        # 获取后端信息
-        print("\n[2/5] 获取后端信息...")
-        info = storage.get_backend_info()
-        print(f"✅ 后端类型: {info.get('type')}")
-        print(f"   描述: {info.get('description')}")
-        for key, value in info.items():
-            if key not in ('type', 'description'):
-                print(f"   {key}: {value}")
-        
-        # 健康检查
-        print("\n[3/5] 执行健康检查...")
-        health = storage.health_check()
-        status = health.get("status")
-        if status == "healthy":
-            print(f"✅ 健康状态: {status}")
-        else:
-            print(f"❌ 健康状态: {status}")
-            print(f"   错误: {health.get('error')}")
-            return False
-        
-        # 读取数据
-        print("\n[4/5] 读取账号数据...")
-        accounts = storage.load_accounts()
-        print(f"✅ 成功读取 {len(accounts)} 个账号")
-        
-        # 写入测试（可选）
-        print("\n[5/5] 测试写入功能...")
-        test_account = {
-            "access_token": "test_token_" + str(os.getpid()),
-            "type": "Free",
-            "status": "测试",
-            "quota": 0,
-            "email": "test@example.com",
-        }
-        
-        # 添加测试账号
-        test_accounts = accounts + [test_account]
-        storage.save_accounts(test_accounts)
-        print("✅ 写入测试账号成功")
-        
-        # 验证写入
-        reloaded = storage.load_accounts()
-        if len(reloaded) == len(test_accounts):
-            print("✅ 验证写入成功")
-        else:
-            print(f"❌ 验证失败: 期望 {len(test_accounts)} 个账号，实际 {len(reloaded)} 个")
-            return False
-        
-        # 恢复原始数据
-        storage.save_accounts(accounts)
-        print("✅ 恢复原始数据")
-        
-        print("\n" + "=" * 60)
-        print("✅ 所有测试通过！")
-        print("=" * 60)
-        return True
-        
-    except Exception as e:
-        print(f"\n❌ 测试失败: {e}")
-        import traceback
-        traceback.print_exc()
-        return False
-
-
-if __name__ == "__main__":
-    success = test_storage()
-    sys.exit(0 if success else 1)
+#!/usr/bin/env python3
+"""
+存储后端测试脚本
+
+用法：
+  python scripts/test_storage.py
+"""
+
+import os
+import sys
+from pathlib import Path
+
+# 添加项目根目录到 Python 路径
+sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
+
+DATA_DIR = Path(__file__).resolve().parents[1] / "data"
+
+from services.storage.factory import create_storage_backend
+
+
+def test_storage():
+    """测试当前配置的存储后端"""
+    DATA_DIR.mkdir(parents=True, exist_ok=True)
+    print("=" * 60)
+    print("ChatGPT2API 存储后端测试")
+    print("=" * 60)
+    
+    # 显示当前配置
+    backend_type = os.getenv("STORAGE_BACKEND", "json")
+    print(f"\n当前存储后端: {backend_type}")
+    
+    if backend_type in ("sqlite", "postgres", "postgresql", "mysql", "database"):
+        database_url = os.getenv("DATABASE_URL", "")
+        if database_url:
+            # 隐藏密码
+            if "://" in database_url and "@" in database_url:
+                protocol, rest = database_url.split("://", 1)
+                if "@" in rest:
+                    credentials, host = rest.split("@", 1)
+                    if ":" in credentials:
+                        username, _ = credentials.split(":", 1)
+                        database_url = f"{protocol}://{username}:****@{host}"
+            print(f"数据库连接: {database_url}")
+        else:
+            print(f"数据库连接: 本地 SQLite (data/accounts.db)")
+    
+    elif backend_type == "git":
+        repo_url = os.getenv("GIT_REPO_URL", "")
+        branch = os.getenv("GIT_BRANCH", "main")
+        file_path = os.getenv("GIT_FILE_PATH", "accounts.json")
+        print(f"Git 仓库: {repo_url}")
+        print(f"Git 分支: {branch}")
+        print(f"文件路径: {file_path}")
+    
+    print("\n" + "=" * 60)
+    
+    try:
+        # 创建存储后端
+        print("\n[1/5] 创建存储后端...")
+        storage = create_storage_backend(DATA_DIR)
+        print("✅ 存储后端创建成功")
+        
+        # 获取后端信息
+        print("\n[2/5] 获取后端信息...")
+        info = storage.get_backend_info()
+        print(f"✅ 后端类型: {info.get('type')}")
+        print(f"   描述: {info.get('description')}")
+        for key, value in info.items():
+            if key not in ('type', 'description'):
+                print(f"   {key}: {value}")
+        
+        # 健康检查
+        print("\n[3/5] 执行健康检查...")
+        health = storage.health_check()
+        status = health.get("status")
+        if status == "healthy":
+            print(f"✅ 健康状态: {status}")
+        else:
+            print(f"❌ 健康状态: {status}")
+            print(f"   错误: {health.get('error')}")
+            return False
+        
+        # 读取数据
+        print("\n[4/5] 读取账号数据...")
+        accounts = storage.load_accounts()
+        print(f"✅ 成功读取 {len(accounts)} 个账号")
+        
+        # 写入测试（可选）
+        print("\n[5/5] 测试写入功能...")
+        test_account = {
+            "access_token": "test_token_" + str(os.getpid()),
+            "type": "Free",
+            "status": "测试",
+            "quota": 0,
+            "email": "test@example.com",
+        }
+        
+        # 添加测试账号
+        test_accounts = accounts + [test_account]
+        storage.save_accounts(test_accounts)
+        print("✅ 写入测试账号成功")
+        
+        # 验证写入
+        reloaded = storage.load_accounts()
+        if len(reloaded) == len(test_accounts):
+            print("✅ 验证写入成功")
+        else:
+            print(f"❌ 验证失败: 期望 {len(test_accounts)} 个账号，实际 {len(reloaded)} 个")
+            return False
+        
+        # 恢复原始数据
+        storage.save_accounts(accounts)
+        print("✅ 恢复原始数据")
+        
+        print("\n" + "=" * 60)
+        print("✅ 所有测试通过！")
+        print("=" * 60)
+        return True
+        
+    except Exception as e:
+        print(f"\n❌ 测试失败: {e}")
+        import traceback
+        traceback.print_exc()
+        return False
+
+
+if __name__ == "__main__":
+    success = test_storage()
+    sys.exit(0 if success else 1)

package/vendor/chatgpt2api/source/scripts/verify_oauth_refresh.py

@@ -1,92 +1,92 @@
-"""验证 OAuth 账号的自动刷新是否生效。
-
-用法（在容器内，工作目录 /app）：
-    uv run python scripts/verify_oauth_refresh.py            # 只读诊断，安全
-    uv run python scripts/verify_oauth_refresh.py --force    # 真正触发一次刷新
-
-只读模式：列出每个账号的 access_token 剩余有效期、是否带 refresh_token、
-          上次刷新时间与上次刷新错误，判断"有没有料可刷"。
---force ：对每个带 refresh_token 的账号强制走一次 refresh_access_token(force=True)，
-          直接验证 refresh_token + 本项目 client_id 能否从 OpenAI 换出新 access_token。
-          这会真实轮换 access_token（新 token 有效，不损坏账号）。
-"""
-from __future__ import annotations
-
-import sys
-
-from services.account_service import account_service
-
-
-def _fmt_remaining(seconds: int | None) -> str:
-    if seconds is None:
-        return "无法解析 exp"
-    if seconds <= 0:
-        return f"已过期 {(-seconds) // 3600}h"
-    return f"{seconds // 3600}h{(seconds % 3600) // 60}m 后过期"
-
-
-def diagnose() -> list[str]:
-    """只读：打印每个账号的刷新就绪状态，返回带 refresh_token 的 token 列表。"""
-    tokens = account_service.list_tokens()
-    print(f"账号总数: {len(tokens)}\n")
-    refreshable: list[str] = []
-    for token in tokens:
-        account = account_service.get_account(token) or {}
-        remaining = account_service._token_expires_in(token)
-        has_rt = bool(str(account.get("refresh_token") or "").strip())
-        if has_rt:
-            refreshable.append(token)
-        print(f"- email={account.get('email') or '(未知)'}")
-        print(f"    access_token[:20]   = {token[:20]}...")
-        print(f"    距过期              = {_fmt_remaining(remaining)}")
-        print(f"    refresh_token       = {'有 ✅' if has_rt else '无 ❌（无法自动刷新）'}")
-        print(f"    last_token_refresh_at    = {account.get('last_token_refresh_at')}")
-        print(f"    last_token_refresh_error = {account.get('last_token_refresh_error')}")
-        print()
-    return refreshable
-
-
-def force_refresh(tokens: list[str]) -> None:
-    """对每个账号 force 刷新一次，并对比前后状态判断成败。"""
-    if not tokens:
-        print("没有带 refresh_token 的账号，无法验证刷新。")
-        return
-    print("=" * 60)
-    print(f"开始对 {len(tokens)} 个账号 force 刷新（真实调用 OpenAI）...\n")
-    ok = 0
-    for token in tokens:
-        before = account_service.get_account(token) or {}
-        new_token = account_service.refresh_access_token(token, force=True, event="manual_verify")
-        after = account_service.get_account(new_token) or {}
-        err = str(after.get("last_token_refresh_error") or "").strip()
-        rotated = new_token != token
-        success = bool(new_token) and not err
-        if success:
-            ok += 1
-        print(f"- email={before.get('email') or '(未知)'}")
-        print(f"    旧 access_token[:20] = {token[:20]}...")
-        print(f"    新 access_token[:20] = {new_token[:20]}...")
-        print(f"    token 是否轮换       = {'是' if rotated else '否（exp 未到刷新窗口时可能返回原值）'}")
-        print(f"    last_token_refresh_at    = {after.get('last_token_refresh_at')}")
-        print(f"    last_token_refresh_error = {after.get('last_token_refresh_error') or '无'}")
-        print(f"    >>> 刷新结果         = {'成功 ✅' if success else '失败 ❌'}")
-        print()
-    print("=" * 60)
-    print(f"汇总: {ok}/{len(tokens)} 个账号刷新成功")
-    if ok == len(tokens):
-        print("✅ 自动刷新机制对这些账号完全可用——refresh_token 与 client_id 匹配。")
-    else:
-        print("❌ 有账号刷新失败，看上面的 last_token_refresh_error，或 docker logs 里的 [oauth-login]/refresh 日志。")
-
-
-def main() -> None:
-    do_force = "--force" in sys.argv[1:]
-    refreshable = diagnose()
-    if do_force:
-        force_refresh(refreshable)
-    else:
-        print("提示: 加 --force 参数可真正触发一次刷新以验证能否成功。")
-
-
-if __name__ == "__main__":
-    main()
+"""验证 OAuth 账号的自动刷新是否生效。
+
+用法（在容器内，工作目录 /app）：
+    uv run python scripts/verify_oauth_refresh.py            # 只读诊断，安全
+    uv run python scripts/verify_oauth_refresh.py --force    # 真正触发一次刷新
+
+只读模式：列出每个账号的 access_token 剩余有效期、是否带 refresh_token、
+          上次刷新时间与上次刷新错误，判断"有没有料可刷"。
+--force ：对每个带 refresh_token 的账号强制走一次 refresh_access_token(force=True)，
+          直接验证 refresh_token + 本项目 client_id 能否从 OpenAI 换出新 access_token。
+          这会真实轮换 access_token（新 token 有效，不损坏账号）。
+"""
+from __future__ import annotations
+
+import sys
+
+from services.account_service import account_service
+
+
+def _fmt_remaining(seconds: int | None) -> str:
+    if seconds is None:
+        return "无法解析 exp"
+    if seconds <= 0:
+        return f"已过期 {(-seconds) // 3600}h"
+    return f"{seconds // 3600}h{(seconds % 3600) // 60}m 后过期"
+
+
+def diagnose() -> list[str]:
+    """只读：打印每个账号的刷新就绪状态，返回带 refresh_token 的 token 列表。"""
+    tokens = account_service.list_tokens()
+    print(f"账号总数: {len(tokens)}\n")
+    refreshable: list[str] = []
+    for token in tokens:
+        account = account_service.get_account(token) or {}
+        remaining = account_service._token_expires_in(token)
+        has_rt = bool(str(account.get("refresh_token") or "").strip())
+        if has_rt:
+            refreshable.append(token)
+        print(f"- email={account.get('email') or '(未知)'}")
+        print(f"    access_token[:20]   = {token[:20]}...")
+        print(f"    距过期              = {_fmt_remaining(remaining)}")
+        print(f"    refresh_token       = {'有 ✅' if has_rt else '无 ❌（无法自动刷新）'}")
+        print(f"    last_token_refresh_at    = {account.get('last_token_refresh_at')}")
+        print(f"    last_token_refresh_error = {account.get('last_token_refresh_error')}")
+        print()
+    return refreshable
+
+
+def force_refresh(tokens: list[str]) -> None:
+    """对每个账号 force 刷新一次，并对比前后状态判断成败。"""
+    if not tokens:
+        print("没有带 refresh_token 的账号，无法验证刷新。")
+        return
+    print("=" * 60)
+    print(f"开始对 {len(tokens)} 个账号 force 刷新（真实调用 OpenAI）...\n")
+    ok = 0
+    for token in tokens:
+        before = account_service.get_account(token) or {}
+        new_token = account_service.refresh_access_token(token, force=True, event="manual_verify")
+        after = account_service.get_account(new_token) or {}
+        err = str(after.get("last_token_refresh_error") or "").strip()
+        rotated = new_token != token
+        success = bool(new_token) and not err
+        if success:
+            ok += 1
+        print(f"- email={before.get('email') or '(未知)'}")
+        print(f"    旧 access_token[:20] = {token[:20]}...")
+        print(f"    新 access_token[:20] = {new_token[:20]}...")
+        print(f"    token 是否轮换       = {'是' if rotated else '否（exp 未到刷新窗口时可能返回原值）'}")
+        print(f"    last_token_refresh_at    = {after.get('last_token_refresh_at')}")
+        print(f"    last_token_refresh_error = {after.get('last_token_refresh_error') or '无'}")
+        print(f"    >>> 刷新结果         = {'成功 ✅' if success else '失败 ❌'}")
+        print()
+    print("=" * 60)
+    print(f"汇总: {ok}/{len(tokens)} 个账号刷新成功")
+    if ok == len(tokens):
+        print("✅ 自动刷新机制对这些账号完全可用——refresh_token 与 client_id 匹配。")
+    else:
+        print("❌ 有账号刷新失败，看上面的 last_token_refresh_error，或 docker logs 里的 [oauth-login]/refresh 日志。")
+
+
+def main() -> None:
+    do_force = "--force" in sys.argv[1:]
+    refreshable = diagnose()
+    if do_force:
+        force_refresh(refreshable)
+    else:
+        print("提示: 加 --force 参数可真正触发一次刷新以验证能否成功。")
+
+
+if __name__ == "__main__":
+    main()