Files
Sipro-Stocks/database/sipro/models/general.py
2024-06-30 04:53:05 +03:00

57 lines
1.9 KiB
Python

from typing import List
from sqlalchemy import ForeignKey
from sqlalchemy.orm import mapped_column, Mapped, relationship
from .base import BaseSiproModel
from .associations import marketplace_warehouses, warehouse_company_warehouses, warehouse_suppliers
class Company(BaseSiproModel):
__tablename__ = 'companies'
id: Mapped[int] = mapped_column(primary_key=True)
warehouse: Mapped["CompanyWarehouse"] = relationship(
back_populates="company"
)
is_denco: Mapped[bool] = mapped_column()
balance: Mapped[int] = mapped_column()
class CompanyWarehouse(BaseSiproModel):
__tablename__ = 'company_warehouses'
id: Mapped[int] = mapped_column(primary_key=True)
company_id: Mapped[int] = mapped_column(ForeignKey('companies.id'))
company: Mapped["Company"] = relationship(back_populates='warehouse')
class Supplier(BaseSiproModel):
__tablename__ = 'suppliers'
id: Mapped[int] = mapped_column(primary_key=True)
is_pseudo: Mapped[bool] = mapped_column()
class Warehouse(BaseSiproModel):
__tablename__ = 'warehouses'
id: Mapped[int] = mapped_column(primary_key=True)
suppliers: Mapped[List["Supplier"]] = relationship(secondary=warehouse_suppliers)
company_warehouses: Mapped[List["CompanyWarehouse"]] = relationship(secondary=warehouse_company_warehouses)
class Marketplace(BaseSiproModel):
__tablename__ = 'marketplaces'
id: Mapped[int] = mapped_column(primary_key=True)
name: Mapped[str] = mapped_column()
auth_data: Mapped[str] = mapped_column()
sell_mixes: Mapped[bool] = mapped_column()
sell_blocks: Mapped[bool] = mapped_column()
sell_warehouse_products: Mapped[bool] = mapped_column()
sell_from_price: Mapped[bool] = mapped_column()
warehouses: Mapped[List["Warehouse"]] = relationship(secondary=marketplace_warehouses)
company_id: Mapped[int] = mapped_column(ForeignKey('companies.id'))
company: Mapped["Company"] = relationship()