bootstrap.py

"""运行时补丁：包装 save_account_to_db 并在成功后发送 webhook。"""

from __future__ import annotations

import json
import os
import sys
import traceback
from urllib import error, request
from message_producer import publish_to_queue

print("正在加载 bootstrap.py...")


def _log(message: str) -> None:
    print(f"[Hook] {message}", file=sys.stderr)


def _install_patch() -> None:
    import utils.db_manager as db_manager

    if getattr(db_manager.save_account_to_db, "__sitecustomize_patched__", False):
        _log("save_account_to_db 已补丁，跳过")
        return

    original_save = db_manager.save_account_to_db

    def patched_save_account_to_db(
        email: str, password: str, token_json_str: str
    ) -> bool:
        ok = original_save(email, password, token_json_str)
        if ok:
            print(f"save_account_to_db 成功，准备发送消息: {email}")
            try:
                publish_to_queue(
                    {
                        "action": "create",
                        "data": {
                            "email": email,
                            "password": password,
                            "token": token_json_str,
                        },
                    }
                )
            except error.URLError as exc:
                _log(f"发送消息失败: {exc}")
            except Exception:
                _log("执行异常")
                traceback.print_exc()
        return ok

    patched_save_account_to_db.__sitecustomize_patched__ = True
    db_manager.save_account_to_db = patched_save_account_to_db
    _log("save_account_to_db 补丁已安装")


try:
    _install_patch()
except Exception:
    _log("补丁安装失败")
    traceback.print_exc()