from sqlalchemy import Column, Integer, String, ForeignKey, Double
from sqlalchemy.orm import relationship, mapped_column, Mapped

from models import BaseModel


class Service(BaseModel):
    __tablename__ = 'services'
    id: Mapped[int] = mapped_column(Integer, autoincrement=True, primary_key=True, index=True)
    name = Column(String, nullable=False, comment='Название услуги')

    category_id = Column(Integer, ForeignKey('service_categories.id'), nullable=False, comment='ID категории услуги')
    category = relationship('ServiceCategory')

    price = Column(Double, nullable=False, comment='Стоимость услуги')


class ServiceCategory(BaseModel):
    __tablename__ = 'service_categories'
    id = Column(Integer, autoincrement=True, primary_key=True, index=True)
    name = Column(String, nullable=False)