Update app.py

app.py

@@ -5,6 +5,8 @@ from flask import Flask, request, jsonify
 from datetime import datetime, timedelta
 import asyncio
 import re
+import time
+from urllib.parse import urljoin
 
 app = Flask(__name__)
 
@@ -13,8 +15,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_DIR = "/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 +86,7 @@ BAN_TRIGGER_PHRASES = [
 ]
 UNBAN_PHRASE = "close username"
 
+
 def make_telegram_request(method, data=None):
     url = f"https://api.telegram.org/bot{TELEGRAM_BOT_TOKEN}/{method}"
     if PHP_PROXY_URL:
@@ -472,5 +479,74 @@ async def unbanUser(chatId, userId):
         await sendTelegramMessage(chatId, f"用户 {userId} 已被解禁。")
         print(f"用户 {userId} 在群组 {chatId} 中被解禁。")
 
+def webdav_upload(filename, data):
+    url = urljoin(WEBDAV_URL, f"{WEBDAV_DIR}{filename}")
+    try:
+        response = requests.put(url, auth=(WEBDAV_USERNAME, WEBDAV_PASSWORD), data=json.dumps(data), headers={'Content-Type': 'application/json'})
+        response.raise_for_status()
+        print(f"WebDAV 上传成功: {filename}")
+    except requests.exceptions.RequestException as e:
+        print(f"WebDAV 上传失败: {filename}, {e}")
+
+def webdav_download(filename):
+    url = urljoin(WEBDAV_URL, f"{WEBDAV_DIR}{filename}")
+    try:
+        response = requests.get(url, auth=(WEBDAV_USERNAME, WEBDAV_PASSWORD))
+        response.raise_for_status()
+        print(f"WebDAV 下载成功: {filename}")
+        return response.json()
+    except requests.exceptions.RequestException as e:
+        print(f"WebDAV 下载失败: {filename}, {e}")
+        return None
+
+def sync_data_to_webdav():
+    while True:
+        try:
+            webdav_upload('chatHistories.json', chatHistories)
+            webdav_upload('GROUP_SETTINGS.json', GROUP_SETTINGS)
+            webdav_upload('USER_SETTINGS.json', USER_SETTINGS)
+            webdav_upload('USER_LAST_ACTIVE.json', USER_LAST_ACTIVE)
+            webdav_upload('GROUP_ACTIVE_USERS.json', GROUP_ACTIVE_USERS)
+            webdav_upload('GROUP_INFO.json', GROUP_INFO)
+            webdav_upload('BANNED_USERS.json', BANNED_USERS)
+        except Exception as e:
+            print(f"WebDAV 同步失败: {e}")
+        time.sleep(60)
+
+def load_data_from_webdav():
+    global chatHistories, GROUP_SETTINGS, USER_SETTINGS, USER_LAST_ACTIVE, GROUP_ACTIVE_USERS, GROUP_INFO, BANNED_USERS
+    
+    loaded_chatHistories = webdav_download('chatHistories.json')
+    if loaded_chatHistories:
+        chatHistories = loaded_chatHistories
+    
+    loaded_GROUP_SETTINGS = webdav_download('GROUP_SETTINGS.json')
+    if loaded_GROUP_SETTINGS:
+        GROUP_SETTINGS = loaded_GROUP_SETTINGS
+    
+    loaded_USER_SETTINGS = webdav_download('USER_SETTINGS.json')
+    if loaded_USER_SETTINGS:
+        USER_SETTINGS = loaded_USER_SETTINGS
+    
+    loaded_USER_LAST_ACTIVE = webdav_download('USER_LAST_ACTIVE.json')
+    if loaded_USER_LAST_ACTIVE:
+       USER_LAST_ACTIVE = {int(k): datetime.fromisoformat(v) for k, v in loaded_USER_LAST_ACTIVE.items()}
+    
+    loaded_GROUP_ACTIVE_USERS = webdav_download('GROUP_ACTIVE_USERS.json')
+    if loaded_GROUP_ACTIVE_USERS:
+      GROUP_ACTIVE_USERS = {int(k): set(v) for k, v in loaded_GROUP_ACTIVE_USERS.items()}
+    
+    loaded_GROUP_INFO = webdav_download('GROUP_INFO.json')
+    if loaded_GROUP_INFO:
+        GROUP_INFO = {int(k): {**v, 'last_active': datetime.fromisoformat(v['last_active'])} for k, v in loaded_GROUP_INFO.items()}
+
+    loaded_BANNED_USERS = webdav_download('BANNED_USERS.json')
+    if loaded_BANNED_USERS:
+        BANNED_USERS = {int(k): datetime.fromisoformat(v) for k, v in loaded_BANNED_USERS.items()}
+
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=int(os.environ.get('PORT', 7860)))
+    load_data_from_webdav()
+    import threading
+    sync_thread = threading.Thread(target=sync_data_to_webdav, daemon=True)
+    sync_thread.start()
+    app.run(host='0.0.0.0', port=int(os.environ.get('PORT', 7860)))