def salvar_dados(self, dados):
retorno = Gasto.create(descricao_previsto=dados['descricao_previsto'],
valor_previsto=dados['valor_previsto'],
total_realizado=dados['total_realizado'],
saldo=dados['saldo'])
return json.dumps(model_to_dict(retorno))
def atualizar(self, id_gasto, dados):
if 'descricao_previsto' in dados:
query = Gasto.update(descricao_previsto=dados['descricao_previsto']) \
.where(Gasto.id == id_gasto)
query.execute()
if 'valor_previsto' in dados:
query = Gasto.update(valor_previsto=dados['valor_previsto']) \
.where(Gasto.id == id_gasto)
query.execute()
if 'total_realizado' in dados:
query = Gasto.update(total_realizado=dados['total_realizado']) \
.where(Gasto.id == id_gasto)
query.execute()
if 'saldo' in dados:
query = Gasto.update(saldo=dados['saldo']) \
.where(Gasto.id == id_gasto)
query.execute()
def atualizar(self, id_gasto, id_realizado, dados):
gasto = Gasto.select().where(Gasto.id == id_gasto)
if not gasto:
raise Exception('nao existe gasto com este valor')
if 'descricao' in dados:
query = Realizado.update(descricao=dados['descricao']) \
.where((Realizado.id == id_realizado),
(Realizado.gasto_id == id_gasto))
query.execute()
if 'valor' in dados:
query = Realizado.update(valor=dados['valor']) \
.where((Realizado.id == id_realizado),
(Realizado.gasto_id == id_gasto))
query.execute()
def selecionar_gasto(self, id_gasto):
gasto = Gasto.select().where(Gasto.id == id_gasto)
if not gasto:
raise Exception('gasto nao encontrado')
return gasto
class CreateTables:
db.drop_tables([Gasto, Realizado])
db.create_tables([Gasto, Realizado], safe=True)
with db.atomic():
Gasto.insert_many([{
"id": 1,
"descricao_previsto": "Teste",
"valor_previsto": 11.2,
"total_realizado": 8,
"saldo": 4
}, {
"id": 2,
"descricao_previsto": "Teste 2",
"valor_previsto": 14,
"total_realizado": 7.5,
"saldo": 6.5
}, {
"id": 3,
"descricao_previsto": "Teste 3",
"valor_previsto": 20,
"total_realizado": 10,
"saldo": 10
}]).execute()
with db.atomic():
Realizado.insert_many([{
"id": 1,
"descricao": "Teste Realizado 1",
"valor": 100.00,
"gasto": 1
}, {
"id": 2,
"descricao": "Teste Realizado 2",
"valor": 220.22,
"gasto": 1
}, {
"id": 3,
"descricao": "Teste Realizado 3",
"valor": 221.10,
"gasto": 1
}, {
"id": 4,
"descricao": "Teste Realizado 4",
"valor": 560.00,
"gasto": 2
}, {
"id": 5,
"descricao": "Teste Realizado 5",
"valor": 212.00,
"gasto": 3
}, {
"id": 6,
"descricao": "Teste Realizado 6",
"valor": 10.00,
"gasto": 2
}, {
"id": 7,
"descricao": "Teste Realizado 7",
"valor": 22.00,
"gasto": 3
}, {
"id": 8,
"descricao": "Teste Realizado 8",
"valor": 20.10,
"gasto": 1
}, {
"id": 9,
"descricao": "Teste Realizado 9",
"valor": 20.01,
"gasto": 2
}, {
"id": 10,
"descricao": "Teste Realizado 10",
"valor": 29.01,
"gasto": 3
}]).execute()
def delete(self, id_gasto):
realizado = Realizado.delete().where(Realizado.gasto_id == id_gasto)
realizado.execute()
retorno = Gasto.delete().where(Gasto.id == id_gasto)
num = retorno.execute()
return {'excluido': num}
def retornar_lista_gasto(self):
retorno_gasto = []
for i in Gasto.select().order_by(Gasto.id):
retorno_gasto.append(i._data)
return json.dumps(retorno_gasto)
def retornar_gasto(self, instancia_id):
gasto = (Gasto.select().where(Gasto.id == instancia_id).dicts().get())
return json.dumps(gasto)
def retorna_novo_id(self):
retorno = Gasto.create()
return json.dumps(model_to_dict(retorno))