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")