Create upload.py

upload.py

@@ -0,0 +1,105 @@
+import time
+import requests
+import os
+
+# 定义登录和上传的URL
+LOGIN_URL = "https://surbao-sur.onrender.com/api/auth/login"
+UPLOAD_URL = "https://surbao-sur.onrender.com/api/fs/form"
+RENAME_URL = "https://surbao-sur.onrender.com/api/fs/rename"
+LIST_URL = "https://surbao-sur.onrender.com/api/fs/list"
+
+def get_token(username, password):
+    """通过登录获取 token"""
+    try:
+        response = requests.post(
+            LOGIN_URL,
+            json={"username": username, "password": password},
+            headers={"Content-Type": "application/json"}
+        )
+        if response.status_code != 200:
+            print(f"请求失败: {response.status_code}")
+            print("响应内容:", response.text)
+            return None
+
+        data = response.json()
+        if data.get("code") == 200 and "data" in data and "token" in data["data"]:
+            print("Token retrieved successfully!")
+            return data["data"]["token"]
+        else:
+            print("无法获取 token:", data)
+            return None
+    except Exception as e:
+        print(f"发生异常: {e}")
+        return None
+
+def getlist(token):
+    """列出文件列表"""
+    try:
+        headers = {"Authorization": token}
+        response = requests.post(
+            LIST_URL,
+            headers=headers,
+            json={"path": "/filen/openwebui", "refresh": True}
+        )
+        print("响应内容:", response.text)
+    except Exception as e:
+        print(f"发生异常: {e}")
+
+def rename_file(token, file_name):
+    """重命名文件"""
+    try:
+        headers = {"Authorization": token}
+        response = requests.post(
+            RENAME_URL,
+            headers=headers,
+            json={"name": "webui_old.db", "path": f"/filen/openwebui/{file_name}"}
+        )
+        if response.status_code == 200:
+            print("文件重命名成功!")
+            print("响应内容:", response.text)
+        else:
+            print(f"文件重命名失败: {response.status_code}")
+            print("响应内容:", response.text)
+    except Exception as e:
+        print(f"文件重命名时发生异常: {e}")
+
+def upload_file(token, file_path, file_name):
+    """上传文件"""
+    try:
+        headers = {
+            "Authorization": token,
+            "File-Path": f"/filen/openwebui/{file_name}",
+            "As-Task": "true"
+        }
+        with open(file_path, "rb") as file:
+            files = {"file": file}
+            response = requests.put(UPLOAD_URL, headers=headers, files=files)
+
+        if response.status_code == 200:
+            print("文件上传成功!")
+            print("响应内容:", response.text)
+        else:
+            print(f"文件上传失败: {response.status_code}")
+            print("响应内容:", response.text)
+    except Exception as e:
+        print(f"文件上传时发生异常: {e}")
+
+def run_script():
+    """主脚本逻辑"""
+    username = os.environ.get("USERNAME")
+    password = os.environ.get("PASSWORD")
+    file_path = "/app/backend/data/webui.db"
+    file_name = "webui.db"
+
+    token = get_token(username, password)
+    if token:
+        getlist(token)
+        rename_file(token, file_name)
+        upload_file(token, file_path, file_name)
+
+if __name__ == "__main__":
+    while True:
+        print("运行脚本中...")
+        run_script()
+        print("脚本执行完成，等待12小时...")
+        time.sleep(3600)  # 等待 12 小时（12 小时 * 60 分钟 * 60 秒）