from sqlalchemy import Column, Integer, String, UniqueConstraint, Date, DateTime
from sqlalchemy.orm import declarative_base
from datetime import datetime, timezone


# Crear la base declarativa
Base = declarative_base()

# Definir el modelo de la tabla
class Registro(Base):
    __tablename__ = 'registros'

    id = Column(Integer, primary_key=True, autoincrement=True)
    organizacion = Column(String, nullable=False)
    ano = Column(Integer, nullable=False)
    mes = Column(Integer, nullable=False)
    dia = Column(Integer, nullable=False)
    fecha_creado = Column(Date, nullable=True)
    cantidad_ioc = Column(Integer, nullable=False)

    # Definir una clave única compuesta
    #__table_args__ = (UniqueConstraint('organizacion', 'ano', 'mes', name='_org_ano_mes_uc'),)


# Se define modelo de usuario
class Usuario(Base):
    __tablename__ = 'usuarios'
    
    id = Column(Integer, primary_key=True, autoincrement=True)
    usuario_sync = Column(String, unique=True, nullable=False)
    organizacion = Column(String, nullable=False)
    creado = Column(DateTime, default=datetime.now(timezone.utc))