feat: add migration for filecodes table and update size field to BigInt

Author: vastsa

apps/base/migrations/migrations_005.py

@@ -0,0 +1,72 @@
+from tortoise import connections
+
+
+def _need_upgrade(columns: list[tuple]) -> bool:
+    for column in columns:
+        # PRAGMA table_info 返回 (cid, name, type, notnull, dflt_value, pk)
+        if column[1] == "size":
+            column_type = (column[2] or "").upper()
+            return "BIGINT" not in column_type
+    return False
+
+
+async def migrate():
+    conn = connections.get("default")
+    result = await conn.execute_query("PRAGMA table_info(filecodes)")
+    columns = result[1] if result and len(result) > 1 else []
+
+    if not columns or not _need_upgrade(columns):
+        return
+
+    await conn.execute_script(
+        """
+        BEGIN;
+        CREATE TABLE IF NOT EXISTS filecodes_new
+        (
+            id             INTEGER                                not null
+                primary key autoincrement,
+            code           VARCHAR(255)                           not null
+                unique,
+            prefix         VARCHAR(255) default ''                not null,
+            suffix         VARCHAR(255) default ''                not null,
+            uuid_file_name VARCHAR(255),
+            file_path      VARCHAR(255),
+            size           BIGINT       default 0                 not null,
+            text           TEXT,
+            expired_at     TIMESTAMP,
+            expired_count  INT          default 0                 not null,
+            used_count     INT          default 0                 not null,
+            created_at     TIMESTAMP    default CURRENT_TIMESTAMP not null,
+            file_hash      VARCHAR(128),
+            is_chunked     BOOL         default False             not null,
+            upload_id      VARCHAR(128)
+        );
+
+        INSERT INTO filecodes_new (id, code, prefix, suffix, uuid_file_name, file_path, size, text,
+                                   expired_at, expired_count, used_count, created_at, file_hash,
+                                   is_chunked, upload_id)
+        SELECT id,
+               code,
+               prefix,
+               suffix,
+               uuid_file_name,
+               file_path,
+               size,
+               text,
+               expired_at,
+               expired_count,
+               used_count,
+               created_at,
+               file_hash,
+               is_chunked,
+               upload_id
+        FROM filecodes;
+
+        DROP TABLE filecodes;
+        ALTER TABLE filecodes_new
+            RENAME TO filecodes;
+        CREATE INDEX IF NOT EXISTS idx_filecodes_code_1c7ee7
+            on filecodes (code);
+        COMMIT;
+        """
+    )

apps/base/models.py

@@ -19,7 +19,7 @@ class FileCodes(models.Model):
     suffix = fields.CharField(max_length=255, default="")
     uuid_file_name = fields.CharField(max_length=255, null=True)
     file_path = fields.CharField(max_length=255, null=True)
-    size = fields.IntField(default=0)
+    size = fields.BigIntField(default=0)
     text = fields.TextField(null=True)
     expired_at = fields.DatetimeField(null=True)
     expired_count = fields.IntField(default=0)

core/settings.py

@@ -65,7 +65,7 @@
     ],
     "themesSelect": "themes/2024",
     "errorMinute": 1,
-    "errorCount": 1,
+    "errorCount": 10,
     "serverWorkers": 1,
     "serverHost": "0.0.0.0",
     "serverPort": 12345,