Home | History | Annotate | Download | only in migrations
      1 ADD_FOREIGN_KEYS = """
      2 ALTER TABLE tko_test_labels_tests DROP FOREIGN KEY tests_labels_tests_ibfk_1;
      3 ALTER TABLE tko_test_labels_tests ADD CONSTRAINT tests_labels_tests_ibfk_1
      4     FOREIGN KEY (testlabel_id) REFERENCES tko_test_labels (id)
      5     ON DELETE CASCADE;
      6 
      7 ALTER TABLE tko_test_labels_tests DROP FOREIGN KEY tests_labels_tests_ibfk_2;
      8 ALTER TABLE tko_test_labels_tests ADD CONSTRAINT tests_labels_tests_ibfk_2
      9     FOREIGN KEY (test_id) REFERENCES tko_tests (test_idx) ON DELETE CASCADE;
     10 """
     11 
     12 DROP_FOREIGN_KEYS = """
     13 ALTER TABLE tko_test_labels_tests DROP FOREIGN KEY tests_labels_tests_ibfk_1;
     14 ALTER TABLE tko_test_labels_tests ADD CONSTRAINT tests_labels_tests_ibfk_1
     15     FOREIGN KEY (testlabel_id) REFERENCES tko_test_labels (id);
     16 
     17 ALTER TABLE tko_test_labels_tests DROP FOREIGN KEY tests_labels_tests_ibfk_2;
     18 ALTER TABLE tko_test_labels_tests ADD CONSTRAINT tests_labels_tests_ibfk_2
     19     FOREIGN KEY (test_id) REFERENCES tko_tests (test_idx);
     20 """
     21 
     22 def migrate_up(mgr):
     23     mgr.execute_script(ADD_FOREIGN_KEYS)
     24 
     25 def migrate_down(mgr):
     26     mgr.execute_script(DROP_FOREIGN_KEYS)
     27