fix(db): add job table to migration 001 and align schema with models

Author: chetanr25

alembic/env.py

@@ -3,9 +3,10 @@
 from alembic import context
 from sqlalchemy import engine_from_config, pool
 
+from sqlmodel import SQLModel
+
 from app.core.config import DATABASE_URL
-from app.models.models import SQLModel 
-from app.models import Template, FormSubmission 
+from app.models import FormSubmission, Job, Template 
 
 config = context.config
 

alembic/versions/001_initial_schema.py

@@ -1,4 +1,4 @@
-"""Initial schema — Template and FormSubmission tables.
+"""Initial schema — Template, FormSubmission, and Job tables.
 
 Revision ID: 001
 Revises:
@@ -24,12 +24,7 @@ def upgrade() -> None:
         sa.Column("name", sqlmodel.sql.sqltypes.AutoString, nullable=False),
         sa.Column("fields", sa.JSON, nullable=False),
         sa.Column("pdf_path", sqlmodel.sql.sqltypes.AutoString, nullable=False),
-        sa.Column(
-            "created_at",
-            sa.DateTime,
-            nullable=False,
-            server_default=sa.text("now()"),
-        ),
+        sa.Column("created_at", sa.DateTime, nullable=False),
     )
 
     op.create_table(
@@ -38,15 +33,32 @@ def upgrade() -> None:
         sa.Column("template_id", sa.Integer, sa.ForeignKey("template.id"), nullable=False),
         sa.Column("input_text", sqlmodel.sql.sqltypes.AutoString, nullable=False),
         sa.Column("output_pdf_path", sqlmodel.sql.sqltypes.AutoString, nullable=False),
-        sa.Column(
-            "created_at",
-            sa.DateTime,
-            nullable=False,
-            server_default=sa.text("now()"),
-        ),
+        sa.Column("created_at", sa.DateTime, nullable=False),
     )
 
+    op.create_table(
+        "job",
+        sa.Column("id", sa.Integer, primary_key=True, autoincrement=True),
+        sa.Column("job_id", sqlmodel.sql.sqltypes.AutoString, nullable=False),
+        sa.Column("celery_task_id", sqlmodel.sql.sqltypes.AutoString, nullable=False),
+        sa.Column("job_type", sqlmodel.sql.sqltypes.AutoString, nullable=False),
+        sa.Column("template_id", sa.Integer, sa.ForeignKey("template.id"), nullable=True),
+        sa.Column("input_text", sqlmodel.sql.sqltypes.AutoString, nullable=True),
+        sa.Column("status", sqlmodel.sql.sqltypes.AutoString, nullable=False),
+        sa.Column("progress_percent", sa.Integer, nullable=False),
+        sa.Column("result_url", sqlmodel.sql.sqltypes.AutoString, nullable=True),
+        sa.Column("error", sa.JSON, nullable=True),
+        sa.Column("model", sqlmodel.sql.sqltypes.AutoString, nullable=True),
+        sa.Column("created_at", sa.DateTime, nullable=False),
+        sa.Column("updated_at", sa.DateTime, nullable=False),
+    )
+    op.create_index("ix_job_job_id", "job", ["job_id"], unique=True)
+    op.create_index("ix_job_celery_task_id", "job", ["celery_task_id"], unique=False)
+
 
 def downgrade() -> None:
+    op.drop_index("ix_job_celery_task_id", table_name="job")
+    op.drop_index("ix_job_job_id", table_name="job")
+    op.drop_table("job")
     op.drop_table("formsubmission")
     op.drop_table("template")

app/models/models.py

@@ -8,7 +8,7 @@
 class Template(SQLModel, table=True):
     id: int | None = Field(default=None, primary_key=True)
     name: str
-    fields: dict = Field(sa_column=Column(JSON))
+    fields: dict = Field(sa_column=Column(JSON, nullable=False))
     pdf_path: str
     created_at: datetime = Field(default_factory=lambda: datetime.now(timezone.utc))