123

database/__init__.py

@@ -0,0 +1,7 @@
+from sqlalchemy.orm import configure_mappers
+
+from .stocks.models import *
+
+from .sipro.models import *
+
+configure_mappers()

database/sipro/__init__.py

@@ -0,0 +1 @@
+from .models import *

database/sipro/models/__init__.py

@@ -0,0 +1,3 @@
+from .base import BaseSiproModel
+from .general import *
+from .products import *

database/sipro/models/associations.py

@@ -0,0 +1,25 @@
+from .base import BaseSiproModel
+
+from sqlalchemy import Table, Column, ForeignKey
+
+marketplace_warehouses = Table(
+    'marketplace_warehouses',
+    BaseSiproModel.metadata,
+    Column('marketplace_id', ForeignKey('marketplaces.id')),
+    Column('warehouse_id', ForeignKey('warehouses.id'))
+)
+
+warehouse_suppliers = Table(
+    'warehouse_suppliers',
+    BaseSiproModel.metadata,
+    Column('warehouse_id', ForeignKey('warehouses.id')),
+    Column('supplier_id', ForeignKey('suppliers.id'))
+)
+
+warehouse_company_warehouses = Table(
+    'warehouse_company_warehouses',
+    BaseSiproModel.metadata,
+    Column('warehouse_id', ForeignKey('warehouses.id')),
+    Column('company_warehouse_id', ForeignKey('company_warehouses.id')),
+
+)

database/sipro/models/base.py

@@ -0,0 +1,10 @@
+from sqlalchemy import MetaData
+from sqlalchemy.ext.asyncio import AsyncAttrs
+from sqlalchemy.orm import DeclarativeBase
+
+
+class BaseSiproModel(AsyncAttrs, DeclarativeBase):
+    metadata = MetaData(schema='public')
+    __table_args__ = {
+        'schema': 'public',
+    }

database/sipro/models/general.py

@@ -0,0 +1,56 @@
+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()

database/sipro/models/products.py

@@ -0,0 +1,60 @@
+from sqlalchemy import ForeignKey
+from sqlalchemy.orm import Mapped, mapped_column, relationship
+
+from .base import BaseSiproModel
+from .general import Marketplace
+
+
+class Product(BaseSiproModel):
+    __tablename__ = 'products'
+    id: Mapped[int] = mapped_column(primary_key=True)
+    denco_article: Mapped[int] = mapped_column(index=True)
+    article: Mapped[str] = mapped_column(index=True)
+
+
+class MarketplaceProduct(BaseSiproModel):
+    __tablename__ = 'marketplace_products'
+    id: Mapped[int] = mapped_column(primary_key=True)
+    marketplace_id: Mapped[int] = mapped_column(ForeignKey('marketplaces.id'))
+    marketplace: Mapped["Marketplace"] = relationship()
+    mp_price_bought: Mapped[int] = mapped_column()
+    price_recommended: Mapped[int] = mapped_column()
+    is_archived: Mapped[bool] = mapped_column()
+
+    product_id: Mapped[int] = mapped_column(ForeignKey("products.id"))
+    product: Mapped["Product"] = relationship()
+
+
+class SupplierProduct(BaseSiproModel):
+    __tablename__ = 'supplier_products'
+    id: Mapped[int] = mapped_column(primary_key=True)
+    supplier_stock: Mapped[int] = mapped_column()
+    supplier_id: Mapped[int] = mapped_column()
+
+    product_id: Mapped[int] = mapped_column(ForeignKey("products.id"))
+    product: Mapped["Product"] = relationship()
+
+    in_block: Mapped[int] = mapped_column()
+
+
+class CompanyWarehouseProduct(BaseSiproModel):
+    __tablename__ = 'company_warehouse_products'
+    id: Mapped[int] = mapped_column(primary_key=True)
+    is_sold: Mapped[bool] = mapped_column()
+    company_warehouse_id: Mapped[int] = mapped_column()
+
+    product_id: Mapped[int] = mapped_column(ForeignKey("products.id"))
+    product: Mapped["Product"] = relationship()
+
+
+class ProductRelation(BaseSiproModel):
+    __tablename__ = "products_relations"
+    id: Mapped[int] = mapped_column(primary_key=True, comment="ID связи в БД")
+
+    master_product_id: Mapped[int] = mapped_column(ForeignKey("products.id"))
+    master_product: Mapped["Product"] = relationship(foreign_keys=[master_product_id])
+
+    slave_product_id: Mapped[int] = mapped_column(ForeignKey("products.id"))
+    slave_product: Mapped["Product"] = relationship(foreign_keys=[slave_product_id])
+
+    relation_type: Mapped[int] = mapped_column(comment="Тип связи (ENUM)")

database/stocks/models/__init__.py

@@ -0,0 +1,2 @@
+from .base import BaseStocksModel
+from .general import *

database/stocks/models/base.py

@@ -0,0 +1,10 @@
+from sqlalchemy import MetaData
+from sqlalchemy.ext.asyncio import AsyncAttrs
+from sqlalchemy.orm import DeclarativeBase
+
+
+class BaseStocksModel(AsyncAttrs, DeclarativeBase):
+    metadata = MetaData(schema='stocks')
+    __table_args__ = {
+        'schema': 'stocks',
+    }

database/stocks/models/general.py

@@ -0,0 +1,10 @@
+from sqlalchemy.orm import mapped_column, Mapped
+
+from .base import BaseStocksModel
+
+
+class DailyStock(BaseStocksModel):
+    __tablename__ = 'daily_stocks'
+    product_id: Mapped[int] = mapped_column(primary_key=True)
+    sold_today: Mapped[int] = mapped_column()
+    test: Mapped[bool] = mapped_column()