from sqlalchemy import Column, Integer, String, ForeignKey, Sequence
from sqlalchemy.orm import relationship

from models import BaseModel


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, 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')

    # 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='Дополнительное поле')


class ProductBarcode(BaseModel):
    __tablename__ = 'product_barcodes'
    product_id = Column(Integer, ForeignKey('products.id'), nullable=False, comment='ID товара', primary_key=True)
    product = relationship('Product', back_populates='barcodes')

    barcode = Column(String, nullable=False, index=True, comment='ШК товара', primary_key=True)