feat: 新增webdav存储

Author: Lan

apps/base/utils.py

@@ -1,8 +1,9 @@
 import datetime
-import uuid
 import os
+import uuid
+
 from fastapi import UploadFile, HTTPException
-from typing import Tuple, Optional
+from typing import Optional, Tuple
 
 from apps.base.dependencies import IPRateLimit
 from apps.base.models import FileCodes

apps/base/views.py

@@ -135,4 +135,8 @@ async def download_file(key: str, code: str, ip: str = Depends(ip_limit['error']
     if not has:
         return APIResponse(code=404, detail='文件不存在')
 
-    return APIResponse(detail=file_code.text) if file_code.text else await file_storage.get_file_response(file_code)
+    return (
+        APIResponse(detail=file_code.text)
+        if file_code.text
+        else await file_storage.get_file_response(file_code)
+    )

core/settings.py

@@ -40,6 +40,9 @@
     'uploadSize': 1024 * 1024 * 10,
     'expireStyle': ['day', 'hour', 'minute', 'forever', 'count'],
     'uploadMinute': 1,
+    'webdav_url': '',
+    'webdav_password': '',
+    'webdav_username': '',
     'opacity': 0.9,
     'background': '',
     'uploadCount': 10,

core/storage.py

@@ -16,7 +16,6 @@
 
 async def delete_expire_files():
     file_storage: FileStorageInterface = storages[settings.file_storage]()
-    print(settings.file_storage)
     while True:
         try:
             # 遍历 share目录下的所有文件夹，删除空的文件夹，并判断父目录是否为空，如果为空也删除

core/tasks.py

@@ -7,9 +7,7 @@
 import random
 import string
 import time
-
-from apps.base.dependencies import IPRateLimit
-
+from core.settings import settings
 
 async def get_random_num():
     """
@@ -46,7 +44,7 @@ async def get_select_token(code: str):
     :param code:
     :return:
     """
-    token = "123456"
+    token = settings.admin_token
     return hashlib.sha256(f"{code}{int(time.time() / 1000)}000{token}".encode()).hexdigest()