add:fronted

apps/base/models.py

@@ -9,16 +9,18 @@ from tortoise import fields
 from tortoise.models import Model
 from tortoise.contrib.pydantic import pydantic_model_creator
 
+from core.utils import get_now
+
 
 class FileCodes(Model):
     id: Optional[int] = fields.IntField(pk=True)
     code: Optional[int] = fields.CharField(description='分享码', max_length=255, index=True, unique=True)
     prefix: Optional[str] = fields.CharField(max_length=255, description='前缀', default='')
     suffix: Optional[str] = fields.CharField(max_length=255, description='后缀', default='')
-    uuid_file_name: Optional[str] = fields.CharField(max_length=255, description='uuid文件名')
-    file_path: Optional[str] = fields.CharField(max_length=255, description='文件路径')
+    uuid_file_name: Optional[str] = fields.CharField(max_length=255, description='uuid文件名', null=True)
+    file_path: Optional[str] = fields.CharField(max_length=255, description='文件路径', null=True)
     size: Optional[int] = fields.IntField(description='文件大小', default=0)
-
+    text: Optional[str] = fields.TextField(description='文本内容', null=True)
     expired_at: Optional[datetime] = fields.DatetimeField(null=True, description='过期时间')
     expired_count: Optional[int] = fields.IntField(description='可用次数', default=0)
     used_count: Optional[int] = fields.IntField(description='已用次数', default=0)
@@ -27,7 +29,7 @@ class FileCodes(Model):
 
     async def is_expired(self):
         if self.expired_at:
-            return self.expired_at < datetime.now()
+            return self.expired_at < await get_now()
         else:
             return self.expired_count != -1 and self.used_count >= self.expired_count
 

apps/base/utils.py

@@ -30,7 +30,7 @@ async def get_file_path_name(file: UploadFile):
     file_uuid = f"{uuid.uuid4().hex}"
     uuid_file_name = f"{file_uuid}{suffix}"
     save_path = f"{path}/{uuid_file_name}"
-    return path, suffix, prefix, uuid_file_name, file_uuid, save_path
+    return path, suffix, prefix, uuid_file_name, save_path
 
 
 async def get_expire_info(expire_value: int, expire_style: str):

apps/base/views.py

@@ -0,0 +1,91 @@
+# @Time    : 2023/8/14 03:59
+# @Author  : Lan
+# @File    : views.py
+# @Software: PyCharm
+from fastapi import APIRouter, Form, UploadFile, File
+from pydantic import BaseModel
+
+from apps.base.models import FileCodes
+from apps.base.utils import get_expire_info, get_file_path_name
+from core.storage import file_storage
+
+share_api = APIRouter(
+    prefix='/share',
+    tags=['分享'],
+)
+
+
+@share_api.post('/text/')
+async def share_text(text: str = Form(...), expire_value: int = Form(default=1, gt=0), expire_style: str = Form(default='day')):
+    expired_at, expired_count, used_count, code = await get_expire_info(expire_value, expire_style)
+    await FileCodes.create(
+        code=code,
+        text=text,
+        expired_at=expired_at,
+        expired_count=expired_count,
+        used_count=used_count,
+        size=len(text),
+        prefix='文本分享'
+    )
+    return {
+        'code': 200,
+        'msg': 'success',
+        'data': {
+            'code': code,
+        }
+    }
+
+
+@share_api.post('/file/')
+async def share_file(expire_value: int = Form(default=1, gt=0), expire_style: str = Form(default='day'), file: UploadFile = File(...)):
+    expired_at, expired_count, used_count, code = await get_expire_info(expire_value, expire_style)
+    path, suffix, prefix, uuid_file_name, save_path = await get_file_path_name(file)
+    await file_storage.save_file(file, save_path)
+    await FileCodes.create(
+        code=code,
+        prefix=prefix,
+        suffix=suffix,
+        uuid_file_name=uuid_file_name,
+        file_path=path,
+        size=file.size,
+        expired_at=expired_at,
+        expired_count=expired_count,
+        used_count=used_count,
+    )
+    return {
+        'code': 200,
+        'msg': 'success',
+        'data': {
+            'code': code,
+            'name': file.filename,
+        }
+    }
+
+
+class SelectFileModel(BaseModel):
+    code: str
+
+
+@share_api.post('/select/')
+async def select_file(data: SelectFileModel):
+    file_code = await FileCodes.filter(code=data.code).first()
+    if not file_code:
+        return {
+            'code': 404,
+            'msg': '文件不存在',
+        }
+    if await file_code.is_expired():
+        return {
+            'code': 403,
+            'msg': '文件已过期',
+        }
+    return {
+        'code': 200,
+        'msg': 'success',
+        'data': {
+            'code': file_code.code,
+            'name': file_code.prefix + file_code.suffix,
+            'size': file_code.size,
+            'text': await file_storage.get_file_url(file_code),
+        }
+    }