blob: 4124b66983f42016a94633001babfcc91df4b69c (
plain)
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
|
import os
import psycopg2
if os.name =='nt':
root = "//WDsentinel/share/CorpCDOs"
elif os.name == 'posix':
root = '/home/share/CorpCDOs'
conn = psycopg2.connect(database="ET",
user="et_user",
password="Serenitas1",
host="debian")
def query_db(conn, sqlstr, **kwargs):
params = kwargs.get('params', None)
one = kwargs.get('one', True)
try:
with conn.cursor() as c:
if params:
c.execute(sqlstr, params)
else:
c.execute(sqlstr)
if one:
return c.fetchone()
else:
return c.fetchall()
except psycopg2.Error:
c.close()
conn.rollback()
return None
|
