Merge pull request #21 from lsst-sqre/tickets/DM-40727

DM-40727: add missing migration

CHANGELOG.rst

@@ -2,6 +2,11 @@
 Change Log
 ==========
 
+0.5.1
+-----
+
+* Add missing migration to set 'systems', 'subsystems', and 'cscs' as nullable fields.
+
 0.5.0
 -----
 

alembic/versions/d9606992ad8d_update_message_table_systems_subsystems_.py

@@ -0,0 +1,40 @@
+"""update message table systems, subsystems and cscs fields
+
+Revision ID: d9606992ad8d
+Revises: b631d21eb6dc
+Create Date: 2023-09-11 17:44:07.705399
+
+"""
+import logging
+
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision = "d9606992ad8d"
+down_revision = "b631d21eb6dc"
+branch_labels = None
+depends_on = None
+
+MESSAGE_TABLE_NAME = "message"
+
+
+def upgrade(log: logging.Logger, table_names: set[str]):
+    if MESSAGE_TABLE_NAME not in table_names:
+        log.info(f"No {MESSAGE_TABLE_NAME} table; nothing to do")
+        return
+    log.info("Set columns 'systems', 'subsystems', and 'cscs' as nullable")
+
+    op.alter_column(MESSAGE_TABLE_NAME, "systems", nullable=True)
+    op.alter_column(MESSAGE_TABLE_NAME, "subsystems", nullable=True)
+    op.alter_column(MESSAGE_TABLE_NAME, "cscs", nullable=True)
+
+
+def downgrade(log: logging.Logger, table_names: set[str]):
+    if MESSAGE_TABLE_NAME not in table_names:
+        log.info(f"No {MESSAGE_TABLE_NAME} table; nothing to do")
+        return
+
+    log.info("Set columns 'systems', 'subsystems', and 'cscs' as not nullable")
+    op.alter_column(MESSAGE_TABLE_NAME, "systems", nullable=False)
+    op.alter_column(MESSAGE_TABLE_NAME, "subsystems", nullable=False)
+    op.alter_column(MESSAGE_TABLE_NAME, "cscs", nullable=False)