from typing import Optional
from sqlalchemy import String, PrimaryKeyConstraint
from sqlalchemy.orm import mapped_column, Mapped
from core.database import Base

class State(Base):
    __tablename__ = "states" #GEstados
    __table_args__ = (
        PrimaryKeyConstraint('m3_key', 'description', name='states_pkey'),
        {"schema": "public"}  
    )

    m3_key: Mapped[str] = mapped_column(String(3), nullable=False)
    description: Mapped[str] = mapped_column(String(50), nullable=False)  # valor legal en español
    mex_key: Mapped[Optional[str]] = mapped_column(String(3))
    ame_key: Mapped[Optional[str]] = mapped_column(String(2))

    def __repr__(self):
        return f"<State(m3_key={self.m3_key}, description={self.description}, mex_key={self.mex_key}, ame_key={self.ame_key})>"