blob: f0cc3fdf8b58ac0cfd94ddfb0911fb6f8ebe11e6 (
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
33
|
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(sqlstr, **kwargs):
c = conn.cursor()
params = kwargs.get('params', None)
one = kwargs.get('one', True)
try:
if params:
c.execute(sqlstr, params)
else:
c.execute(sqlstr)
except psycopg2.Error:
c.close()
conn.rollback()
return None
if one:
result = c.fetchone()
else:
result = c.fetchall()
c.close()
return result
|
