From e5bdee6cb7156b93696159897b47047ee612bbaa Mon Sep 17 00:00:00 2001
From: Luc Didry <luc@didry.org>
Date: Wed, 28 Feb 2024 14:16:04 +0100
Subject: [PATCH] =?UTF-8?q?=F0=9F=97=83=20=E2=80=94=20Alembic=20migration?=
 =?UTF-8?q?=20for=20adding=20severity=20to=20Task?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ..._add_severity_to_task_and_add_severity_.py | 33 +++++++++++++++++++
 1 file changed, 33 insertions(+)
 create mode 100644 alembic/versions/e99bc35702c9_add_severity_to_task_and_add_severity_.py

diff --git a/alembic/versions/e99bc35702c9_add_severity_to_task_and_add_severity_.py b/alembic/versions/e99bc35702c9_add_severity_to_task_and_add_severity_.py
new file mode 100644
index 0000000..80c1f13
--- /dev/null
+++ b/alembic/versions/e99bc35702c9_add_severity_to_task_and_add_severity_.py
@@ -0,0 +1,33 @@
+"""Add severity to Task and add severity level UNKNOWN
+
+Revision ID: e99bc35702c9
+Revises: 7d480e6f1112
+Create Date: 2024-02-28 14:14:22.519918
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = 'e99bc35702c9'
+down_revision: Union[str, None] = '7d480e6f1112'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.execute("ALTER TYPE severity ADD VALUE 'unknown'")
+    op.add_column('tasks', sa.Column('severity', sa.Enum('ok', 'warning', 'critical', 'unknown', name='severity'), nullable=False))
+    op.add_column('tasks', sa.Column('last_severity_update', sa.DateTime(), nullable=True))
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column('tasks', 'last_severity_update')
+    op.drop_column('tasks', 'severity')
+    # ### end Alembic commands ###