class Contato(object):
def __init__(self):
self.con = Conexao()
def criar_contato(self):
self.cursor = self.con.cursor()
try:
self.nome = raw_input("Entre com um nome: ")
self.endereco = raw_input("Entre com o endereco: ")
self.cursor.execute("INSERT INTO contato (nome, endereco) VALUES ('{0}', '{1}')".format(self.nome, self.endereco))
self.con.salvar_no_banco() # salvar as alteracoes no banco
print "Dados {0} || {1} inseridos com sucesso".format(self.nome, self.endereco)
except MySQLdb.IntegrityError:
logging.warn("Falha ao tentar inserir os valores: {0}, {1}".format(self.nome, self.endereco))
finally:
self.cursor.close()
def listar_contatos(self):
self.cursor = self.con.cursor()
print "Buscando dados.:"
dados = self.cursor.execute("SELECT * FROM contato")
if dados:
for linha in self.cursor.fetchall():
print "ID: {0}\nNome: {1}\nEndereco: {2}\n\n".format(linha[0], linha[1], linha[2])
else:
print "Nenhum dado encontrado"
self.cursor.close()
## TODO Implementar os metodos exluir e editar, e fazer verificacao de entrada.
def fechar_agenda(self):
self.con.fechar_conexao()
def editar(self, nome, matricula):
try:
conexao = Conexao().conexaoDatabase()
cursor = conexao.cursor()
editar = f"UPDATE aluno SET matricula='{matricula}' WHERE nome='{nome}';"
cursor.execute(editar)
conexao.commit()
return f"O aluno {nome} teve seus dados atualizados"
except (Exception, psycopg2.Error) as error:
print("\nFalha ao atualizar o registro\n", error)
finally:
if (conexao):
cursor.close()
conexao.close()
def inserir(self, cpf, nome, email):
try:
conexao = Conexao().get_connection()
cursor = conexao.cursor()
inserir = """insert into pessoas (cpf, nome, email) values ('{0}','{1}','{2}')""".format(
cpf, nome, email)
cursor.execute(inserir)
conexao.commit()
print("Cadastro inserido com sucesso!")
except (Exception, psycopg2.DatabaseError) as error:
print("Oppss! Algum erro aconteceu :/", error)
finally:
if (conexao):
cursor.close()
conexao.close()
def remover(self, matricula):
try:
conexao = Conexao().conexaoDatabase()
cursor = conexao.cursor()
remover = f"DELETE FROM aluno WHERE matricula='{matricula}';"
cursor.execute(remover)
conexao.commit()
return f"O aluno com a matricula {matricula} foi removido"
except (Exception, psycopg2.Error) as error:
print("\nFalha ao remover o registro\n", error)
finally:
if (conexao):
cursor.close()
conexao.close()
print("A conexão com PostgreSQL foi encerrada\n")
def cadastrar(self, nome, matricula):
try:
conexao = Conexao().conexaoDatabase()
cursor = conexao.cursor()
#cadastrar os registros na tabela
cadastrar = f"INSERT INTO aluno (nome, matricula) VALUES ('{nome}', '{matricula}');"
cursor.execute(cadastrar)
#comitar os registros no database
conexao.commit()
return f"Aluno {nome} e com a matricula {matricula} foi cadastrado com sucesso no banco de dados"
#exceção para se ocorrer um erro ao cadastrar os registros na tabela
except (Exception, psycopg2.DatabaseError) as error:
print("\nFalha ao cadastrar o registro\n", error)
#se tudo ocorrer corretamente, a conexão com o database é encerrada
finally:
if conexao:
cursor.close()
conexao.close()
print("A conexão com PostgreSQL foi encerrada\n")
def busca_cpf(self, cpf):
try:
conexao = Conexao().get_connection()
cursor = conexao.cursor()
busca = "select * from pessoas where cpf = '{0}'".format(cpf)
cursor.execute(busca)
pessoas = cursor.fetchall()
if len(pessoas) >= 1:
return True
else:
return False
except (Exception, psycopg2.DatabaseError) as error:
print("Oppss! Algum erro aconteceu :/", error)
finally:
if (conexao):
cursor.close()
conexao.close()
def busca_pessoa_cpf(self, cpf):
try:
conexao = Conexao().get_connection()
cursor = conexao.cursor()
busca = "select * from pessoas where cpf like '%{0}%' and status = 1".format(
cpf)
cursor.execute(busca)
pessoas = cursor.fetchall()
for i in pessoas:
print("Nome: ", i[1])
print("CPF: ", i[0])
print("E-mail: ", i[2])
except (Exception, psycopg2.DatabaseError) as error:
print("Oppss! Algum erro aconteceu :/", error)
finally:
if (conexao):
cursor.close()
conexao.close()
def consultar(self):
try:
conexao = Conexao().conexaoDatabase()
cursor = conexao.cursor()
select = f"SELECT * FROM aluno;"
cursor.execute(select)
aluno = cursor.fetchall()
if len(aluno):
return aluno
else:
return "Nenhum registro foi encontrado"
except (Exception, psycopg2.Error) as error:
print("\nFalha ao consultar o registro\n", error)
finally:
if (conexao):
cursor.close()
conexao.close()
print("A conexão com PostgreSQL foi encerrada\n")
