def registrar_ponto(
colaborador_id: str = Query(..., title="ID colaborador",
description="Identificador único do identificador.")) -> Dict[str, str]:
mes = str(pendulum.now().month)
try:
colaborador = Colaborador.find(colaborador_id=colaborador_id)
if not colaborador:
return gen_mensagem("Não existe colaborador para os parametros informados.")
ultimo_ponto = Ponto.find(colaborador_id=colaborador_id, mes=mes)
ultimo_ponto = ultimo_ponto.pop() if ultimo_ponto else {'colaborador_id': colaborador_id}
ponto = Ponto(**ultimo_ponto)
if ponto.e_entrada() and ponto.registrar_entrada_in():
return gen_mensagem("Ponto registrado com sucesso.", [ponto.dict()])
elif ponto.e_virada() and ponto.registrar_virada_in():
return gen_mensagem("Ponto registrado com sucesso.", [ponto.dict()])
elif ponto.registrar_saida_in():
return gen_mensagem("Ponto registrado com sucesso.", [ponto.dict()])
return gen_mensagem("Problema ao registrar o ponto.")
except:
# TODO EXCEPT
return gen_mensagem("Erro ao registrar ponto")
def test_save_find(web):
col1.save()
col_test = Colaborador.find(cpf=col1.cpf)
colaborador_id = col_test[0]['_id']
col1.remover()
ponto_salvo = Ponto(colaborador_id=colaborador_id)
ponto_salvo.criar_registro_ES()
ponto_rec = Ponto.find(colaborador_id=ponto_salvo.colaborador_id,
data=pendulum.now().format(date_format))
ponto_salvo.remover()
assert ponto_rec, ponto_salvo
from api.models.ColaboradorModel import Colaborador from api.models.PontoModel import Ponto col1 = Colaborador(nome="col_001", endereco="end_col_001", cpf='11111111111') col2 = Colaborador(nome="col_002", endereco="end_col_002", cpf='22222222222') col3 = Colaborador(nome="col_003", endereco="end_col_003", cpf='33333333333') ponto1 = Ponto(colaborador_id="11111111111") ponto2 = Ponto(colaborador_id="22222222222")
def test_remove(web):
col1.remover()
col_recuperado = Colaborador.find(cpf=col1.cpf)
assert str(None), str(col_recuperado)
def test_save_find(web):
col1.save()
col_recuperado = Colaborador.find(cpf=col1.cpf)
col_recuperado = Colaborador(**col_recuperado[0])
assert col1, col_recuperado
def test_instance_erro_values():
with pytest.raises(Exception):
assert Colaborador(nome="Andr", cpf='12131213')
def test_instance_erro_empty():
with pytest.raises(Exception):
assert Colaborador()