54 lines
1.9 KiB
Python
54 lines
1.9 KiB
Python
import datetime
|
|
from enum import StrEnum
|
|
from typing import TYPE_CHECKING
|
|
|
|
from sqlalchemy import FLOAT, INTEGER, DateTime, Enum, ForeignKey
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
|
|
from db.base import Base
|
|
|
|
if TYPE_CHECKING:
|
|
from db.models import User
|
|
|
|
|
|
class OrderStatus(StrEnum):
|
|
PENDING = "pending"
|
|
PAID = "paid"
|
|
DISCARDED = "discarded"
|
|
|
|
|
|
class OrderAddon(Base):
|
|
__tablename__ = "order_addons"
|
|
|
|
id: Mapped[int] = mapped_column(
|
|
INTEGER, nullable=False, unique=True, autoincrement=True, primary_key=True
|
|
)
|
|
order_id: Mapped[int] = mapped_column(ForeignKey("orders.id"), nullable=False)
|
|
addon_id: Mapped[str] = mapped_column(ForeignKey("addons.id"), nullable=False)
|
|
|
|
order: Mapped["Order"] = relationship("Order", back_populates="addons", lazy="selectin")
|
|
|
|
|
|
class Order(Base):
|
|
__tablename__ = "orders"
|
|
|
|
id: Mapped[int] = mapped_column(
|
|
INTEGER, nullable=False, autoincrement=True, unique=True, primary_key=True
|
|
)
|
|
user_id: Mapped[int] = mapped_column(ForeignKey("users.id"), nullable=False)
|
|
|
|
devices: Mapped[int] = mapped_column(INTEGER, nullable=False)
|
|
duration_days: Mapped[int] = mapped_column(INTEGER, nullable=False)
|
|
total_amount: Mapped[float] = mapped_column(FLOAT, nullable=False)
|
|
balance_amount: Mapped[float] = mapped_column(FLOAT, nullable=False, default=0)
|
|
status: Mapped[OrderStatus] = mapped_column(
|
|
Enum(OrderStatus, name="orderstatus"), nullable=False, default=OrderStatus.PENDING
|
|
)
|
|
applies_at: Mapped[datetime.datetime | None] = mapped_column(DateTime(True), nullable=True)
|
|
applied_at: Mapped[datetime.datetime | None] = mapped_column(DateTime(True), nullable=True)
|
|
|
|
user: Mapped["User"] = relationship("User", back_populates="orders", lazy="selectin")
|
|
addons: Mapped[list["OrderAddon"]] = relationship(
|
|
"OrderAddon", back_populates="order", lazy="selectin"
|
|
)
|