refact: some codes

apps/admin/services.py

@@ -0,0 +1,123 @@
+import os
+import time
+
+from core.response import APIResponse
+from core.storage import FileStorageInterface, storages
+from core.settings import settings
+from apps.base.models import FileCodes, KeyValue
+from apps.base.utils import get_expire_info, get_file_path_name
+from fastapi import HTTPException
+from core.settings import data_root
+
+
+class FileService:
+    def __init__(self):
+        self.file_storage: FileStorageInterface = storages[settings.file_storage]()
+
+    async def delete_file(self, file_id: int):
+        file_code = await FileCodes.get(id=file_id)
+        await self.file_storage.delete_file(file_code)
+        await file_code.delete()
+
+    async def list_files(self, page: int, size: int):
+        offset = (page - 1) * size
+        files = await FileCodes.all().limit(size).offset(offset)
+        total = await FileCodes.all().count()
+        return files, total
+
+    async def download_file(self, file_id: int):
+        file_code = await FileCodes.filter(id=file_id).first()
+        if not file_code:
+            raise HTTPException(status_code=404, detail='文件不存在')
+        if file_code.text:
+            return APIResponse(detail=file_code.text)
+        else:
+            return await self.file_storage.get_file_response(file_code)
+
+    async def share_local_file(self, item):
+        local_file = LocalFileClass(item.filename)
+        if not await local_file.exists():
+            raise HTTPException(status_code=404, detail='文件不存在')
+
+        text = await local_file.read()
+        expired_at, expired_count, used_count, code = await get_expire_info(item.expire_value, item.expire_style)
+        path, suffix, prefix, uuid_file_name, save_path = await get_file_path_name(item)
+
+        await self.file_storage.save_file(text, save_path)
+
+        await FileCodes.create(
+            code=code,
+            prefix=prefix,
+            suffix=suffix,
+            uuid_file_name=uuid_file_name,
+            file_path=path,
+            size=local_file.size,
+            expired_at=expired_at,
+            expired_count=expired_count,
+            used_count=used_count,
+        )
+
+        return {
+            'code': code,
+            'name': local_file.file,
+        }
+
+
+class ConfigService:
+    def get_config(self):
+        return settings.items()
+
+    async def update_config(self, data: dict):
+        admin_token = data.get('admin_token')
+        if admin_token is None or admin_token == '':
+            raise HTTPException(status_code=400, detail='管理员密码不能为空')
+
+        for key, value in data.items():
+            if key not in settings.default_config:
+                continue
+            if key in ['errorCount', 'errorMinute', 'max_save_seconds', 'onedrive_proxy', 'openUpload', 'port', 's3_proxy', 'uploadCount', 'uploadMinute', 'uploadSize']:
+                data[key] = int(value)
+            elif key in ['opacity']:
+                data[key] = float(value)
+            else:
+                data[key] = value
+
+        await KeyValue.filter(key='settings').update(value=data)
+        for k, v in data.items():
+            settings.__setattr__(k, v)
+
+
+class LocalFileService:
+    async def list_files(self):
+        files = []
+        for file in os.listdir(data_root / 'local'):
+            files.append(LocalFileClass(file))
+        return files
+
+    async def delete_file(self, filename: str):
+        file = LocalFileClass(filename)
+        if await file.exists():
+            await file.delete()
+            return '删除成功'
+        raise HTTPException(status_code=404, detail='文件不存在')
+
+
+class LocalFileClass:
+    def __init__(self, file):
+        self.file = file
+        self.path = data_root / 'local' / file
+        self.ctime = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(os.path.getctime(self.path)))
+        self.size = os.path.getsize(self.path)
+
+    async def read(self):
+        return open(self.path, 'rb')
+
+    async def write(self, data):
+        with open(self.path, 'w') as f:
+            f.write(data)
+
+    async def delete(self):
+        os.remove(self.path)
+
+    async def exists(self):
+        return os.path.exists(self.path)