def test_caso_todas_as_informacoes_esteja_ok_um_agendamento_deve_ser_criado(
self):
"""
Caso todos as informações estejam ok um agendamento deve ser criado.
"""
tags = [
'time-0|1',
'time-1|1,0',
'time-2|2,1,0',
]
Agendamento(pensador=self.pensador, times_tag=tags).save()
assert_equals(Agendamento.objects(pensador=self.pensador).count(), 1)
Agendamento.objects(pensador=self.pensador).delete()
def populate_agendamento(file, reset=False):
csv_file = read_csv(file, skip=None)
for c in csv_file:
if len(c) > 11:
email = c[11]
else:
email = None
if c[6] == '-':
return
s = Servico.query.get(c[6])
if not s:
s = Servico(cod_servico=c[6], servico=c[7])
db.session.add(s)
if c[10] != '0':
if c[10][2] == '0':
c[10] = replacer(c[10], '9', 2)
a = Agendamento(cod_agendamento=c[2],
nome_requerente=c[8],
data_agendamento=c[4],
data_solicitacao_agendamento=c[5],
celular=c[10],
telefone_fixo=c[9],
email=email,
cod_aps="{:08d}".format(int(c[0])))
try:
s.agendamentos.append(a)
db.session.commit()
except (sqlalchemy.orm.exc.FlushError,
sqlalchemy.exc.IntegrityError) as e:
db.session.rollback()
print("repeated {0}".format(c[8]))
pass
catalogurl = 'http://{0}:7000'.format(ES_HOST)
agendamentos = db.session.query(Agendamento).all()
index_docs(catalogurl, agendamentoindex, 'agendamento', 3, agendamentos,
reset)
