Update app.py

app.py

@@ -7,17 +7,14 @@ 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')
-
 WEBDAV_URL = os.environ.get('WEBDAV_URL')
 WEBDAV_USERNAME = os.environ.get('WEBDAV_USERNAME')
 WEBDAV_PASSWORD = os.environ.get('WEBDAV_PASSWORD')
@@ -25,13 +22,11 @@ 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，只用文字格式。",
@@ -40,22 +35,12 @@ 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": "清空当前会话的聊天记录"},
@@ -70,8 +55,11 @@ 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|你|你们)",
@@ -96,8 +84,8 @@ BAN_TRIGGER_PHRASES = [
     r"(?:AI|bot|机器人).*(?:真|真是|简直|太).*(?:垃圾|废物|没用|蠢|笨|傻|弱)",
 ]
 UNBAN_PHRASE = "close username"
+BACKUP_INTERVAL = 60  # 1分钟
 
-# ------------------ Telegram API 请求 ------------------
 def make_telegram_request(method, data=None):
     url = f"https://api.telegram.org/bot{TELEGRAM_BOT_TOKEN}/{method}"
     if PHP_PROXY_URL:
@@ -116,7 +104,6 @@ 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"
@@ -139,7 +126,6 @@ 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'):
@@ -244,14 +230,12 @@ 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 '):
@@ -289,7 +273,6 @@ 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()
@@ -364,7 +347,6 @@ 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]
@@ -372,7 +354,6 @@ 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']
@@ -387,7 +368,6 @@ async def handleCallbackQuery(callbackQuery):
         },
       })
 
-# ------------------ 发送 Telegram 消息 ------------------
 async def sendTelegramMessage(chatId, text, options={}):
     url = f"https://api.telegram.org/bot{TELEGRAM_BOT_TOKEN}/sendMessage"
     if PHP_PROXY_URL:
@@ -407,7 +387,6 @@ 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:
@@ -424,7 +403,6 @@ async def editTelegramMessage(chatId, message_id, text, options={}):
     except requests.exceptions.RequestException as e:
         print(f'编辑 Telegram 消息失败: {e}')
 
-# ------------------ 获取帮助消息 ------------------
 def getHelpMessage():
     return f"""
 可用指令：
@@ -454,13 +432,11 @@ def getHelpMessage():
   - 机器人具有攻击性，请谨慎使用。
 """
 
-# ------------------ 更新 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:
@@ -473,12 +449,10 @@ 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:
@@ -494,101 +468,106 @@ 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
+def backup_data():
     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,
+        'USER_LAST_ACTIVE': USER_LAST_ACTIVE,
         '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}"
+    json_data = json.dumps(data)
     
+    timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
+    filename = f"tg_bot/backup_{timestamp}.json"
+
+    curl_command = [
+        'curl',
+        '-k',
+        '-u', f'{WEBDAV_USERNAME}:{WEBDAV_PASSWORD}',
+        '-X', 'PUT',
+        '-d', json_data,
+        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}")
+        process = subprocess.Popen(curl_command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        stdout, stderr = process.communicate()
+        if process.returncode == 0:
+            print(f"备份成功，已上传到 {WEBDAV_URL}/{filename}")
+        else:
+            print(f"备份失败，curl 返回码: {process.returncode} 错误信息: {stderr.decode()}")
     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}"
+        print(f"执行 curl 命令时发生错误: {e}")
 
+def load_data():
+    global chatHistories, GROUP_SETTINGS, USER_SETTINGS, GROUP_INFO, USER_LAST_ACTIVE, BANNED_USERS
+    
+    curl_command = [
+        'curl',
+        '-k',
+        '-u', f'{WEBDAV_USERNAME}:{WEBDAV_PASSWORD}',
+        f'{WEBDAV_URL}/tg_bot/'
+    ]
+  
     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}")
+        process = subprocess.Popen(curl_command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        stdout, stderr = process.communicate()
+        if process.returncode == 0:
+           output = stdout.decode()
+           json_files = re.findall(r'backup_(\d+)\.json', output)
+           if json_files:
+              latest_file = sorted(json_files, reverse=True)[0]
+              filename = f"tg_bot/backup_{latest_file}.json"
+              download_command = [
+                  'curl',
+                  '-k',
+                  '-u', f'{WEBDAV_USERNAME}:{WEBDAV_PASSWORD}',
+                  f'{WEBDAV_URL}/{filename}'
+              ]
+              process = subprocess.Popen(download_command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+              stdout, stderr = process.communicate()
+              if process.returncode == 0:
+                  try:
+                      data = json.loads(stdout.decode())
+                      chatHistories = data.get('chatHistories', {})
+                      GROUP_SETTINGS = data.get('GROUP_SETTINGS', {})
+                      USER_SETTINGS = data.get('USER_SETTINGS', {})
+                      GROUP_INFO = data.get('GROUP_INFO', {})
+                      USER_LAST_ACTIVE = data.get('USER_LAST_ACTIVE', {})
+                      BANNED_USERS = data.get('BANNED_USERS', {})
+                      print(f"从 {WEBDAV_URL}/{filename} 加载数据成功")
+                      return
+                  except json.JSONDecodeError:
+                      print(f"加载数据失败: {filename} JSON 解析错误")
+              else:
+                  print(f"下载文件失败，curl 返回码: {process.returncode} 错误信息: {stderr.decode()}")
+           else:
+              print("WebDAV 中未找到备份文件，初始化数据")
+        else:
+          print(f"列出文件失败，curl 返回码: {process.returncode} 错误信息: {stderr.decode()}")
     except Exception as e:
-         print(f"WebDAV 数据加载时发生错误: {e}")
+        print(f"执行 curl 命令时发生错误: {e}")
 
-def webdav_upload_loop():
+def schedule_backup():
     while True:
-        save_data_to_webdav()
-        time.sleep(60)
+        backup_data()
+        time.sleep(BACKUP_INTERVAL)
 
 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)))
+    load_data()
+    import threading
+    backup_thread = threading.Thread(target=schedule_backup, daemon=True)
+    backup_thread.start()
+    app.run(host='0.0.0.0', port=int(os.environ.get('PORT', 7860)))