added mark for cancellation in DB

Author: sanderegg

packages/postgres-database/src/simcore_postgres_database/migration/versions/10097b74cf5a_added_cancellation_mark.py

@@ -0,0 +1,35 @@
+"""added_cancellation_mark
+
+Revision ID: 10097b74cf5a
+Revises: fce5d231e16d
+Create Date: 2024-11-01 17:24:21.002659+00:00
+
+"""
+import sqlalchemy as sa
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision = "10097b74cf5a"
+down_revision = "fce5d231e16d"
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column(
+        "comp_runs",
+        sa.Column(
+            "marked_for_cancellation",
+            sa.Boolean(),
+            server_default=sa.text("false"),
+            nullable=False,
+        ),
+    )
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column("comp_runs", "marked_for_cancellation")
+    # ### end Alembic commands ###

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

@@ -1,6 +1,7 @@
 """ Computational Runs Table
 
 """
+
 import sqlalchemy as sa
 from sqlalchemy.dialects.postgresql import JSONB
 from sqlalchemy.sql import func
@@ -106,5 +107,12 @@
         nullable=False,
         doc="the run uses on demand clusters",
     ),
+    sa.Column(
+        "marked_for_cancellation",
+        sa.Boolean(),
+        server_default=sa.sql.expression.false(),
+        nullable=False,
+        doc="the run was marked for cancellation",
+    ),
     sa.UniqueConstraint("project_uuid", "user_id", "iteration"),
 )