WIP: add single database migration

src/api/qualicharge/migrations/versions/fd6fc1cd048a_create_static_schema.py → src/api/qualicharge/migrations/versions/2664a0b4ce11_add_static_schema.py

@@ -1,8 +1,8 @@
-"""create static schema
+"""add static schema
 
-Revision ID: fd6fc1cd048a
+Revision ID: 2664a0b4ce11
 Revises:
-Create Date: 2024-04-12 14:16:26.196085
+Create Date: 2024-04-23 14:51:04.993896
 
 """
 
@@ -13,8 +13,9 @@
 import sqlmodel
 from alembic import op
 
+
 # revision identifiers, used by Alembic.
-revision: str = "fd6fc1cd048a"
+revision: str = "2664a0b4ce11"
 down_revision: Union[str, None] = None
 branch_labels: Union[str, Sequence[str], None] = None
 depends_on: Union[str, Sequence[str], None] = None
@@ -24,20 +25,30 @@ def upgrade() -> None:
     # ### commands auto generated by Alembic - please adjust! ###
     op.create_table(
         "amenageur",
+        sa.Column("created_at", sa.DateTime(timezone=True), nullable=False),
+        sa.Column("updated_at", sa.DateTime(timezone=True), nullable=False),
         sa.Column("id", sqlmodel.sql.sqltypes.GUID(), nullable=False),
         sa.Column("nom_amenageur", sqlmodel.sql.sqltypes.AutoString(), nullable=True),
         sa.Column("siren_amenageur", sqlmodel.sql.sqltypes.AutoString(), nullable=True),
         sa.Column("contact_amenageur", sa.String(), nullable=True),
+        sa.CheckConstraint("created_at <= updated_at", name="pre-creation-update"),
         sa.PrimaryKeyConstraint("id"),
+        sa.UniqueConstraint("nom_amenageur", "siren_amenageur", "contact_amenageur"),
     )
     op.create_table(
         "enseigne",
+        sa.Column("created_at", sa.DateTime(timezone=True), nullable=False),
+        sa.Column("updated_at", sa.DateTime(timezone=True), nullable=False),
         sa.Column("id", sqlmodel.sql.sqltypes.GUID(), nullable=False),
         sa.Column("nom_enseigne", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
+        sa.CheckConstraint("created_at <= updated_at", name="pre-creation-update"),
         sa.PrimaryKeyConstraint("id"),
+        sa.UniqueConstraint("nom_enseigne"),
     )
     op.create_table(
         "localisation",
+        sa.Column("created_at", sa.DateTime(timezone=True), nullable=False),
+        sa.Column("updated_at", sa.DateTime(timezone=True), nullable=False),
         sa.Column("id", sqlmodel.sql.sqltypes.GUID(), nullable=False),
         sa.Column(
             "adresse_station", sqlmodel.sql.sqltypes.AutoString(), nullable=False
@@ -56,20 +67,37 @@ def upgrade() -> None:
             ),
             nullable=False,
         ),
+        sa.CheckConstraint("created_at <= updated_at", name="pre-creation-update"),
         sa.PrimaryKeyConstraint("id"),
+        sa.UniqueConstraint("adresse_station", "coordonneesXY"),
     )
+    # op.create_index(
+    #     "idx_localisation_coordonneesXY",
+    #     "localisation",
+    #     ["coordonneesXY"],
+    #     unique=False,
+    #     postgresql_using="gist",
+    # )
     op.create_table(
         "operateur",
+        sa.Column("created_at", sa.DateTime(timezone=True), nullable=False),
+        sa.Column("updated_at", sa.DateTime(timezone=True), nullable=False),
         sa.Column("id", sqlmodel.sql.sqltypes.GUID(), nullable=False),
         sa.Column("nom_operateur", sqlmodel.sql.sqltypes.AutoString(), nullable=True),
         sa.Column("contact_operateur", sa.String(), nullable=False),
         sa.Column(
             "telephone_operateur", sqlmodel.sql.sqltypes.AutoString(), nullable=True
         ),
+        sa.CheckConstraint("created_at <= updated_at", name="pre-creation-update"),
         sa.PrimaryKeyConstraint("id"),
+        sa.UniqueConstraint(
+            "nom_operateur", "contact_operateur", "telephone_operateur"
+        ),
     )
     op.create_table(
         "station",
+        sa.Column("created_at", sa.DateTime(timezone=True), nullable=False),
+        sa.Column("updated_at", sa.DateTime(timezone=True), nullable=False),
         sa.Column("id", sqlmodel.sql.sqltypes.GUID(), nullable=False),
         sa.Column(
             "id_station_itinerance", sqlmodel.sql.sqltypes.AutoString(), nullable=False
@@ -106,10 +134,11 @@ def upgrade() -> None:
         sa.Column("num_pdl", sqlmodel.sql.sqltypes.AutoString(), nullable=True),
         sa.Column("date_maj", sa.DateTime(), nullable=False),
         sa.Column("date_mise_en_service", sa.DateTime(), nullable=True),
-        sa.Column("amenageur_id", sqlmodel.sql.sqltypes.GUID(), nullable=False),
-        sa.Column("operateur_id", sqlmodel.sql.sqltypes.GUID(), nullable=False),
-        sa.Column("enseigne_id", sqlmodel.sql.sqltypes.GUID(), nullable=False),
-        sa.Column("localisation_id", sqlmodel.sql.sqltypes.GUID(), nullable=False),
+        sa.Column("amenageur_id", sqlmodel.sql.sqltypes.GUID(), nullable=True),
+        sa.Column("operateur_id", sqlmodel.sql.sqltypes.GUID(), nullable=True),
+        sa.Column("enseigne_id", sqlmodel.sql.sqltypes.GUID(), nullable=True),
+        sa.Column("localisation_id", sqlmodel.sql.sqltypes.GUID(), nullable=True),
+        sa.CheckConstraint("created_at <= updated_at", name="pre-creation-update"),
         sa.ForeignKeyConstraint(
             ["amenageur_id"],
             ["amenageur.id"],
@@ -132,10 +161,12 @@ def upgrade() -> None:
         op.f("ix_station_id_station_itinerance"),
         "station",
         ["id_station_itinerance"],
-        unique=False,
+        unique=True,
     )
     op.create_table(
         "pointdecharge",
+        sa.Column("created_at", sa.DateTime(timezone=True), nullable=False),
+        sa.Column("updated_at", sa.DateTime(timezone=True), nullable=False),
         sa.Column("id", sqlmodel.sql.sqltypes.GUID(), nullable=False),
         sa.Column(
             "id_pdc_itinerance", sqlmodel.sql.sqltypes.AutoString(), nullable=False
@@ -169,7 +200,8 @@ def upgrade() -> None:
         ),
         sa.Column("observations", sqlmodel.sql.sqltypes.AutoString(), nullable=True),
         sa.Column("cable_t2_attache", sa.Boolean(), nullable=True),
-        sa.Column("station_id", sqlmodel.sql.sqltypes.GUID(), nullable=False),
+        sa.Column("station_id", sqlmodel.sql.sqltypes.GUID(), nullable=True),
+        sa.CheckConstraint("created_at <= updated_at", name="pre-creation-update"),
         sa.ForeignKeyConstraint(
             ["station_id"],
             ["station.id"],
@@ -194,6 +226,11 @@ def downgrade() -> None:
     op.drop_index(op.f("ix_station_id_station_itinerance"), table_name="station")
     op.drop_table("station")
     op.drop_table("operateur")
+    op.drop_index(
+        "idx_localisation_coordonneesXY",
+        table_name="localisation",
+        postgresql_using="gist",
+    )
     op.drop_table("localisation")
     op.drop_table("enseigne")
     op.drop_table("amenageur")