A trick to get past the update issues, hopefully

schema/versions/c3f9f47e7b4f_ws_717_fix_capability_name_foreign_keys.py

@@ -27,6 +27,11 @@ def upgrade():
         UPDATE capability_executions SET capability_name = v.capability_name FROM capability_versions v
         WHERE capability_executions.capability_version_number = v.version_number AND capability_executions.capability_request_id = v.capability_request_id;
         ALTER TABLE capability_executions ALTER COLUMN capability_name SET NOT NULL;
+        
+        UPDATE capability_executions
+        SET state = new_state.new_state
+        FROM (VALUES ('ExecutingWorkflow', 'Executing'), ('Created', 'Start'), ('Waiting', 'Queued'), ('Awaiting QA', 'Complete')) new_state(old_state, new_state)
+        WHERE state = new_state.old_state;
 
         ALTER TABLE capability_executions ADD CONSTRAINT capability_execution_state_fkey FOREIGN KEY (capability_name, state) REFERENCES capability_states(capability_name, state);
     """
@@ -38,5 +43,10 @@ def downgrade():
         """
         ALTER TABLE capability_executions DROP COLUMN capability_name;
         ALTER TABLE capability_versions DROP COLUMN capability_name;
+        
+        UPDATE capability_executions
+        SET state = new_state.old_state
+        FROM (VALUES ('ExecutingWorkflow', 'Executing'), ('Created', 'Start'), ('Waiting', 'Queued'), ('Awaiting QA', 'Complete')) new_state(old_state, new_state)
+        WHERE state = new_state.new_state;
     """
     )