61 lines
2.1 KiB
Python
61 lines
2.1 KiB
Python
from typing import TYPE_CHECKING
|
|
|
|
from sqlalchemy import ForeignKey, ForeignKeyConstraint
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
|
|
from models.base import BaseModel
|
|
from models.mixins import PriceMixin
|
|
|
|
if TYPE_CHECKING:
|
|
from models import Deal, Service, Product
|
|
|
|
|
|
class DealProduct(BaseModel):
|
|
__tablename__ = "fulfillment_base_deal_products"
|
|
|
|
deal_id: Mapped[int] = mapped_column(ForeignKey("deals.id"), primary_key=True)
|
|
product_id: Mapped[int] = mapped_column(
|
|
ForeignKey("fulfillment_base_products.id"), primary_key=True
|
|
)
|
|
|
|
quantity: Mapped[int] = mapped_column(default=1)
|
|
comment: Mapped[str] = mapped_column(comment="Комментарий к товару")
|
|
|
|
deal: Mapped["Deal"] = relationship(backref="deal_products")
|
|
product: Mapped["Product"] = relationship()
|
|
|
|
product_services: Mapped[list["DealProductService"]] = relationship(
|
|
back_populates="deal_product",
|
|
primaryjoin="and_(DealProduct.deal_id==DealProductService.deal_id, DealProduct.product_id==DealProductService.product_id)",
|
|
)
|
|
|
|
|
|
class DealProductService(BaseModel, PriceMixin):
|
|
__tablename__ = "fulfillment_base_deal_products_services"
|
|
|
|
deal_id: Mapped[int] = mapped_column(primary_key=True)
|
|
product_id: Mapped[int] = mapped_column(primary_key=True)
|
|
service_id: Mapped[int] = mapped_column(
|
|
ForeignKey("fulfillment_base_services.id"), primary_key=True
|
|
)
|
|
|
|
is_fixed_price: Mapped[bool] = mapped_column(
|
|
default=False, server_default="0", comment="Фиксированная цена"
|
|
)
|
|
|
|
deal_product: Mapped["DealProduct"] = relationship(
|
|
back_populates="product_services",
|
|
primaryjoin="and_(DealProductService.deal_id==DealProduct.deal_id, DealProductService.product_id==DealProduct.product_id)",
|
|
)
|
|
service: Mapped["Service"] = relationship()
|
|
|
|
__table_args__ = (
|
|
ForeignKeyConstraint(
|
|
["deal_id", "product_id"],
|
|
[
|
|
"fulfillment_base_deal_products.deal_id",
|
|
"fulfillment_base_deal_products.product_id",
|
|
],
|
|
),
|
|
)
|