Kris
102b98adea
本次提交将woc-bridge项目重构为模块化包结构，按职责拆分多个子域： 1. 新增models层定义所有Pydantic数据模型与统一错误体系 2. 拆分db/ui/messaging/routes等业务域模块 3. 实现基础API路由：状态查询、截图、登录、媒体获取等 4. 重构tools脚本的模块导入路径 5. 补充版本号与能力清单定义 6. 完善全局配置与依赖管理 整体完成项目从单文件脚本到可维护的包结构迁移，为后续功能开发打下基础。
69 lines
2.2 KiB
Python
69 lines
2.2 KiB
Python
"""基础模型：错误码 / BridgeError 异常 / ErrorResponse。"""
|
||
|
||
from __future__ import annotations
|
||
|
||
from typing import Any, Optional
|
||
|
||
from pydantic import BaseModel, Field
|
||
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# 错误码定义（与 spec 一致）
|
||
# ---------------------------------------------------------------------------
|
||
# 每个错误码对应一个 HTTP 状态码
|
||
ERROR_CODES: dict[str, int] = {
|
||
"WECHAT_NOT_RUNNING": 503,
|
||
"WECHAT_NOT_LOGGED_IN": 401,
|
||
"WINDOW_NOT_FOUND": 503,
|
||
"CONTACT_NOT_FOUND": 404,
|
||
"SEND_FAILED": 500,
|
||
"DB_LOCKED": 503,
|
||
"DB_NOT_FOUND": 500,
|
||
"INVALID_PARAMS": 400,
|
||
"BRIDGE_INTERNAL_ERROR": 500,
|
||
"LOGIN_TIMEOUT": 408,
|
||
"MEDIA_NOT_FOUND": 404,
|
||
"RATE_LIMITED": 429,
|
||
"DB_ENCRYPTED": 503,
|
||
"DB_NEED_INIT": 503,
|
||
"DB_INIT_IN_PROGRESS": 503,
|
||
"DB_KEY_INVALID": 503,
|
||
"LOGOUT_FAILED": 500,
|
||
"RESTART_TIMEOUT": 408,
|
||
"REVOKE_WINDOW_EXPIRED": 409,
|
||
}
|
||
|
||
|
||
class BridgeError(Exception):
|
||
"""bridge 统一业务异常。
|
||
|
||
路由层直接 raise BridgeError(code="...", message="...", details=...)，
|
||
由全局异常处理器捕获后转换为统一 ErrorResponse 并设置对应 HTTP 状态码。
|
||
"""
|
||
|
||
def __init__(
|
||
self,
|
||
code: str = "BRIDGE_INTERNAL_ERROR",
|
||
message: str = "bridge internal error",
|
||
http_status: Optional[int] = None,
|
||
details: Optional[Any] = None,
|
||
) -> None:
|
||
self.code = code
|
||
self.message = message
|
||
# 若未显式指定 http_status，则查表取默认值；查不到则回落到 500
|
||
self.http_status = http_status if http_status is not None else ERROR_CODES.get(code, 500)
|
||
self.details = details
|
||
super().__init__(f"[{code}] {message}")
|
||
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# 通用错误响应
|
||
# ---------------------------------------------------------------------------
|
||
class ErrorResponse(BaseModel):
|
||
"""统一错误响应结构。"""
|
||
|
||
success: bool = Field(default=False, description="固定为 false")
|
||
error: dict[str, Any] = Field(
|
||
description="错误详情，含 code/message/details 三个字段"
|
||
)
|