-
Notifications
You must be signed in to change notification settings - Fork 0
/
database.py
45 lines (39 loc) · 1.03 KB
/
database.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
import psycopg2
# Função para criar conexão no banco
def conect_db():
db = psycopg2.connect(host='localhost',
dbname='infosProducao',
user='postgres',
password='postgres',
port='5432')
return db
# Função para criar ou dropar uma tabela no banco
def create_drop_db(sql):
con = conect_db()
cur = con.cursor()
cur.execute(sql)
con.commit()
con.close()
# Função para inserir dados no banco
def insert_delete_db(sql):
con = conect_db()
cur = con.cursor()
try:
cur.execute(sql)
con.commit()
except (Exception, psycopg2.DatabaseError) as error:
print("Error: %s" % error)
con.rollback()
cur.close()
return 1
cur.close()
def consult_db(sql):
con = conect_db()
cur = con.cursor()
cur.execute(sql)
recset = cur.fetchall()
registros = []
for rec in recset:
registros.append(rec)
con.close()
return registros