add server default to dag field

Author: matusdrobuliak66

packages/postgres-database/src/simcore_postgres_database/migration/versions/bc978c56397b_add_comp_run_snapshot_tasks_table_2.py

@@ -1,8 +1,8 @@
 """add comp_run_snapshot_tasks table
 
-Revision ID: 4e9ac9ea7eca
+Revision ID: e89eae27fb3f
 Revises: 278daef7e99d
-Create Date: 2025-05-28 14:00:23.502544+00:00
+Create Date: 2025-05-29 16:52:00.435268+00:00
 
 """
 
@@ -11,7 +11,7 @@
 from sqlalchemy.dialects import postgresql
 
 # revision identifiers, used by Alembic.
-revision = "4e9ac9ea7eca"
+revision = "e89eae27fb3f"
 down_revision = "278daef7e99d"
 branch_labels = None
 depends_on = None
@@ -69,13 +69,34 @@ def upgrade():
     op.add_column(
         "comp_runs",
         sa.Column(
-            "dag_adjacency_list", postgresql.JSONB(astext_type=sa.Text()), nullable=True
+            "dag_adjacency_list",
+            postgresql.JSONB(astext_type=sa.Text()),
+            server_default=sa.text("'{}'::jsonb"),
+            nullable=False,
         ),
     )
     # ### end Alembic commands ###
     op.execute("ALTER TABLE comp_run_snapshot_tasks ADD COLUMN node_class nodeclass;")
     op.execute("ALTER TABLE comp_run_snapshot_tasks ADD COLUMN state statetype;")
 
+    op.alter_column(
+        "comp_run_snapshot_tasks",
+        "state",
+        existing_type=postgresql.ENUM(
+            "NOT_STARTED",
+            "PUBLISHED",
+            "PENDING",
+            "RUNNING",
+            "SUCCESS",
+            "FAILED",
+            "ABORTED",
+            "WAITING_FOR_RESOURCES",
+            "WAITING_FOR_CLUSTER",
+            name="statetype",
+        ),
+        nullable=False,
+    )
+
 
 def downgrade():
     # ### commands auto generated by Alembic - please adjust! ###

packages/postgres-database/src/simcore_postgres_database/migration/versions/4e9ac9ea7eca_add_comp_run_snapshot_tasks_table.py renamed to packages/postgres-database/src/simcore_postgres_database/migration/versions/e89eae27fb3f_add_comp_run_snapshot_tasks_table.py

@@ -99,7 +99,11 @@
         doc="the run uses on demand clusters",
     ),
     sa.Column(
-        "dag_adjacency_list", JSONB, doc="Adjacency list for the pipeline's graph"
+        "dag_adjacency_list",
+        JSONB,
+        doc="Adjacency list for the pipeline's graph",
+        server_default=sa.text("'{}'::jsonb"),
+        nullable=False,
     ),
     sa.UniqueConstraint("project_uuid", "user_id", "iteration"),
     sa.Index("ix_comp_runs_user_id", "user_id"),

packages/postgres-database/src/simcore_postgres_database/models/comp_runs.py

@@ -36,7 +36,7 @@ async def sample_comp_runs(asyncpg_engine: AsyncEngine):
                     "processed": None,
                     "metadata": None,
                     "use_on_demand_clusters": False,
-                    "dag_adjacency_list": None,
+                    "dag_adjacency_list": {},
                 },
                 {
                     "run_id": 2,
@@ -57,7 +57,7 @@ async def sample_comp_runs(asyncpg_engine: AsyncEngine):
                     "processed": None,
                     "metadata": None,
                     "use_on_demand_clusters": False,
-                    "dag_adjacency_list": None,
+                    "dag_adjacency_list": {},
                 },
                 {
                     "run_id": 3,
@@ -78,7 +78,7 @@ async def sample_comp_runs(asyncpg_engine: AsyncEngine):
                     "processed": None,
                     "metadata": None,
                     "use_on_demand_clusters": False,
-                    "dag_adjacency_list": None,
+                    "dag_adjacency_list": {},
                 },
                 {
                     "run_id": 4,
@@ -99,7 +99,7 @@ async def sample_comp_runs(asyncpg_engine: AsyncEngine):
                     "processed": None,
                     "metadata": None,
                     "use_on_demand_clusters": False,
-                    "dag_adjacency_list": None,
+                    "dag_adjacency_list": {},
                 },
             ],
         )