hoàn thành bước tạo docker compose

2026-08-05 14:33:10 +07:00 · 2026-03-31 14:47:29 +07:00
parent c362692c51
commit 5d5ec7fb1a
16 changed files with 427 additions and 3 deletions
@@ -0,0 +1,102 @@
+"""initial schema: users, pdfs, annotations
+
+Revision ID: 0001
+Revises:
+Create Date: 2026-03-31
+"""
+
+revision = "0001"
+down_revision = None
+branch_labels = None
+depends_on = None
+
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.dialects import postgresql
+
+
+def upgrade() -> None:
+    # ── users ─────────────────────────────────────────────────────────────────
+    op.create_table(
+        "users",
+        sa.Column("id", sa.Integer(), nullable=False),
+        sa.Column("username", sa.String(length=64), nullable=False),
+        sa.Column("email", sa.String(length=255), nullable=False),
+        sa.Column("password_hash", sa.String(length=255), nullable=False),
+        sa.Column(
+            "created_at",
+            sa.DateTime(timezone=True),
+            nullable=False,
+            server_default=sa.text("now()"),
+        ),
+        sa.PrimaryKeyConstraint("id"),
+    )
+    op.create_index("ix_users_id", "users", ["id"])
+    op.create_index("ix_users_username", "users", ["username"], unique=True)
+    op.create_index("ix_users_email", "users", ["email"], unique=True)
+
+    # ── pdfs ──────────────────────────────────────────────────────────────────
+    op.create_table(
+        "pdfs",
+        sa.Column("id", sa.Integer(), nullable=False),
+        sa.Column("user_id", sa.Integer(), nullable=False),
+        sa.Column("title", sa.String(length=255), nullable=False),
+        sa.Column("file_path", sa.Text(), nullable=False),
+        sa.Column("total_pages", sa.Integer(), nullable=True),
+        sa.Column(
+            "created_at",
+            sa.DateTime(timezone=True),
+            nullable=False,
+            server_default=sa.text("now()"),
+        ),
+        sa.ForeignKeyConstraint(["user_id"], ["users.id"], ondelete="CASCADE"),
+        sa.PrimaryKeyConstraint("id"),
+    )
+    op.create_index("ix_pdfs_id", "pdfs", ["id"])
+    op.create_index("ix_pdfs_user_id", "pdfs", ["user_id"])
+
+    # ── annotations ───────────────────────────────────────────────────────────
+    op.create_table(
+        "annotations",
+        sa.Column("id", sa.Integer(), nullable=False),
+        sa.Column("pdf_id", sa.Integer(), nullable=False),
+        sa.Column("user_id", sa.Integer(), nullable=False),
+        sa.Column("page_number", sa.Integer(), nullable=False),
+        sa.Column(
+            "canvas_data",
+            postgresql.JSONB(astext_type=sa.Text()),
+            nullable=False,
+            server_default="{}",
+        ),
+        sa.Column(
+            "updated_at",
+            sa.DateTime(timezone=True),
+            nullable=False,
+            server_default=sa.text("now()"),
+        ),
+        sa.ForeignKeyConstraint(["pdf_id"], ["pdfs.id"], ondelete="CASCADE"),
+        sa.ForeignKeyConstraint(["user_id"], ["users.id"], ondelete="CASCADE"),
+        sa.PrimaryKeyConstraint("id"),
+        sa.UniqueConstraint(
+            "pdf_id", "user_id", "page_number", name="uq_annotation_pdf_user_page"
+        ),
+    )
+    op.create_index("ix_annotations_id", "annotations", ["id"])
+    op.create_index("ix_annotations_pdf_id", "annotations", ["pdf_id"])
+    op.create_index("ix_annotations_user_id", "annotations", ["user_id"])
+
+
+def downgrade() -> None:
+    op.drop_index("ix_annotations_user_id", table_name="annotations")
+    op.drop_index("ix_annotations_pdf_id", table_name="annotations")
+    op.drop_index("ix_annotations_id", table_name="annotations")
+    op.drop_table("annotations")
+
+    op.drop_index("ix_pdfs_user_id", table_name="pdfs")
+    op.drop_index("ix_pdfs_id", table_name="pdfs")
+    op.drop_table("pdfs")
+
+    op.drop_index("ix_users_email", table_name="users")
+    op.drop_index("ix_users_username", table_name="users")
+    op.drop_index("ix_users_id", table_name="users")
+    op.drop_table("users")