feat:New database migration, new adaptation root path configuration

2025-02-16 21:38:44 +08:00
parent 4a1d77036f
commit d736a7b03a
7 changed files with 208 additions and 58 deletions
@@ -0,0 +1,52 @@
+from tortoise import connections
+
+
+async def create_file_codes_table():
+    conn = connections.get("default")
+    await conn.execute_script(
+        """
+        CREATE TABLE IF NOT EXISTS filecodes
+        (
+            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           INT          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
+        );
+        CREATE INDEX IF NOT EXISTS idx_filecodes_code_1c7ee7
+            on filecodes (code);
+    """
+    )
+
+
+async def create_key_value_table():
+    conn = connections.get("default")
+    await conn.execute_script(
+        """
+        CREATE TABLE IF NOT EXISTS keyvalue
+        (
+            id         INTEGER                             not null
+                primary key autoincrement,
+            key        VARCHAR(255)                        not null
+                unique,
+            value      JSON,
+            created_at TIMESTAMP default CURRENT_TIMESTAMP not null
+        );
+        CREATE INDEX IF NOT EXISTS idx_keyvalue_key_eab890
+            on keyvalue (key);
+    """
+    )
+
+
+async def migrate():
+    await create_file_codes_table()
+    await create_key_value_table()