gistfile1.txt

from sqlalchemy.orm import Session, SessionExtension
from sqlalchemy import Column, String, Integer, Boolean, DateTime

from .database import Base, FuncionarioSessionLocal

class FuncionarioData(Base):
    """
    classe que representa o objeto funcionario no banco de dados
    """
    __tablename__ = "autenticacao_funcionario"
    id = Column(Integer, primary_key=True, index=True)
    matricula = Column(Integer)
    cpf = Column(String)
    nome = Column(String, unique=True, index=True)
    cod_setor = Column(String)
    setor = Column(String)
    funcao = Column(String)
    cod_cargo = Column(Integer)
    cargo = Column(String)
    cod_funcao = Column(Integer)
    cod_vinculo = Column(Integer)
    vinculo = Column(String)
    cod_secretaria = Column(Integer)
    secretaria = Column(String)
    data_admissao = Column(DateTime)
    data_desligamento = Column(DateTime)
    localizacao = Column(String)
    ativo = Column(Boolean)
    admitido_exonerado = Column(Boolean)
    mes_referencia = Column(Integer)
    ano_referencia = Column(Integer)
    data_atualizacao = Column(DateTime)


    def get_db():
        try:
            db = FuncionarioSessionLocal()
            yield db
        finally:
            db.close()

    def fetchall(db: Session):
        """
        Função para retornar todos os funcionários cadastrados no banco
        :return:
        """

        return db.query(FuncionarioData).limit(1000).all()