ody_migr_db_handler.PGSession Class Reference

Public Member Functions
def	__init__ (self)
def	__enter__ (self)
def	__exit__ (self, exception_type, exception_value, _traceback)

Public Attributes
	conn

Detailed Description

Context manager for PG database connection

TODO: investigate connection pooling:
http://initd.org/psycopg/docs/pool.html

Definition at line 27 of file ody_migr_db_handler.py.

Constructor & Destructor Documentation

__init__()

def ody_migr_db_handler.PGSession.__init__

(

self

)

initialize database manager

Definition at line 34 of file ody_migr_db_handler.py.

    def __init__(self):
        """initialize database manager"""
        self.conn = None

Member Function Documentation

__enter__()

def ody_migr_db_handler.PGSession.__enter__

(

self

)

open database connection

Definition at line 38 of file ody_migr_db_handler.py.

    def __enter__(self):
        """open database connection"""
        try:
            self.conn = psycopg2.connect(("host={} dbname={} "
                                          "user={} password={}")
                                         .format(
                                             PG_DB_HOST,
                                             PG_DB_NAME,
                                             PG_DB_USER,
                                             PG_DB_PWD
            ))
            return self.conn

        except psycopg2.OperationalError as err:
            raise SystemExit("Unable to connect!{}".format(err))

__exit__()

def ody_migr_db_handler.PGSession.__exit__	(		self,
			exception_type,
			exception_value,
			_traceback
	)

close database connection

Definition at line 54 of file ody_migr_db_handler.py.

    def __exit__(self, exception_type, exception_value, _traceback):
        """close database connection"""
        if exception_type:
            LOGGER.error(exception_value)
        if self.conn is not None:
            self.conn.close()


# DATABASE UTILITIES

---

The documentation for this class was generated from the following file:

• app/tools/bin/ody_migr_db_handler.py