Update

geointel/backend/alembic/versions/202601120001_dataset_storage_metadata.py

@@ -0,0 +1,27 @@
+"""Add dataset storage metadata columns."""
+
+from alembic import op
+import sqlalchemy as sa
+
+
+revision = "202601120001"
+down_revision = "202601110001"
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    op.add_column("datasets", sa.Column("original_filename", sa.Text(), nullable=True))
+    op.add_column("datasets", sa.Column("stored_filename", sa.Text(), nullable=True))
+    op.add_column("datasets", sa.Column("content_type", sa.Text(), nullable=True))
+    op.add_column("datasets", sa.Column("size_bytes", sa.Integer(), nullable=True))
+    op.add_column("datasets", sa.Column("checksum_sha256", sa.Text(), nullable=True))
+    op.alter_column("datasets", "status", server_default="uploaded")
+
+
+def downgrade() -> None:
+    op.drop_column("datasets", "checksum_sha256")
+    op.drop_column("datasets", "size_bytes")
+    op.drop_column("datasets", "content_type")
+    op.drop_column("datasets", "stored_filename")
+    op.drop_column("datasets", "original_filename")