Commit 4f9820b7 authored by Huihui, Jonathan's avatar Huihui, Jonathan
Browse files

pylint/flake8

parent 0a771691

src/common/mixins/postgres.py

+16 −7
Original line number Diff line number Diff line
@@ -4,6 +4,8 @@ 
import psycopg2

from common.env import check_multi_environment as cme

from common.env import check_environment as ce





class PostgresMixin():

    """Serve common connection method for postgres.
@@ -11,14 +13,21 @@ class PostgresMixin(): 
       operating system variable:

           - DATABASE_SEARCH_PATH

    """

    DEFAULT_DB = cme('DATABASE_DB_PG', 'postgres', 'DATABASE_DB')

    DEFAULT_USER = cme('DATABASE_USER_PG', 'postgres', 'DATABASE_USER')

    DEFAULT_PW = cme('DATABASE_PW_PG', 'postgres', 'DATABASE_PW')

    DEFAULT_HOST = cme('DATABASE_HOST_PG', "localhost", 'DATABASE_HOST')

    DEFAULT_PORT = int(cme('DATABASE_PORT_PG', 5432, 'DATABASE_PORT'))

    DEFAULT_DB = cme('DATABASE_DB_PG', 'postgres',

                     'DATABASE_DB', 'postgres')

    DEFAULT_USER = cme('DATABASE_USER_PG', 'postgres',

                       'DATABASE_USER', 'postgres')

    DEFAULT_PW = cme('DATABASE_PW_PG', 'postgres',

                     'DATABASE_PW', 'postgres')

    DEFAULT_HOST = cme('DATABASE_HOST_PG', "localhost",

                       'DATABASE_HOST', "localhost")

    DEFAULT_PORT = int(cme('DATABASE_PORT_PG', 5432,

                           'DATABASE_PORT', 5432))

    DEFAULT_TIMEOUT = ce('DATABASE_TIMEOUT', 60)

    DEFAULT_SCHEMA = cme('DATABASE_SCHEMA_PG', 'public', 'DATABASE_SCHEMA')

    DEFAULT_ENGINE = cme('DATABASE_ENGINE_PG', 'postgresql', 'DATABASE_ENGINE')

    DEFAULT_SCHEMA = cme('DATABASE_SCHEMA_PG', 'public',

                         'DATABASE_SCHEMA', 'public')

    DEFAULT_ENGINE = cme('DATABASE_ENGINE_PG', 'postgresql',

                         'DATABASE_ENGINE', 'postgresql')

    DEFAULT_SEARCH_PATH = ce('DATABASE_SEARCH_PATH', 'public')



    # define a URI string if URI is perferred to connect