Compare commits

...

3 commits

Author SHA1 Message Date
Benjamin Renard 3c2da7f85e Fix ident and code cleaning 2022-01-11 13:06:47 +01:00
Benjamin Renard 8dbb067996 PgDB: add possibility to provide query parameters on doSelect() 2021-07-12 12:16:42 +02:00
Benjamin Renard 198d6021b4 PgDB: fix connect() return and improve logging 2021-07-12 12:16:32 +02:00
4 changed files with 512 additions and 500 deletions

18
PgDB.py
View file

@ -28,11 +28,20 @@ class PgDB(object):
def connect(self):
if self.con == 0:
try:
con = psycopg2.connect("dbname='%s' user='%s' host='%s' password='%s'" % (self.db,self.user,self.host,self.pwd))
con = psycopg2.connect(
dbname=self.db,
user=self.user,
host=self.host,
password=self.pwd
)
self.con = con
except Exception:
logging.fatal('An error occured during Postgresql database connection.', exc_info=1)
logging.fatal(
'An error occured during Postgresql database connection (%s@%s, database=%s).',
self.user, self.host, self.db, exc_info=1
)
sys.exit(1)
return True
def close(self):
if self.con:
@ -61,10 +70,13 @@ class PgDB(object):
self.con.rollback()
return False
def doSelect(self,sql):
def doSelect(self,sql,params):
cursor = self.con.cursor()
try:
if params is None:
cursor.execute(sql)
else:
cursor.execute(sql,params)
results = cursor.fetchall()
return results
except Exception: