From 2e761c690ba63e4c9fe925b6753d7ec0d6c2137a Mon Sep 17 00:00:00 2001
From: UPToZ <uptoz@qq.com>
Date: Tue, 3 Jun 2025 09:38:20 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E9=80=9A=E7=9F=A5=E5=BC=82?=
 =?UTF-8?q?=E5=B8=B8=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 TeiBa/tb_signin.py | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

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}")