blob: 07d44a29a7c9e98e97b8c79ad6adc882675cab63 (
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
32
|
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:
c = conn.cursor()
if params:
c.execute(sqlstr, params)
else:
c.execute(sqlstr)
if one:
return c.fetchone()
else:
return c.fetchall()
c.close()
except psycopg2.Error:
c.close()
conn.rollback()
return None
|
