Update app.py

app.py

@@ -5,6 +5,9 @@ from flask import Flask, request, jsonify
 from datetime import datetime, timedelta
 import asyncio
 import re
+import time
+from threading import Thread
+from webdav3.client import Client
 
 app = Flask(__name__)
 
@@ -13,8 +16,12 @@ 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')
+WEBDAV_DIRECTORY = '/tg_bot'
 
-if not all([TELEGRAM_BOT_TOKEN, AI_API_ENDPOINT, AI_API_KEY, AI_MODEL]):
+if not all([TELEGRAM_BOT_TOKEN, AI_API_ENDPOINT, AI_API_KEY, AI_MODEL, WEBDAV_URL, WEBDAV_USERNAME, WEBDAV_PASSWORD]):
     raise ValueError("请设置所有必要的环境变量")
 
 AI_API_HEADERS = {
@@ -80,6 +87,64 @@ BAN_TRIGGER_PHRASES = [
 ]
 UNBAN_PHRASE = "close username"
 
+webdav_options = {
+    'webdav_hostname': WEBDAV_URL,
+    'webdav_login': WEBDAV_USERNAME,
+    'webdav_password': WEBDAV_PASSWORD
+}
+webdav_client = Client(webdav_options)
+
+def save_data_to_webdav():
+    data = {
+        "chatHistories": chatHistories,
+        "GROUP_SETTINGS": GROUP_SETTINGS,
+        "USER_SETTINGS": USER_SETTINGS,
+        "USER_LAST_ACTIVE": {k: v.isoformat() for k, v in USER_LAST_ACTIVE.items()},
+        "GROUP_ACTIVE_USERS": {k: list(v) for k, v in GROUP_ACTIVE_USERS.items()},
+        "GROUP_INFO": {k: {**v, 'last_active': v['last_active'].isoformat()} for k, v in GROUP_INFO.items()},
+        "BANNED_USERS": {k: v.isoformat() for k, v in BANNED_USERS.items()}
+    }
+    try:
+        file_name = f"{WEBDAV_DIRECTORY}/data.json"
+        with open("data.json", "w") as f:
+            json.dump(data, f)
+        
+        if not webdav_client.check(WEBDAV_DIRECTORY):
+            webdav_client.mkdir(WEBDAV_DIRECTORY)
+
+        webdav_client.upload_sync(remote_path=file_name, local_path="data.json")
+        print(f"数据已上传至 WebDAV: {file_name}")
+    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
+    try:
+        file_name = f"{WEBDAV_DIRECTORY}/data.json"
+        if webdav_client.check(file_name):
+            webdav_client.download_sync(remote_path=file_name, local_path="data.json")
+            with open("data.json", "r") as f:
+                data = json.load(f)
+            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()}
+            GROUP_ACTIVE_USERS = {int(k): set(v) for k, v in data.get("GROUP_ACTIVE_USERS", {}).items()}
+            GROUP_INFO = {int(k): {**v, 'last_active': datetime.fromisoformat(v['last_active'])} for k, v in data.get("GROUP_INFO", {}).items()}
+            BANNED_USERS = {int(k): datetime.fromisoformat(v) for k, v in data.get("BANNED_USERS", {}).items()}
+
+            print(f"从 WebDAV 加载数据成功: {file_name}")
+        else:
+            print("WebDAV 上没有找到数据文件，初始化为空数据。")
+            save_data_to_webdav()
+    except Exception as e:
+        print(f"从 WebDAV 加载数据失败: {e}")
+
+def webdav_sync_thread():
+    while True:
+        save_data_to_webdav()
+        time.sleep(60)
+
 def make_telegram_request(method, data=None):
     url = f"https://api.telegram.org/bot{TELEGRAM_BOT_TOKEN}/{method}"
     if PHP_PROXY_URL:
@@ -134,7 +199,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:
@@ -156,7 +221,6 @@ async def handleTelegramUpdate(update):
         else:
             GROUP_INFO[chatId]['last_active'] = datetime.now()
 
-
     if not userMessage:
         return
 
@@ -273,7 +337,7 @@ async def processAiMessage(chatId, userMessage, fromUserId, message_id, fromUser
         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 +348,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)
@@ -473,4 +537,6 @@ async def unbanUser(chatId, userId):
         print(f"用户 {userId} 在群组 {chatId} 中被解禁。")
 
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=int(os.environ.get('PORT', 7860)))
+    load_data_from_webdav()
+    Thread(target=webdav_sync_thread, daemon=True).start()
+    app.run(host='0.0.0.0', port=int(os.environ.get('PORT', 7860)))