def buscar(self, numero):
with connect(self._dados_con) as conn:
cur = conn.cursor()
cur.execute('SELECT * FROM conta WHERE codigo = %s', [numero])
linha = cur.fetchall()
conta = Conta(linha[0][1], linha[0][2], linha[0][3], linha[0][4])
conta.codigo = linha[0][0]
daocl = clienteDao()
busc = daocl.buscar(int(linha[0][5]))
conta.addCliente(busc)
return conta
conn.commit()
cur.close()
def listar(self):
with connect(self._dados_con) as conn:
cur = conn.cursor()
lista_conta = []
cur.execute('SELECT * FROM conta')
for linha in cur.fetchall():
conta = Conta(linha[1], linha[2], linha[3], linha[4])
conta.codigo = linha[0]
daocl = clienteDao()
busc = daocl.buscar(int(linha[5]))
conta.addCliente(busc)
lista_conta.append(conta)
return lista_conta
conn.commit()
cur.close()
def salvarConta():
numero = request.form["numero"]
agencia = request.form["agencia"]
saldo = request.form["saldo"]
tipo = request.form["tipo"]
cliente = request.form["cliente"]
daocliente = clienteDao()
c = daocliente.buscar(int(cliente))
conta = Conta(numero, agencia, saldo, tipo)
conta.addCliente(c)
daoconta = contaDao()
if (request.values.has_key("codigo") == True):
cod = request.form["codigo"]
conta.codigo = int(cod)
daoconta.salvar(conta)
return redirect('/conta/listar')
