129 lines
5.0 KiB
Python
129 lines
5.0 KiB
Python
from datetime import datetime
|
|
from enum import IntEnum, unique
|
|
from typing import Optional, TYPE_CHECKING
|
|
|
|
from sqlalchemy import Column, Integer, String, DateTime, ForeignKey, Boolean
|
|
from sqlalchemy.orm import relationship, backref, Mapped, mapped_column
|
|
|
|
from models.base import BaseModel
|
|
from .marketplace import BaseMarketplace
|
|
from .board import Board
|
|
from .status import DealStatus
|
|
from .shipping import Pallet, Box
|
|
from .shipping_warehouse import ShippingWarehouse
|
|
|
|
if TYPE_CHECKING:
|
|
from . import (
|
|
DealBillRequest,
|
|
ServicePriceCategory,
|
|
DealGroup,
|
|
User,
|
|
)
|
|
|
|
|
|
@unique
|
|
class DealStatusEnum(IntEnum):
|
|
CREATED = 0
|
|
AWAITING_ACCEPTANCE = 1
|
|
READY_FOR_WORK = 2
|
|
PACKAGING = 3
|
|
AWAITING_SHIPMENT = 4
|
|
IN_DELIVERY = 5
|
|
AWAITING_PAYMENT = 6
|
|
COMPLETED = 7
|
|
CANCELLED = 8
|
|
|
|
|
|
class DealPriceCategory(BaseModel):
|
|
__tablename__ = 'deal_price_category'
|
|
deal_id: Mapped[int] = mapped_column(ForeignKey('deals.id'), primary_key=True, unique=True)
|
|
category_id: Mapped[int] = mapped_column(ForeignKey('service_price_category.id'), primary_key=True)
|
|
|
|
|
|
class Deal(BaseModel):
|
|
__tablename__ = 'deals'
|
|
id = Column(Integer, autoincrement=True, primary_key=True, index=True)
|
|
name = Column(String, nullable=False, comment='Название сделки')
|
|
created_at = Column(DateTime, nullable=False, comment='Дата создания')
|
|
|
|
client_id = Column(Integer, ForeignKey('clients.id', ondelete='CASCADE'), nullable=False, comment='ID клиента')
|
|
client = relationship('Client', backref=backref('deals', cascade="all, delete-orphan"))
|
|
|
|
status_history = relationship('DealStatusHistory', back_populates='deal', cascade="all, delete-orphan")
|
|
|
|
is_deleted = Column(Boolean, nullable=False, server_default='0', default=False, comment='Удалена')
|
|
is_completed = Column(Boolean, nullable=False, server_default='0', default=False, comment='Завершена')
|
|
is_locked: Mapped[bool] = mapped_column(default=False, server_default='0')
|
|
is_accounted: Mapped[bool] = mapped_column(default=True, server_default='1')
|
|
|
|
current_status_id: Mapped[int] = mapped_column(
|
|
ForeignKey("deal_statuses.id"),
|
|
nullable=False,
|
|
comment='Текущий статус',
|
|
)
|
|
status: Mapped["DealStatus"] = relationship(lazy="selectin")
|
|
|
|
|
|
shipping_warehouse_id: Mapped[int] = mapped_column(ForeignKey('shipping_warehouses.id'), nullable=True)
|
|
shipping_warehouse: Mapped["ShippingWarehouse"] = relationship()
|
|
|
|
base_marketplace_key: Mapped[str] = mapped_column(ForeignKey("base_marketplaces.key"), nullable=True)
|
|
base_marketplace: Mapped["BaseMarketplace"] = relationship(lazy="joined")
|
|
|
|
delivery_date: Mapped[Optional[datetime]] = mapped_column(nullable=True)
|
|
receiving_slot_date: Mapped[Optional[datetime]] = mapped_column(nullable=True)
|
|
|
|
services = relationship(
|
|
'DealService',
|
|
back_populates='deal',
|
|
cascade="all, delete-orphan",
|
|
order_by="desc(DealService.service_id)"
|
|
)
|
|
|
|
products = relationship(
|
|
'DealProduct',
|
|
back_populates='deal',
|
|
cascade="all, delete-orphan",
|
|
order_by="desc(DealProduct.product_id)"
|
|
)
|
|
|
|
board_id: Mapped[int] = mapped_column(ForeignKey('boards.id'), nullable=True, server_default='1')
|
|
board: Mapped[Board] = relationship(
|
|
"Board",
|
|
back_populates="deals",
|
|
)
|
|
|
|
# TODO remake with sequence
|
|
lexorank = Column(String, nullable=False, comment='Lexorank', index=True)
|
|
|
|
comment = Column(String, nullable=False, server_default='', comment='Коментарий к заданию')
|
|
bill_request: Mapped[Optional['DealBillRequest']] = relationship(back_populates='deal', lazy='joined')
|
|
category: Mapped[Optional["ServicePriceCategory"]] = relationship('ServicePriceCategory',
|
|
secondary=DealPriceCategory.__table__,
|
|
lazy='joined')
|
|
group: Mapped[Optional["DealGroup"]] = relationship(
|
|
'DealGroup',
|
|
secondary='deal_relations',
|
|
lazy='joined',
|
|
back_populates='deals'
|
|
)
|
|
|
|
manager_id: Mapped[int] = mapped_column(ForeignKey('users.id'), nullable=True)
|
|
manager: Mapped[Optional["User"]] = relationship(back_populates='managed_deals', lazy='joined')
|
|
|
|
pallets: Mapped[list[Pallet]] = relationship(back_populates='deal', lazy='selectin')
|
|
boxes: Mapped[list[Box]] = relationship(back_populates='deal', lazy='selectin')
|
|
|
|
employees: Mapped[list['DealEmployees']] = relationship(back_populates='deal', lazy='selectin')
|
|
|
|
|
|
class DealEmployees(BaseModel):
|
|
__tablename__ = 'deal_employees'
|
|
user_id: Mapped[int] = mapped_column(ForeignKey('users.id'), primary_key=True)
|
|
user: Mapped['User'] = relationship('User', back_populates='deals', lazy='selectin')
|
|
|
|
deal_id: Mapped[int] = mapped_column(ForeignKey('deals.id'), primary_key=True)
|
|
deal: Mapped[Deal] = relationship('Deal', back_populates='employees', lazy='selectin')
|
|
|
|
created_at: Mapped[datetime] = mapped_column()
|