diff --git a/TeiBa/tb_signin.py b/TeiBa/tb_signin.py index 96b99a7..da9a56a 100644 --- a/TeiBa/tb_signin.py +++ b/TeiBa/tb_signin.py @@ -8,6 +8,7 @@ import hashlib import json import os import random +import sys import time from typing import Optional, Union @@ -280,12 +281,32 @@ def get_tieba_cookies(): print(f"从环境变量读取到 {len(cookies)} 个账号") return cookies +# 添加青龙脚本根目录到Python路径 +QL_SCRIPTS_DIR = '/ql/scripts' # 青龙脚本默认目录 +sys.path.append(QL_SCRIPTS_DIR) + +# 添加notify可能存在的其他路径 +POSSIBLE_PATHS = [ + '/ql', # 青龙根目录 + '/ql/data/scripts', # 新版青龙数据目录 + '/ql/scripts/notify', # 自定义通知目录 + os.path.dirname(__file__) # 当前脚本目录 +] + +for path in POSSIBLE_PATHS: + if os.path.exists(os.path.join(path, 'notify.py')): + sys.path.append(path) + break + # 发送青龙面板通知 def send_notification(title, content): try: # 尝试导入青龙的通知模块 from notify import send return send(title, content) + except ImportError: + print("⚠️ 无法加载通知模块,请检查路径配置") + return lambda title, content: None # 创建空函数防止报错 except Exception as e: print(f"通知发送失败: {e}") print(f"通知内容: {title}\n{content}")