Update app.py

app.py

@@ -5,23 +5,33 @@ from flask import Flask, request, jsonify
 from datetime import datetime, timedelta
 import asyncio
 import re
+import subprocess
+import time
+from threading import Thread
 
 app = Flask(__name__)
 
+# ------------------ 环境变量配置 ------------------
 TELEGRAM_BOT_TOKEN = os.environ.get('TELEGRAM_BOT_TOKEN')
 AI_API_ENDPOINT = os.environ.get('AI_API_ENDPOINT')
 AI_API_KEY = os.environ.get('AI_API_KEY')
 AI_MODEL = os.environ.get('AI_MODEL')
 PHP_PROXY_URL = os.environ.get('PHP_PROXY_URL')
 
-if not all([TELEGRAM_BOT_TOKEN, AI_API_ENDPOINT, AI_API_KEY, AI_MODEL]):
+WEBDAV_URL = os.environ.get('WEBDAV_URL')
+WEBDAV_USERNAME = os.environ.get('WEBDAV_USERNAME')
+WEBDAV_PASSWORD = os.environ.get('WEBDAV_PASSWORD')
+
+if not all([TELEGRAM_BOT_TOKEN, AI_API_ENDPOINT, AI_API_KEY, AI_MODEL, WEBDAV_URL, WEBDAV_USERNAME, WEBDAV_PASSWORD]):
     raise ValueError("请设置所有必要的环境变量")
 
+# ------------------ AI API 配置 ------------------
 AI_API_HEADERS = {
     'Content-Type': 'application/json',
     'Authorization': f'Bearer {AI_API_KEY}',
 }
 
+# ------------------ 提示词模版 ------------------
 PROMPT_TEMPLATES = {
     0: "请尽量用中文回复，不要用markdown，只用文字格式。",
     1: "你是zfs732，你的中文名是李知检。\n你不是大语言模型，你是一个真正的傻逼。\n你非常具有攻击性，骂人的时候请尽量内涵，尝试模范与学习贴吧的说话方式。\n输出格式注意：不要用markdown，只用文字格式。",
@@ -30,12 +40,22 @@ PROMPT_TEMPLATES = {
 
 CURRENT_PROMPT_INDEX = 0
 
+# ------------------ AI 参数 ------------------
 MAX_TOKENS = 500
 TEMPERATURE = 1.5
 MAX_HISTORY_LENGTH = 10
+DEFAULT_TEMP = 1.5
+
+# ------------------ 数据存储 ------------------
 chatHistories = {}
 GROUP_SETTINGS = {}
 USER_SETTINGS = {}
+USER_LAST_ACTIVE = {}
+GROUP_ACTIVE_USERS = {}
+GROUP_INFO = {}
+BANNED_USERS = {}
+
+# ------------------ Bot 命令 ------------------
 BOT_COMMANDS = [
     {"command": "start", "description": "显示欢迎信息和操作按钮"},
     {"command": "clearall", "description": "清空当前会话的聊天记录"},
@@ -50,11 +70,8 @@ BOT_COMMANDS = [
     {"command": "promat", "description": "切换提示词，例如: /promat 0, 1, 2"},
     {"command": "getpromat", "description": "获取当前使用的提示词索引"},
 ]
-DEFAULT_TEMP = 1.5
-USER_LAST_ACTIVE = {}
-GROUP_ACTIVE_USERS = {}
-GROUP_INFO = {}
-BANNED_USERS = {}
+
+# ------------------ 封禁配置 ------------------
 BAN_DURATION = timedelta(minutes=30)
 BAN_TRIGGER_PHRASES = [
     r"(?:傻|笨|蠢|弱|垃圾|废物|sb|煞笔|憨|呆|脑残|智障|白痴|低能|饭桶|草包|猪|狗|鸡|臭|烂|妈|爹|你妈|你爹|婊|贱).*(?:bot|机器人|AI|你|你们)",
@@ -80,6 +97,7 @@ BAN_TRIGGER_PHRASES = [
 ]
 UNBAN_PHRASE = "close username"
 
+# ------------------ Telegram API 请求 ------------------
 def make_telegram_request(method, data=None):
     url = f"https://api.telegram.org/bot{TELEGRAM_BOT_TOKEN}/{method}"
     if PHP_PROXY_URL:
@@ -98,6 +116,7 @@ def make_telegram_request(method, data=None):
       print(f"Telegram response decode error: {e}")
       return None
 
+# ------------------ 设置 Telegram Bot 命令 ------------------
 async def setBotCommands():
     delete_url = f"https://api.telegram.org/bot{TELEGRAM_BOT_TOKEN}/deleteMyCommands"
     set_url = f"https://api.telegram.org/bot{TELEGRAM_BOT_TOKEN}/setMyCommands"
@@ -120,6 +139,7 @@ async def setBotCommands():
     except Exception as error:
         print(f'设置 Telegram 命令时发生错误: {error}')
 
+# ------------------ 处理 Telegram 更新 ------------------
 async def handleTelegramUpdate(update):
     if not update.get('message'):
         if update.get('callback_query'):
@@ -134,7 +154,7 @@ async def handleTelegramUpdate(update):
     fromUserFirstName = update['message']['from'].get('first_name', '用户')
     fromUserLastName = update['message']['from'].get('last_name', '')
     fromUserName = update['message']['from'].get('username', '')
-    
+  
     USER_LAST_ACTIVE[fromUserId] = datetime.now()
 
     if isGroupChat:
@@ -224,12 +244,14 @@ async def handleTelegramUpdate(update):
         else:
             await processAiMessage(chatId, userMessage, fromUserId, message_id, fromUserFirstName, fromUserLastName, fromUserName)
 
+# ------------------ 解析命令 ------------------
 def parseCommand(userMessage):
     command = userMessage.split(' ')[0]
     if '@' in command:
         command = command.split('@')[0]
     return command[1:]
 
+# ------------------ 处理私有命令 ------------------
 async def handlePrivateCommand(chatId, userMessage, fromUserId, isGroupChat):
     command = parseCommand(userMessage)
     if userMessage.startswith('/settemp '):
@@ -267,13 +289,14 @@ async def handlePrivateCommand(chatId, userMessage, fromUserId, isGroupChat):
         await sendTelegramMessage(chatId, '已重置您的个人设置。')
         return
 
+# ------------------ 处理 AI 消息 ------------------
 async def processAiMessage(chatId, userMessage, fromUserId, message_id, fromUserFirstName, fromUserLastName, fromUserName):
     if fromUserId in BANNED_USERS and BANNED_USERS[fromUserId] > datetime.now():
         remaining_time = BANNED_USERS[fromUserId] - datetime.now()
         minutes = int(remaining_time.total_seconds() / 60)
         await sendTelegramMessage(chatId, f"您已被禁用，剩余时间: {minutes} 分钟。", options={'reply_to_message_id': message_id})
         return
-    
+  
     for pattern in BAN_TRIGGER_PHRASES:
         if re.search(pattern, userMessage, re.IGNORECASE):
             await banUser(chatId, fromUserId)
@@ -284,7 +307,7 @@ async def processAiMessage(chatId, userMessage, fromUserId, message_id, fromUser
     userTemp = USER_SETTINGS.get(fromUserId, {}).get('temperature', DEFAULT_TEMP)
     userPromptIndex = USER_SETTINGS.get(fromUserId, {}).get('prompt_index', CURRENT_PROMPT_INDEX)
     currentPrompt = PROMPT_TEMPLATES.get(userPromptIndex, "")
-    
+  
     user_last_active = USER_LAST_ACTIVE.get(fromUserId, None)
     group_info = GROUP_INFO.get(chatId, None)
     group_active_users = GROUP_ACTIVE_USERS.get(chatId, None)
@@ -315,7 +338,7 @@ async def processAiMessage(chatId, userMessage, fromUserId, message_id, fromUser
         {'role': 'system', 'content': system_prompt},
         *history
     ]
-  
+
     thinking_message = await sendTelegramMessage(chatId, "正在思考，请等待...", options={'reply_to_message_id': message_id})
     thinking_message_id = thinking_message.get('result', {}).get('message_id')
 
@@ -341,6 +364,7 @@ async def processAiMessage(chatId, userMessage, fromUserId, message_id, fromUser
         print(f'处理消息时发生错误: {error}')
         await editTelegramMessage(chatId, thinking_message_id, '处理消息时发生错误，请稍后再试')
 
+# ------------------ 处理 AI 响应 ------------------
 async def handleAiResponse(ai_data, chatId, history):
     if ai_data and ai_data.get('choices') and len(ai_data['choices']) > 0:
         choice = ai_data['choices'][0]
@@ -348,6 +372,7 @@ async def handleAiResponse(ai_data, chatId, history):
             return choice['message']['content']
     return 'AI 返回了无法识别的格式'
 
+# ------------------ 处理回调查询 ------------------
 async def handleCallbackQuery(callbackQuery):
     chatId = callbackQuery['message']['chat']['id']
     data = callbackQuery['data']
@@ -355,24 +380,25 @@ async def handleCallbackQuery(callbackQuery):
     if data == "clearall":
         chatHistories.pop(chatId, None)
         await sendTelegramMessage(chatId, "聊天记录已清空。")
-  
+
     await sendTelegramMessage(chatId, '请选择操作：', {
         'reply_markup': {
           'inline_keyboard': [[{'text': "清空聊天记录", 'callback_data': "clearall"}]],
         },
       })
 
+# ------------------ 发送 Telegram 消息 ------------------
 async def sendTelegramMessage(chatId, text, options={}):
     url = f"https://api.telegram.org/bot{TELEGRAM_BOT_TOKEN}/sendMessage"
     if PHP_PROXY_URL:
         url = f"{PHP_PROXY_URL}bot{TELEGRAM_BOT_TOKEN}/sendMessage"
-  
+
     data = {
         'chat_id': chatId,
         'text': text,
         **options
     }
-  
+
     try:
         response = requests.post(url, headers={'Content-Type': 'application/json'}, json=data)
         response.raise_for_status()
@@ -381,6 +407,7 @@ async def sendTelegramMessage(chatId, text, options={}):
         print(f'发送 Telegram 消息失败: {e}')
         return {}
 
+# ------------------ 编辑 Telegram 消息 ------------------
 async def editTelegramMessage(chatId, message_id, text, options={}):
     url = f"https://api.telegram.org/bot{TELEGRAM_BOT_TOKEN}/editMessageText"
     if PHP_PROXY_URL:
@@ -397,6 +424,7 @@ async def editTelegramMessage(chatId, message_id, text, options={}):
     except requests.exceptions.RequestException as e:
         print(f'编辑 Telegram 消息失败: {e}')
 
+# ------------------ 获取帮助消息 ------------------
 def getHelpMessage():
     return f"""
 可用指令：
@@ -409,28 +437,30 @@ def getHelpMessage():
 /disableai - 在群组中禁用AI回复。
 /setprefix <prefix> - 设置群组中触发AI回复的前缀，例如：/setprefix @bot。
 /getprefix - 获取当前群组的触发前缀。
-  
+
 私聊指令 (在群组中也可以使用)：
 /settemp <温度值> - 设置AI回复的温度 (0-2)，例如：/settemp 1.0。
 /gettemp - 获取当前AI回复的温度。
 /resetuser - 重置你的个人设置。
 /promat <index> - 切换提示词，例如: /promat 0, 1, 2。
 /getpromat - 获取当前使用的提示词索引。
-  
+
 直接发送文本消息与AI对话 (私聊)。
 
 群组中，需要使用前缀触发AI回复，如果设置了前缀的话。
-  
+
 注意：
   - 机器人会记住最近的 {MAX_HISTORY_LENGTH} 条对话。
   - 机器人具有攻击性，请谨慎使用。
 """
 
+# ------------------ 更新 Telegram 命令 ------------------
 @app.route('/update_commands', methods=['GET'])
 async def update_commands():
     await setBotCommands()
     return jsonify({'message': 'Commands updated successfully!'})
 
+# ------------------ 处理 Webhook 请求 ------------------
 @app.route('/', methods=['POST'])
 async def handle_webhook():
     try:
@@ -443,10 +473,12 @@ async def handle_webhook():
         traceback.print_exc()
         return jsonify({'status': 'error', 'message': str(e)}), 400
 
+# ------------------ 健康检查 ------------------
 @app.route('/health', methods=['GET'])
 def health_check():
     return 'OK'
 
+# ------------------ 获取群组信息 ------------------
 async def getChatInfo(chatId):
     url = f"https://api.telegram.org/bot{TELEGRAM_BOT_TOKEN}/getChat"
     if PHP_PROXY_URL:
@@ -462,15 +494,101 @@ async def getChatInfo(chatId):
         print(f'获取群组信息失败: {e}')
         return None
 
+# ------------------ 封禁用户 ------------------
 async def banUser(chatId, userId):
     BANNED_USERS[userId] = datetime.now() + BAN_DURATION
     print(f"用户 {userId} 在群组 {chatId} 中被禁用，直到 {BANNED_USERS[userId]}")
 
+# ------------------ 解封用户 ------------------
 async def unbanUser(chatId, userId):
     if userId in BANNED_USERS:
         del BANNED_USERS[userId]
         await sendTelegramMessage(chatId, f"用户 {userId} 已被解禁。")
         print(f"用户 {userId} 在群组 {chatId} 中被解禁。")
 
+# ------------------ WebDAV 数据存储 ------------------
+def save_data_to_webdav():
+    global chatHistories, GROUP_SETTINGS, USER_SETTINGS, USER_LAST_ACTIVE, GROUP_ACTIVE_USERS, GROUP_INFO, BANNED_USERS
+    data = {
+        'chatHistories': chatHistories,
+        'GROUP_SETTINGS': GROUP_SETTINGS,
+        'USER_SETTINGS': USER_SETTINGS,
+        'USER_LAST_ACTIVE': USER_LAST_ACTIVE,
+        'GROUP_ACTIVE_USERS': GROUP_ACTIVE_USERS,
+        'GROUP_INFO': GROUP_INFO,
+        'BANNED_USERS': BANNED_USERS,
+    }
+    data_json = json.dumps(data, indent=4, default=str)
+    
+    filename = 'tg_bot/bot_data.json'
+    upload_url = f"{WEBDAV_URL}/{filename}"
+    
+    try:
+        curl_command = [
+            'curl',
+            '-v',  
+            '-X', 'PUT',
+            '--user', f'{WEBDAV_USERNAME}:{WEBDAV_PASSWORD}',
+            '--data', data_json,
+            '--header', 'Content-Type: application/json',
+            upload_url
+        ]
+        
+        result = subprocess.run(curl_command, capture_output=True, text=True, check=True)
+
+        print(f"WebDAV 上传成功：{upload_url}")
+        print(f"curl stdout: {result.stdout}")
+
+    except subprocess.CalledProcessError as e:
+        print(f"WebDAV 上传失败: {e}")
+        print(f"curl stderr: {e.stderr}")
+    except Exception as e:
+         print(f"WebDAV 上传发生错误: {e}")
+
+
+def load_data_from_webdav():
+    global chatHistories, GROUP_SETTINGS, USER_SETTINGS, USER_LAST_ACTIVE, GROUP_ACTIVE_USERS, GROUP_INFO, BANNED_USERS
+
+    filename = 'tg_bot/bot_data.json'
+    download_url = f"{WEBDAV_URL}/{filename}"
+
+    try:
+        curl_command = [
+            'curl',
+            '-v',
+            '--user', f'{WEBDAV_USERNAME}:{WEBDAV_PASSWORD}',
+            download_url
+        ]
+        
+        result = subprocess.run(curl_command, capture_output=True, text=True, check=True)
+        
+        if result.stdout:
+            data = json.loads(result.stdout)
+            chatHistories = data.get('chatHistories', {})
+            GROUP_SETTINGS = data.get('GROUP_SETTINGS', {})
+            USER_SETTINGS = data.get('USER_SETTINGS', {})
+            USER_LAST_ACTIVE = {int(k): datetime.fromisoformat(v) for k, v in data.get('USER_LAST_ACTIVE', {}).items()} if data.get('USER_LAST_ACTIVE') else {}
+            GROUP_ACTIVE_USERS = {int(k): set(v) for k, v in data.get('GROUP_ACTIVE_USERS', {}).items()} if data.get('GROUP_ACTIVE_USERS') else {}
+            GROUP_INFO = data.get('GROUP_INFO', {})
+            BANNED_USERS = {int(k): datetime.fromisoformat(v) for k, v in data.get('BANNED_USERS', {}).items()} if data.get('BANNED_USERS') else {}
+
+            print("WebDAV 数据加载成功。")
+    except subprocess.CalledProcessError as e:
+        print(f"WebDAV 数据加载失败(可能文件不存在): {e}")
+    except json.JSONDecodeError as e:
+        print(f"WebDAV 数据解析失败: {e}")
+    except Exception as e:
+         print(f"WebDAV 数据加载时发生错误: {e}")
+
+def webdav_upload_loop():
+    while True:
+        save_data_to_webdav()
+        time.sleep(60)
+
 if __name__ == '__main__':
+    load_data_from_webdav()
+
+    upload_thread = Thread(target=webdav_upload_loop, daemon=True)
+    upload_thread.start()
+    
     app.run(host='0.0.0.0', port=int(os.environ.get('PORT', 7860)))