forked from cartsnitch/api
Coupon Carl
b7e6f637a7
Consolidate API gateway service into monorepo. Squashed from https://github.com/cartsnitch/api main (89bacb1). Co-Authored-By: Paperclip <noreply@paperclip.ing>
53 lines
2.1 KiB
Python
53 lines
2.1 KiB
Python
"""Store and StoreLocation models."""
|
|
|
|
import uuid
|
|
from typing import TYPE_CHECKING
|
|
|
|
from sqlalchemy import Float, ForeignKey, String
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
|
|
from cartsnitch_api.constants import StoreSlug
|
|
from cartsnitch_api.models.base import Base, TimestampMixin, UUIDPrimaryKeyMixin
|
|
|
|
if TYPE_CHECKING:
|
|
from cartsnitch_api.models.coupon import Coupon
|
|
from cartsnitch_api.models.price import PriceHistory
|
|
from cartsnitch_api.models.purchase import Purchase
|
|
from cartsnitch_api.models.user import UserStoreAccount
|
|
|
|
|
|
class Store(UUIDPrimaryKeyMixin, TimestampMixin, Base):
|
|
"""Supported retailer."""
|
|
|
|
__tablename__ = "stores"
|
|
|
|
name: Mapped[str] = mapped_column(String(100), nullable=False)
|
|
slug: Mapped[StoreSlug] = mapped_column(String(20), nullable=False, unique=True)
|
|
logo_url: Mapped[str | None] = mapped_column(String(500))
|
|
website_url: Mapped[str | None] = mapped_column(String(500))
|
|
|
|
# Relationships
|
|
locations: Mapped[list["StoreLocation"]] = relationship(back_populates="store")
|
|
purchases: Mapped[list["Purchase"]] = relationship(back_populates="store")
|
|
user_accounts: Mapped[list["UserStoreAccount"]] = relationship(back_populates="store")
|
|
price_histories: Mapped[list["PriceHistory"]] = relationship(back_populates="store")
|
|
coupons: Mapped[list["Coupon"]] = relationship(back_populates="store")
|
|
|
|
|
|
class StoreLocation(UUIDPrimaryKeyMixin, TimestampMixin, Base):
|
|
"""Physical store location."""
|
|
|
|
__tablename__ = "store_locations"
|
|
|
|
store_id: Mapped[uuid.UUID] = mapped_column(ForeignKey("stores.id"), nullable=False)
|
|
address: Mapped[str] = mapped_column(String(300), nullable=False)
|
|
city: Mapped[str] = mapped_column(String(100), nullable=False)
|
|
state: Mapped[str] = mapped_column(String(2), nullable=False)
|
|
zip: Mapped[str] = mapped_column(String(10), nullable=False)
|
|
lat: Mapped[float | None] = mapped_column(Float)
|
|
lng: Mapped[float | None] = mapped_column(Float)
|
|
|
|
# Relationships
|
|
store: Mapped["Store"] = relationship(back_populates="locations")
|
|
purchases: Mapped[list["Purchase"]] = relationship(back_populates="store_location")
|