def salvar(self, compra):
produto = Produto_dao.findById(compra.id_produto)
if produto != None:
produto.quantidade += compra.quantidade
compra.salvar()
return {'mensagem': 'Compra cadastrado com sucesso'}
else:
return {'mensagem': 'Produto da compra nao existe'}
def findById(id_produto):
return Produto_dao.findById(id_produto)