(PC-31395)[API] feat: add idAtProvider in PriceCategory

api/alembic_version_conflict_detection.txt

@@ -1,2 +1,2 @@
-869f0d3be788 (pre) (head)
-63fa42fd8352 (post) (head)
+8523f3e2d7d6 (pre) (head)
+404b3075d1a4 (post) (head)

api/src/pcapi/alembic/versions/20240903T113527_8523f3e2d7d6_add_idatprovider_in_price_category.py

@@ -0,0 +1,21 @@
+"""Add idAtProvider in `price_category`
+"""
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# pre/post deployment: pre
+# revision identifiers, used by Alembic.
+revision = "8523f3e2d7d6"
+down_revision = "869f0d3be788"
+branch_labels: tuple[str] | None = None
+depends_on: list[str] | None = None
+
+
+def upgrade() -> None:
+    op.add_column("price_category", sa.Column("idAtProvider", sa.Text(), nullable=True))
+
+
+def downgrade() -> None:
+    op.drop_column("price_category", "idAtProvider")

api/src/pcapi/alembic/versions/20240903T114208_404b3075d1a4_add_unique_index_on_price_category_idatprovider.py

@@ -0,0 +1,34 @@
+"""Add unique index on (`idAProvider`, `offerId`) in `price_category` table
+"""
+
+from alembic import op
+
+
+# pre/post deployment: post
+# revision identifiers, used by Alembic.
+revision = "404b3075d1a4"
+down_revision = "63fa42fd8352"
+branch_labels: tuple[str] | None = None
+depends_on: list[str] | None = None
+
+
+def upgrade() -> None:
+    with op.get_context().autocommit_block():
+        op.create_index(
+            "unique_ix_offer_id_id_at_provider",
+            "price_category",
+            ["offerId", "idAtProvider"],
+            unique=True,
+            postgresql_concurrently=True,
+            if_not_exists=True,
+        )
+
+
+def downgrade() -> None:
+    with op.get_context().autocommit_block():
+        op.drop_index(
+            "unique_ix_offer_id_id_at_provider",
+            table_name="price_category",
+            postgresql_concurrently=True,
+            if_exists=True,
+        )

api/src/pcapi/core/offers/models.py

@@ -1243,6 +1243,11 @@ class PriceCategory(PcObject, Base, Model):
         "PriceCategoryLabel", back_populates="priceCategory"
     )
     stocks: sa_orm.Mapped[list["Stock"]] = relationship("Stock", back_populates="priceCategory", cascade="all")
+    idAtProvider = sa.Column(sa.Text, nullable=True)
+
+    # First step : Create a unique index on offerId/idAtProvider
+    # Next step : (TO DO) Create a unique constraint based on this index
+    sa.Index("unique_ix_offer_id_id_at_provider", offerId, idAtProvider, unique=True)
 
     @property
     def label(self) -> str: