Update app.py

app.py

@@ -526,10 +526,37 @@ async def unbanUser(chatId, userId):
 
 def webdav_upload(filename, data):
     url = urljoin(WEBDAV_URL, f"{WEBDAV_DIR}{filename}")
+    headers = {'Content-Type': 'application/json'}
+    
     try:
-        response = requests.put(url, auth=(WEBDAV_USERNAME, WEBDAV_PASSWORD), data=json.dumps(data), headers={'Content-Type': 'application/json'})
-        response.raise_for_status()
+      #尝试创建目录
+        dir_url = urljoin(WEBDAV_URL, f"{WEBDAV_DIR}")
+        response = requests.request('MKCOL', dir_url, auth=(WEBDAV_USERNAME, WEBDAV_PASSWORD))
+        if response.status_code == 201:
+            print(f"WebDAV 目录创建成功: {WEBDAV_DIR}")
+        elif response.status_code == 405:
+            print(f"WebDAV 目录已存在: {WEBDAV_DIR}")
+        elif response.status_code != 409:
+            response.raise_for_status()
+
+      # 检查文件是否存在
+        check_response = requests.request('PROPFIND', url, auth=(WEBDAV_USERNAME, WEBDAV_PASSWORD))
+
+        if check_response.status_code == 207:
+            print(f"WebDAV 文件已存在，使用PUT更新: {filename}")
+            response = requests.put(url, auth=(WEBDAV_USERNAME, WEBDAV_PASSWORD), data=json.dumps(data, default=str), headers=headers)
+            response.raise_for_status()
+        else:
+            print(f"WebDAV 文件不存在，使用COPY创建: {filename}")
+            copy_url = urljoin(WEBDAV_URL, f"{WEBDAV_DIR}{filename}.tmp")
+            response = requests.put(copy_url, auth=(WEBDAV_USERNAME, WEBDAV_PASSWORD), data=json.dumps(data, default=str), headers=headers)
+            response.raise_for_status()
+            response = requests.request('COPY', copy_url, auth=(WEBDAV_USERNAME, WEBDAV_PASSWORD), headers={'Destination': url})
+            response.raise_for_status()
+            response = requests.request('DELETE', copy_url, auth=(WEBDAV_USERNAME, WEBDAV_PASSWORD))
+            response.raise_for_status()
         print(f"WebDAV 上传成功: {filename}")
+
     except requests.exceptions.RequestException as e:
         print(f"WebDAV 上传失败: {filename}, {e}")