fix: #362

Author: vastsa

apps/base/migrations/migrations_004.py

@@ -0,0 +1,14 @@
+from tortoise import connections
+
+
+async def add_save_path_to_uploadchunk():
+    conn = connections.get("default")
+    await conn.execute_script(
+        """
+        ALTER TABLE uploadchunk ADD COLUMN save_path VARCHAR(512) NULL;
+        """
+    )
+
+
+async def migrate():
+    await add_save_path_to_uploadchunk()

apps/base/models.py

@@ -49,6 +49,7 @@ class UploadChunk(models.Model):
     file_size = fields.BigIntField()
     chunk_size = fields.IntField()
     file_name = fields.CharField(max_length=255)
+    save_path = fields.CharField(max_length=512, null=True)
     created_at = fields.DatetimeField(auto_now_add=True)
     completed = fields.BooleanField(default=False)
 
@@ -66,6 +67,7 @@ class KeyValue(Model):
 
 class PresignUploadSession(models.Model):
     """预签名上传会话模型"""
+
     id = fields.IntField(pk=True)
     upload_id = fields.CharField(max_length=36, unique=True, index=True)
     file_name = fields.CharField(max_length=255)
@@ -85,4 +87,6 @@ async def is_expired(self):
 file_codes_pydantic = pydantic_model_creator(FileCodes, name="FileCodes")
 upload_chunk_pydantic = pydantic_model_creator(UploadChunk, name="UploadChunk")
 key_value_pydantic = pydantic_model_creator(KeyValue, name="KeyValue")
-presign_upload_session_pydantic = pydantic_model_creator(PresignUploadSession, name="PresignUploadSession")
+presign_upload_session_pydantic = pydantic_model_creator(
+    PresignUploadSession, name="PresignUploadSession"
+)