Create a CANCELLING state type (#7794)

Co-authored-by: Zach Angell <[email protected]>
PrefectHQ · Feb 6, 2023 · fb66864 · fb66864
1 parent 3311cff
commit fb66864
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 2 deletions.
diff --git a/...s/versions/postgresql/2022_12_06_164028_9326a6aee18b_add_cancelling_to_state_type_enum.py b/...s/versions/postgresql/2022_12_06_164028_9326a6aee18b_add_cancelling_to_state_type_enum.py
@@ -0,0 +1,22 @@
+"""Add CANCELLING to state type enum
+
+Revision ID: 9326a6aee18b
+Revises: f7587d6c5776
+Create Date: 2022-12-06 16:40:28.282753
+
+"""
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision = "9326a6aee18b"
+down_revision = "f7587d6c5776"
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    op.execute("ALTER TYPE state_type ADD VALUE IF NOT EXISTS 'CANCELLING';")
+
+
+def downgrade():
+    pass
diff --git a/...s/versions/postgresql/2023_01_08_180142_d481d5058a19_rename_worker_pools_to_work_pools.py b/...s/versions/postgresql/2023_01_08_180142_d481d5058a19_rename_worker_pools_to_work_pools.py
@@ -1,7 +1,7 @@
 """Rename Worker Pools to Work Pools
 
 Revision ID: d481d5058a19
-Revises: f7587d6c5776
+Revises: 9326a6aee18b
 Create Date: 2023-01-08 18:01:42.559990
 
 """
@@ -12,7 +12,7 @@
 
 # revision identifiers, used by Alembic.
 revision = "d481d5058a19"
-down_revision = "f7587d6c5776"
+down_revision = "9326a6aee18b"
 branch_labels = None
 depends_on = None