Add unique constraint in flexdata table

lsst-dm · Sep 16, 2024 · 8f45734 · 8f45734
1 parent d2fcde9
commit 8f45734
Show file tree

Hide file tree

Showing 6 changed files with 164 additions and 41 deletions.
diff --git a/alembic/latiss/versions/53707815663e_change_primary_key_to_day_obs_seq_num.py b/alembic/latiss/versions/53707815663e_change_primary_key_to_day_obs_seq_num.py
@@ -21,6 +21,10 @@
 
 def upgrade() -> None:
     # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_constraint("fk_exposure_id", "ccdexposure", schema="cdb_latiss", type_="foreignkey")
+    op.drop_constraint("fk_obs_id", "exposure_flexdata", schema="cdb_latiss", type_="foreignkey")
+    op.drop_constraint("un_exposure_id_detector", "ccdexposure", schema="cdb_latiss", type_="unique")
+    op.drop_constraint("un_day_obs_seq_num", "exposure", schema="cdb_latiss", type_="unique")
     op.add_column(
         "ccdexposure",
         sa.Column(
@@ -41,7 +45,6 @@ def upgrade() -> None:
         ),
         schema="cdb_latiss",
     )
-    op.drop_constraint("un_exposure_id_detector", "ccdexposure", schema="cdb_latiss", type_="unique")
     op.create_unique_constraint(
         "un_ccdexposure_ccdexposure_id", "ccdexposure", ["ccdexposure_id"], schema="cdb_latiss"
     )
@@ -51,17 +54,6 @@ def upgrade() -> None:
         ["day_obs", "seq_num", "detector"],
         schema="cdb_latiss",
     )
-    op.drop_constraint("fk_exposure_id", "ccdexposure", schema="cdb_latiss", type_="foreignkey")
-    op.create_foreign_key(
-        "fk_ccdexposure_day_obs_seq_num",
-        "ccdexposure",
-        "exposure",
-        ["day_obs", "seq_num"],
-        ["day_obs", "seq_num"],
-        source_schema="cdb_latiss",
-        referent_schema="cdb_latiss",
-    )
-    op.drop_constraint("un_day_obs_seq_num", "exposure", schema="cdb_latiss", type_="unique")
     op.create_unique_constraint(
         "un_exposure_day_obs_seq_num", "exposure", ["day_obs", "seq_num"], schema="cdb_latiss"
     )
@@ -86,7 +78,6 @@ def upgrade() -> None:
         ),
         schema="cdb_latiss",
     )
-    op.drop_constraint("fk_obs_id", "exposure_flexdata", schema="cdb_latiss", type_="foreignkey")
     op.create_foreign_key(
         "fk_exposure_flexdata_day_obs_seq_num",
         "exposure_flexdata",
@@ -135,6 +126,15 @@ def upgrade() -> None:
         source_schema="cdb_latiss",
         referent_schema="cdb_latiss",
     )
+    op.create_foreign_key(
+        "fk_ccdexposure_day_obs_seq_num",
+        "ccdexposure",
+        "exposure",
+        ["day_obs", "seq_num"],
+        ["day_obs", "seq_num"],
+        source_schema="cdb_latiss",
+        referent_schema="cdb_latiss",
+    )
     # ### end Alembic commands ###
 
     # Added by hand: copy day_obs and seq_num into the ccdexposure,

diff --git a/alembic/latiss/versions/56077b746de8_add_unique_constraint_for_day_obs_seq_.py b/alembic/latiss/versions/56077b746de8_add_unique_constraint_for_day_obs_seq_.py
@@ -0,0 +1,39 @@
+"""add unique constraint for day obs seq num key
+
+Revision ID: 56077b746de8
+Revises: 53707815663e
+Create Date: 2024-09-16 17:31:22.643951+00:00
+
+"""
+
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.dialects import mysql
+from sqlalchemy.dialects import postgresql
+
+# revision identifiers, used by Alembic.
+revision: str = "56077b746de8"
+down_revision: Union[str, None] = "53707815663e"
+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.create_unique_constraint(
+        "un_exposure_flexdata_day_obs_seq_num_key",
+        "exposure_flexdata",
+        ["day_obs", "seq_num", "key"],
+        schema="cdb_latiss",
+    )
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_constraint(
+        "un_exposure_flexdata_day_obs_seq_num_key", "exposure_flexdata", schema="cdb_latiss", type_="unique"
+    )
+    # ### end Alembic commands ###
diff --git a/alembic/lsstcomcam/versions/67319407b5fc_add_unique_constraint_for_day_obs_seq_.py b/alembic/lsstcomcam/versions/67319407b5fc_add_unique_constraint_for_day_obs_seq_.py
@@ -0,0 +1,42 @@
+"""add unique constraint for day obs seq num key
+
+Revision ID: 67319407b5fc
+Revises: bf7ed261cc12
+Create Date: 2024-09-16 17:31:24.124147+00:00
+
+"""
+
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.dialects import mysql
+from sqlalchemy.dialects import postgresql
+
+# revision identifiers, used by Alembic.
+revision: str = "67319407b5fc"
+down_revision: Union[str, None] = "bf7ed261cc12"
+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.create_unique_constraint(
+        "un_exposure_flexdata_day_obs_seq_num_key",
+        "exposure_flexdata",
+        ["day_obs", "seq_num", "key"],
+        schema="cdb_lsstcomcam",
+    )
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_constraint(
+        "un_exposure_flexdata_day_obs_seq_num_key",
+        "exposure_flexdata",
+        schema="cdb_lsstcomcam",
+        type_="unique",
+    )
+    # ### end Alembic commands ###
diff --git a/alembic/lsstcomcam/versions/bf7ed261cc12_change_primary_key_to_day_obs_seq_num.py b/alembic/lsstcomcam/versions/bf7ed261cc12_change_primary_key_to_day_obs_seq_num.py
@@ -21,6 +21,13 @@
 
 def upgrade() -> None:
     # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_constraint("fk_obs_id", "exposure_flexdata", schema="cdb_lsstcomcam", type_="foreignkey")
+    op.drop_constraint("fk_exposure_id", "ccdexposure", schema="cdb_lsstcomcam", type_="foreignkey")
+    op.drop_constraint("un_exposure_id_detector", "ccdexposure", schema="cdb_lsstcomcam", type_="unique")
+    op.drop_constraint("un_day_obs_seq_num", "exposure", schema="cdb_lsstcomcam", type_="unique")
+    op.create_unique_constraint(
+        "un_ccdexposure_ccdexposure_id", "ccdexposure", ["ccdexposure_id"], schema="cdb_lsstcomcam"
+    )
     op.add_column(
         "ccdexposure",
         sa.Column(
@@ -41,26 +48,12 @@ def upgrade() -> None:
         ),
         schema="cdb_lsstcomcam",
     )
-    op.drop_constraint("un_exposure_id_detector", "ccdexposure", schema="cdb_lsstcomcam", type_="unique")
-    op.create_unique_constraint(
-        "un_ccdexposure_ccdexposure_id", "ccdexposure", ["ccdexposure_id"], schema="cdb_lsstcomcam"
-    )
     op.create_unique_constraint(
         "un_ccdexposure_day_obs_seq_num_detector",
         "ccdexposure",
         ["day_obs", "seq_num", "detector"],
         schema="cdb_lsstcomcam",
     )
-    op.drop_constraint("fk_exposure_id", "ccdexposure", schema="cdb_lsstcomcam", type_="foreignkey")
-    op.create_foreign_key(
-        "fk_ccdexposure_day_obs_seq_num",
-        "ccdexposure",
-        "exposure",
-        ["day_obs", "seq_num"],
-        ["day_obs", "seq_num"],
-        source_schema="cdb_lsstcomcam",
-        referent_schema="cdb_lsstcomcam",
-    )
     op.add_column(
         "ccdvisit1_quicklook",
         sa.Column(
@@ -71,7 +64,6 @@ def upgrade() -> None:
         ),
         schema="cdb_lsstcomcam",
     )
-    op.drop_constraint("un_day_obs_seq_num", "exposure", schema="cdb_lsstcomcam", type_="unique")
     op.create_unique_constraint(
         "un_exposure_day_obs_seq_num", "exposure", ["day_obs", "seq_num"], schema="cdb_lsstcomcam"
     )
@@ -98,7 +90,6 @@ def upgrade() -> None:
         ),
         schema="cdb_lsstcomcam",
     )
-    op.drop_constraint("fk_obs_id", "exposure_flexdata", schema="cdb_lsstcomcam", type_="foreignkey")
     op.create_foreign_key(
         "fk_exposure_flexdata_day_obs_seq_num",
         "exposure_flexdata",
@@ -167,6 +158,15 @@ def upgrade() -> None:
         source_schema="cdb_lsstcomcam",
         referent_schema="cdb_lsstcomcam",
     )
+    op.create_foreign_key(
+        "fk_ccdexposure_day_obs_seq_num",
+        "ccdexposure",
+        "exposure",
+        ["day_obs", "seq_num"],
+        ["day_obs", "seq_num"],
+        source_schema="cdb_lsstcomcam",
+        referent_schema="cdb_lsstcomcam",
+    )
     # ### end Alembic commands ###
 
     # Added by hand: copy day_obs and seq_num into the ccdexposure,

diff --git a/alembic/lsstcomcamsim/versions/02f64409522c_add_unique_constraint_for_day_obs_seq_.py b/alembic/lsstcomcamsim/versions/02f64409522c_add_unique_constraint_for_day_obs_seq_.py
@@ -0,0 +1,42 @@
+"""add unique constraint for day obs seq num key
+
+Revision ID: 02f64409522c
+Revises: 5f50b32c44fc
+Create Date: 2024-09-16 17:31:25.545455+00:00
+
+"""
+
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.dialects import mysql
+from sqlalchemy.dialects import postgresql
+
+# revision identifiers, used by Alembic.
+revision: str = "02f64409522c"
+down_revision: Union[str, None] = "5f50b32c44fc"
+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.create_unique_constraint(
+        "un_exposure_flexdata_day_obs_seq_num_key",
+        "exposure_flexdata",
+        ["day_obs", "seq_num", "key"],
+        schema="cdb_lsstcomcamsim",
+    )
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_constraint(
+        "un_exposure_flexdata_day_obs_seq_num_key",
+        "exposure_flexdata",
+        schema="cdb_lsstcomcamsim",
+        type_="unique",
+    )
+    # ### end Alembic commands ###
diff --git a/alembic/lsstcomcamsim/versions/5f50b32c44fc_change_primary_key_to_day_obs_seq_num.py b/alembic/lsstcomcamsim/versions/5f50b32c44fc_change_primary_key_to_day_obs_seq_num.py
@@ -41,7 +41,10 @@ def upgrade() -> None:
         ),
         schema="cdb_lsstcomcamsim",
     )
+    op.drop_constraint("fk_obs_id", "exposure_flexdata", schema="cdb_lsstcomcamsim", type_="foreignkey")
+    op.drop_constraint("fk_exposure_id", "ccdexposure", schema="cdb_lsstcomcamsim", type_="foreignkey")
     op.drop_constraint("un_exposure_id_detector", "ccdexposure", schema="cdb_lsstcomcamsim", type_="unique")
+    op.drop_constraint("un_day_obs_seq_num", "exposure", schema="cdb_lsstcomcamsim", type_="unique")
     op.create_unique_constraint(
         "un_ccdexposure_ccdexposure_id", "ccdexposure", ["ccdexposure_id"], schema="cdb_lsstcomcamsim"
     )
@@ -51,16 +54,6 @@ def upgrade() -> None:
         ["day_obs", "seq_num", "detector"],
         schema="cdb_lsstcomcamsim",
     )
-    op.drop_constraint("fk_exposure_id", "ccdexposure", schema="cdb_lsstcomcamsim", type_="foreignkey")
-    op.create_foreign_key(
-        "fk_ccdexposure_day_obs_seq_num",
-        "ccdexposure",
-        "exposure",
-        ["day_obs", "seq_num"],
-        ["day_obs", "seq_num"],
-        source_schema="cdb_lsstcomcamsim",
-        referent_schema="cdb_lsstcomcamsim",
-    )
     op.add_column(
         "ccdvisit1_quicklook",
         sa.Column(
@@ -71,7 +64,6 @@ def upgrade() -> None:
         ),
         schema="cdb_lsstcomcamsim",
     )
-    op.drop_constraint("un_day_obs_seq_num", "exposure", schema="cdb_lsstcomcamsim", type_="unique")
     op.create_unique_constraint(
         "un_exposure_day_obs_seq_num", "exposure", ["day_obs", "seq_num"], schema="cdb_lsstcomcamsim"
     )
@@ -98,7 +90,6 @@ def upgrade() -> None:
         ),
         schema="cdb_lsstcomcamsim",
     )
-    op.drop_constraint("fk_obs_id", "exposure_flexdata", schema="cdb_lsstcomcamsim", type_="foreignkey")
     op.create_foreign_key(
         "fk_exposure_flexdata_day_obs_seq_num",
         "exposure_flexdata",
@@ -167,6 +158,15 @@ def upgrade() -> None:
         source_schema="cdb_lsstcomcamsim",
         referent_schema="cdb_lsstcomcamsim",
     )
+    op.create_foreign_key(
+        "fk_ccdexposure_day_obs_seq_num",
+        "ccdexposure",
+        "exposure",
+        ["day_obs", "seq_num"],
+        ["day_obs", "seq_num"],
+        source_schema="cdb_lsstcomcamsim",
+        referent_schema="cdb_lsstcomcamsim",
+    )
     # ### end Alembic commands ###
 
     # Added by hand: copy day_obs and seq_num into the ccdexposure,