You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
285 lines
14 KiB
285 lines
14 KiB
"""initial schema
|
|
|
|
Revision ID: 20260413_0001
|
|
Revises:
|
|
Create Date: 2026-04-13 00:00:00.000000
|
|
"""
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision = "20260413_0001"
|
|
down_revision = None
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
op.create_table(
|
|
"users",
|
|
sa.Column("id", sa.String(), nullable=False),
|
|
sa.Column("username", sa.String(), nullable=False),
|
|
sa.Column("password_hash", sa.String(), nullable=False),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
sa.UniqueConstraint("username"),
|
|
)
|
|
|
|
op.create_table(
|
|
"directories",
|
|
sa.Column("id", sa.String(), nullable=False),
|
|
sa.Column("parent_id", sa.String(), nullable=True),
|
|
sa.Column("name", sa.String(), nullable=False),
|
|
sa.Column("path_key", sa.String(), nullable=False),
|
|
sa.Column("deleted_at", sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.ForeignKeyConstraint(["parent_id"], ["directories.id"]),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
sa.UniqueConstraint("path_key"),
|
|
)
|
|
op.create_index("idx_directories_parent_id", "directories", ["parent_id"])
|
|
op.create_index("idx_directories_path_key", "directories", ["path_key"])
|
|
|
|
op.create_table(
|
|
"backends",
|
|
sa.Column("id", sa.String(), nullable=False),
|
|
sa.Column("name", sa.String(), nullable=False),
|
|
sa.Column("type", sa.String(), nullable=False),
|
|
sa.Column("stability_class", sa.String(), nullable=False),
|
|
sa.Column("read_priority", sa.Integer(), nullable=False),
|
|
sa.Column("write_priority", sa.Integer(), nullable=False),
|
|
sa.Column("is_enabled", sa.Boolean(), server_default="1", nullable=False),
|
|
sa.Column("config_json", sa.Text(), nullable=False),
|
|
sa.Column("capacity_hint_bytes", sa.Integer(), nullable=True),
|
|
sa.Column("last_health_status", sa.String(), nullable=True),
|
|
sa.Column("last_health_checked_at", sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
sa.UniqueConstraint("name"),
|
|
)
|
|
op.create_index("idx_backends_enabled", "backends", ["is_enabled"])
|
|
op.create_index("idx_backends_priority", "backends", ["write_priority", "read_priority"])
|
|
|
|
op.create_table(
|
|
"upload_sessions",
|
|
sa.Column("id", sa.String(), nullable=False),
|
|
sa.Column("directory_id", sa.String(), nullable=True),
|
|
sa.Column("filename", sa.String(), nullable=True),
|
|
sa.Column("total_size_bytes", sa.Integer(), nullable=False),
|
|
sa.Column("received_size_bytes", sa.Integer(), nullable=False),
|
|
sa.Column("status", sa.String(), nullable=False),
|
|
sa.Column("temp_path", sa.String(), nullable=False),
|
|
sa.Column("tus_upload_url", sa.String(), nullable=True),
|
|
sa.Column("upload_metadata_json", sa.Text(), nullable=True),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.ForeignKeyConstraint(["directory_id"], ["directories.id"]),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
op.create_index("idx_upload_sessions_status", "upload_sessions", ["status"])
|
|
op.create_index("idx_upload_sessions_updated_at", "upload_sessions", ["updated_at"])
|
|
|
|
op.create_table(
|
|
"file_entries",
|
|
sa.Column("id", sa.String(), nullable=False),
|
|
sa.Column("directory_id", sa.String(), nullable=False),
|
|
sa.Column("name", sa.String(), nullable=False),
|
|
sa.Column("mime_type", sa.String(), nullable=True),
|
|
sa.Column("size_bytes", sa.Integer(), nullable=False),
|
|
sa.Column("current_version_id", sa.String(), nullable=True),
|
|
sa.Column("is_deleted", sa.Boolean(), server_default="0", nullable=False),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.ForeignKeyConstraint(["directory_id"], ["directories.id"]),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
op.create_index("idx_file_entries_current_version_id", "file_entries", ["current_version_id"])
|
|
op.create_index(
|
|
"idx_file_entries_directory_name",
|
|
"file_entries",
|
|
["directory_id", "name", "is_deleted"],
|
|
unique=True,
|
|
)
|
|
|
|
op.create_table(
|
|
"file_versions",
|
|
sa.Column("id", sa.String(), nullable=False),
|
|
sa.Column("file_entry_id", sa.String(), nullable=False),
|
|
sa.Column("content_hash", sa.String(), nullable=False),
|
|
sa.Column("size_bytes", sa.Integer(), nullable=False),
|
|
sa.Column("storage_layout", sa.String(), nullable=False),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.ForeignKeyConstraint(["file_entry_id"], ["file_entries.id"]),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
op.create_index("idx_file_versions_content_hash", "file_versions", ["content_hash"])
|
|
op.create_index("idx_file_versions_file_entry_id", "file_versions", ["file_entry_id"])
|
|
|
|
op.create_table(
|
|
"blobs",
|
|
sa.Column("id", sa.String(), nullable=False),
|
|
sa.Column("file_version_id", sa.String(), nullable=False),
|
|
sa.Column("blob_index", sa.Integer(), nullable=False),
|
|
sa.Column("kind", sa.String(), nullable=False),
|
|
sa.Column("content_hash", sa.String(), nullable=False),
|
|
sa.Column("size_bytes", sa.Integer(), nullable=False),
|
|
sa.Column("logical_offset", sa.Integer(), server_default="0", nullable=False),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.ForeignKeyConstraint(["file_version_id"], ["file_versions.id"]),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
op.create_index("idx_blobs_content_hash", "blobs", ["content_hash"])
|
|
op.create_index("idx_blobs_file_version_index", "blobs", ["file_version_id", "blob_index"], unique=True)
|
|
|
|
op.create_table(
|
|
"blob_replicas",
|
|
sa.Column("id", sa.String(), nullable=False),
|
|
sa.Column("blob_id", sa.String(), nullable=False),
|
|
sa.Column("backend_id", sa.String(), nullable=False),
|
|
sa.Column("storage_key", sa.String(), nullable=False),
|
|
sa.Column("status", sa.String(), nullable=False),
|
|
sa.Column("etag", sa.String(), nullable=True),
|
|
sa.Column("checksum", sa.String(), nullable=True),
|
|
sa.Column("size_bytes", sa.Integer(), nullable=False),
|
|
sa.Column("last_verified_at", sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.ForeignKeyConstraint(["backend_id"], ["backends.id"]),
|
|
sa.ForeignKeyConstraint(["blob_id"], ["blobs.id"]),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
op.create_index("idx_blob_replicas_backend_status", "blob_replicas", ["backend_id", "status"])
|
|
op.create_index("idx_blob_replicas_blob_backend", "blob_replicas", ["blob_id", "backend_id"], unique=True)
|
|
op.create_index("idx_blob_replicas_storage_key", "blob_replicas", ["storage_key"])
|
|
|
|
op.create_table(
|
|
"upload_session_parts",
|
|
sa.Column("id", sa.String(), nullable=False),
|
|
sa.Column("upload_session_id", sa.String(), nullable=False),
|
|
sa.Column("part_number", sa.Integer(), nullable=False),
|
|
sa.Column("byte_offset", sa.Integer(), nullable=False),
|
|
sa.Column("size_bytes", sa.Integer(), nullable=False),
|
|
sa.Column("checksum", sa.String(), nullable=True),
|
|
sa.Column("status", sa.String(), nullable=False),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.ForeignKeyConstraint(["upload_session_id"], ["upload_sessions.id"]),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
op.create_index(
|
|
"idx_upload_session_parts_unique",
|
|
"upload_session_parts",
|
|
["upload_session_id", "part_number"],
|
|
unique=True,
|
|
)
|
|
op.create_index("idx_upload_session_parts_upload_id", "upload_session_parts", ["upload_session_id"])
|
|
|
|
op.create_table(
|
|
"jobs",
|
|
sa.Column("id", sa.String(), nullable=False),
|
|
sa.Column("kind", sa.String(), nullable=False),
|
|
sa.Column("status", sa.String(), nullable=False),
|
|
sa.Column("payload_json", sa.Text(), nullable=False),
|
|
sa.Column("attempt_count", sa.Integer(), server_default="0", nullable=False),
|
|
sa.Column("max_attempts", sa.Integer(), server_default="5", nullable=False),
|
|
sa.Column("run_after", sa.DateTime(timezone=True), nullable=False),
|
|
sa.Column("last_error", sa.Text(), nullable=True),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
op.create_index("idx_jobs_kind_status", "jobs", ["kind", "status"])
|
|
op.create_index("idx_jobs_status_run_after", "jobs", ["status", "run_after"])
|
|
|
|
op.create_table(
|
|
"preview_artifacts",
|
|
sa.Column("id", sa.String(), nullable=False),
|
|
sa.Column("file_version_id", sa.String(), nullable=False),
|
|
sa.Column("artifact_type", sa.String(), nullable=False),
|
|
sa.Column("blob_id", sa.String(), nullable=False),
|
|
sa.Column("status", sa.String(), nullable=False),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.ForeignKeyConstraint(["blob_id"], ["blobs.id"]),
|
|
sa.ForeignKeyConstraint(["file_version_id"], ["file_versions.id"]),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
op.create_index(
|
|
"idx_preview_artifacts_version_type",
|
|
"preview_artifacts",
|
|
["file_version_id", "artifact_type"],
|
|
unique=True,
|
|
)
|
|
|
|
op.create_table(
|
|
"cache_entries",
|
|
sa.Column("id", sa.String(), nullable=False),
|
|
sa.Column("blob_id", sa.String(), nullable=False),
|
|
sa.Column("local_path", sa.String(), nullable=False),
|
|
sa.Column("cache_kind", sa.String(), nullable=False),
|
|
sa.Column("size_bytes", sa.Integer(), nullable=False),
|
|
sa.Column("status", sa.String(), nullable=False),
|
|
sa.Column("last_accessed_at", sa.DateTime(timezone=True), nullable=False),
|
|
sa.Column("expires_at", sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.ForeignKeyConstraint(["blob_id"], ["blobs.id"]),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
op.create_index("idx_cache_entries_blob_id", "cache_entries", ["blob_id"])
|
|
op.create_index("idx_cache_entries_expires_at", "cache_entries", ["expires_at"])
|
|
op.create_index("idx_cache_entries_last_accessed_at", "cache_entries", ["last_accessed_at"])
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_index("idx_cache_entries_last_accessed_at", table_name="cache_entries")
|
|
op.drop_index("idx_cache_entries_expires_at", table_name="cache_entries")
|
|
op.drop_index("idx_cache_entries_blob_id", table_name="cache_entries")
|
|
op.drop_table("cache_entries")
|
|
|
|
op.drop_index("idx_preview_artifacts_version_type", table_name="preview_artifacts")
|
|
op.drop_table("preview_artifacts")
|
|
|
|
op.drop_index("idx_jobs_status_run_after", table_name="jobs")
|
|
op.drop_index("idx_jobs_kind_status", table_name="jobs")
|
|
op.drop_table("jobs")
|
|
|
|
op.drop_index("idx_upload_session_parts_upload_id", table_name="upload_session_parts")
|
|
op.drop_index("idx_upload_session_parts_unique", table_name="upload_session_parts")
|
|
op.drop_table("upload_session_parts")
|
|
|
|
op.drop_index("idx_blob_replicas_storage_key", table_name="blob_replicas")
|
|
op.drop_index("idx_blob_replicas_blob_backend", table_name="blob_replicas")
|
|
op.drop_index("idx_blob_replicas_backend_status", table_name="blob_replicas")
|
|
op.drop_table("blob_replicas")
|
|
|
|
op.drop_index("idx_blobs_file_version_index", table_name="blobs")
|
|
op.drop_index("idx_blobs_content_hash", table_name="blobs")
|
|
op.drop_table("blobs")
|
|
|
|
op.drop_index("idx_file_versions_file_entry_id", table_name="file_versions")
|
|
op.drop_index("idx_file_versions_content_hash", table_name="file_versions")
|
|
op.drop_table("file_versions")
|
|
|
|
op.drop_index("idx_file_entries_directory_name", table_name="file_entries")
|
|
op.drop_index("idx_file_entries_current_version_id", table_name="file_entries")
|
|
op.drop_table("file_entries")
|
|
|
|
op.drop_index("idx_upload_sessions_updated_at", table_name="upload_sessions")
|
|
op.drop_index("idx_upload_sessions_status", table_name="upload_sessions")
|
|
op.drop_table("upload_sessions")
|
|
|
|
op.drop_index("idx_backends_priority", table_name="backends")
|
|
op.drop_index("idx_backends_enabled", table_name="backends")
|
|
op.drop_table("backends")
|
|
|
|
op.drop_index("idx_directories_path_key", table_name="directories")
|
|
op.drop_index("idx_directories_parent_id", table_name="directories")
|
|
op.drop_table("directories")
|
|
|
|
op.drop_table("users")
|
|
|