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)

    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='Дополнительное поле')
    images = relationship('ProductImage',
                          back_populates='product',
                          lazy='selectin',
                          cascade="all, delete-orphan")



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)