from sqlalchemy import Column, INTEGER, VARCHAR, DATE, ForeignKey
from sqlalchemy.orm import relationship

from database import ModelBase
from persona.models import Persona


class Agente(ModelBase):
    __tablename__ = 'tb_agentes'

    id = Column('idAgente', INTEGER(), primary_key=True, nullable=False)
    legajo = Column('Legajo', VARCHAR(length=10))
    observaciones = Column('Observaciones', VARCHAR(length=250))

    causa_baja = Column('CausaBaja', INTEGER())
    fecha_alta = Column('FechaAlta', DATE(), nullable=False)
    fecha_baja = Column('FechaBaja', DATE())

    # Relaciones
    persona_id = Column('Persona', INTEGER(), ForeignKey('tb_Personas.idPersona'), nullable=False)
    # persona = relationship(Persona, primaryjoin="Agente.Persona == Persona.idPersona", lazy='dynamic')

    def __str__(self):
        return f"{self.persona_id}"

    def __repr__(self):
        return f"<Agente: {self}>"