Update 0451_create_db_users.py

After further discussions with the team, we have decided that we should make our downgrades more representative of a reversal of what an upgrade does, but while also safely been tested for upgrade and downgrade paths
cds-snc · May 29, 2024 · 94d6a55 · 94d6a55
1 parent 1cd72c9
commit 94d6a55
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/migrations/versions/0451_create_db_users.py b/migrations/versions/0451_create_db_users.py
@@ -22,7 +22,9 @@ def upgrade():
 
 
 def downgrade():
-    return
+    for role in roles:
+        op.execute(f"REVOKE {super_role} FROM {role};")
+        op.execute(f"DROP ROLE IF EXISTS {role};")
 
 
 def create_role_if_not_exist(role):