53 lines
1.5 KiB
Python
53 lines
1.5 KiB
Python
import psycopg2
|
|
import psycopg2.extras
|
|
from chalicelib.utils.helper import environ
|
|
|
|
PG_CONFIG = {"host": environ["pg_host"],
|
|
"database": environ["pg_dbname"],
|
|
"user": environ["pg_user"],
|
|
"password": environ["pg_password"],
|
|
"port": int(environ["pg_port"])}
|
|
|
|
from psycopg2 import pool
|
|
|
|
try:
|
|
postgreSQL_pool = psycopg2.pool.ThreadedConnectionPool(6, 20, **PG_CONFIG)
|
|
if (postgreSQL_pool):
|
|
print("Connection pool created successfully")
|
|
except (Exception, psycopg2.DatabaseError) as error:
|
|
print("Error while connecting to PostgreSQL", error)
|
|
raise error
|
|
|
|
|
|
# finally:
|
|
# # closing database connection.
|
|
# # use closeall method to close all the active connection if you want to turn of the application
|
|
# if (postgreSQL_pool):
|
|
# postgreSQL_pool.closeall
|
|
# print("PostgreSQL connection pool is closed")
|
|
|
|
class PostgresClient:
|
|
connection = None
|
|
cursor = None
|
|
|
|
def __init__(self):
|
|
self.connection = postgreSQL_pool.getconn()
|
|
|
|
def __enter__(self):
|
|
if self.cursor is None:
|
|
self.cursor = self.connection.cursor(cursor_factory=psycopg2.extras.RealDictCursor)
|
|
return self.cursor
|
|
|
|
def __exit__(self, *args):
|
|
try:
|
|
self.connection.commit()
|
|
self.cursor.close()
|
|
except:
|
|
print("Error while committing/closing PG-connection", error)
|
|
raise error
|
|
finally:
|
|
postgreSQL_pool.putconn(self.connection)
|
|
|
|
|
|
def close():
|
|
pass
|