feat: 导入 MemRelay 初始源码

2026-09-23 21:40:46 +08:00
commit 0ade3bb318
167 changed files with 53511 additions and 0 deletions
@@ -0,0 +1,42 @@
+"""Add upload task metadata."""
+
+from collections.abc import Sequence
+
+import sqlalchemy as sa
+from alembic import op
+
+revision: str = "20260802_0002"
+down_revision: str | Sequence[str] | None = "20260802_0001"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade() -> None:
+    with op.batch_alter_table("upload_tasks") as batch:
+        batch.add_column(sa.Column("project_id", sa.String(36), nullable=True))
+        batch.add_column(sa.Column("description", sa.Text(), nullable=True))
+        batch.add_column(sa.Column("version", sa.String(100), nullable=True))
+        batch.add_column(sa.Column("purpose", sa.String(200), nullable=True))
+        batch.add_column(sa.Column("tags_json", sa.Text(), nullable=False, server_default="[]"))
+        batch.add_column(
+            sa.Column("overwrite_allowed", sa.Boolean(), nullable=False, server_default=sa.false())
+        )
+        batch.create_foreign_key(
+            "fk_upload_tasks_project_id", "projects", ["project_id"], ["id"], ondelete="SET NULL"
+        )
+        batch.create_index("ix_upload_tasks_project_id", ["project_id"])
+
+
+def downgrade() -> None:
+    with op.batch_alter_table("upload_tasks") as batch:
+        batch.drop_index("ix_upload_tasks_project_id")
+        batch.drop_constraint("fk_upload_tasks_project_id", type_="foreignkey")
+        for column in (
+            "overwrite_allowed",
+            "tags_json",
+            "purpose",
+            "version",
+            "description",
+            "project_id",
+        ):
+            batch.drop_column(column)