75 lines
3.3 KiB
Python
75 lines
3.3 KiB
Python
from typing import TYPE_CHECKING
|
|
|
|
from sqlalchemy import Column, Integer, String, ForeignKey, Sequence
|
|
from sqlalchemy.orm import relationship, Mapped
|
|
from sqlalchemy.testing.schema import mapped_column
|
|
|
|
from models import BaseModel
|
|
|
|
if TYPE_CHECKING:
|
|
from models import Marketplace
|
|
|
|
|
|
class Product(BaseModel):
|
|
__tablename__ = 'products'
|
|
id = Column(Integer, autoincrement=True, primary_key=True, index=True)
|
|
name = Column(String, nullable=False, index=True)
|
|
article = Column(String, nullable=False, default='', server_default='', index=True)
|
|
factory_article = Column(String, nullable=False, default='', server_default='', index=True)
|
|
|
|
client_id = Column(Integer, ForeignKey('clients.id'), nullable=False, comment='ID сделки')
|
|
client = relationship('Client', back_populates='products')
|
|
barcodes = relationship('ProductBarcode', back_populates='product', cascade="all, delete-orphan")
|
|
|
|
barcode_template_id = Column(Integer, ForeignKey('barcode_templates.id'), nullable=True)
|
|
barcode_template = relationship('BarcodeTemplate', lazy='joined')
|
|
|
|
barcode_image = relationship('ProductBarcodeImage', back_populates='product', lazy='joined', uselist=False)
|
|
|
|
# Attributes
|
|
# TODO move to another table
|
|
brand = Column(String, nullable=True, comment='Бренд')
|
|
color = Column(String, nullable=True, comment='Цвет')
|
|
composition = Column(String, nullable=True, comment='Состав')
|
|
size = Column(String, nullable=True, comment='Размер')
|
|
additional_info = Column(String, nullable=True, comment='Дополнительное поле')
|
|
images = relationship('ProductImage',
|
|
back_populates='product',
|
|
lazy='selectin',
|
|
cascade="all, delete-orphan")
|
|
|
|
wildberries_products = relationship('WildberriesProduct',
|
|
back_populates='product',
|
|
cascade="all, delete-orphan",
|
|
uselist=True)
|
|
|
|
ozon_products = relationship('OzonProduct',
|
|
back_populates='product',
|
|
cascade="all, delete-orphan",
|
|
uselist=True)
|
|
|
|
|
|
class ProductImage(BaseModel):
|
|
__tablename__ = 'product_images'
|
|
id = Column(Integer, autoincrement=True, primary_key=True, index=True)
|
|
product_id = Column(Integer, ForeignKey('products.id'), nullable=False)
|
|
product: Mapped["Product"] = relationship(back_populates='images')
|
|
|
|
image_url = Column(String, nullable=False)
|
|
|
|
|
|
class ProductBarcode(BaseModel):
|
|
__tablename__ = 'product_barcodes'
|
|
product_id = Column(Integer, ForeignKey('products.id'), nullable=False, comment='ID товара', primary_key=True)
|
|
product: Mapped["Product"] = relationship(back_populates='barcodes')
|
|
|
|
barcode = Column(String, nullable=False, index=True, comment='ШК товара', primary_key=True)
|
|
|
|
|
|
class ProductBarcodeImage(BaseModel):
|
|
__tablename__ = 'product_barcode_images'
|
|
product_id = Column(Integer, ForeignKey('products.id'), primary_key=True, comment='ID товара')
|
|
product: Mapped["Product"] = relationship(back_populates='barcode_image')
|
|
|
|
filename = Column(String, nullable=False)
|