Create relationship between Price & Product

alembic/versions/20231121_2013_7103fb49908f_add_price_product_relationship.py

@@ -0,0 +1,32 @@
+"""Add Price.product relationship
+
+Revision ID: 7103fb49908f
+Revises: df15f7cacf48
+Create Date: 2023-11-21 20:13:21.521496
+
+"""
+from typing import Sequence, Union
+
+import sqlalchemy as sa
+
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision: str = "7103fb49908f"
+down_revision: Union[str, None] = "df15f7cacf48"
+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.add_column("prices", sa.Column("product_id", sa.Integer(), nullable=True))
+    op.create_foreign_key(None, "prices", "products", ["product_id"], ["id"])
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_constraint(None, "prices", type_="foreignkey")
+    op.drop_column("prices", "product_id")
+    # ### end Alembic commands ###

app/models.py

@@ -41,6 +41,8 @@ class Product(Base):
     off_quantity = Column(Integer)
     off_image_url = Column(String)
 
+    prices: Mapped[list["Price"]] = relationship(back_populates="product")
+
     created = Column(DateTime(timezone=True), server_default=func.now())
     updated = Column(DateTime(timezone=True), onupdate=func.now())
 
@@ -87,6 +89,8 @@ class Price(Base):
     id = Column(Integer, primary_key=True, index=True)
 
     product_code = Column(String, index=True)
+    product_id: Mapped[int] = mapped_column(ForeignKey("products.id"), nullable=True)
+    product: Mapped[Product] = relationship(back_populates="prices")
 
     price = Column(Numeric(precision=10, scale=2))
     currency = Column(CurrencyType)