connect.py

import psycopg2
from config import config

def connect():
    """
    This method is just to check connections with the database.
    """
    conn = None
    try:
        # read connection params
        params = config()

        # connect to the PostgreSQL Server
        print('Connecting to the pdSQL database...')
        conn = psycopg2.connect(**params)

        # create a cursor
        cur = conn.cursor()

        # execute a statement
        print('PostgreSQL database version:')
        cur.execute('SELECT version()')

        # display the PostgreSQL database server versiohn
        db_version = cur.fetchone()
        print(db_version)

        # close the communication with the PostgreSQL
        cur.close()

    except (Exception, psycopg2.DatabaseError) as error:
        print(error)
    finally:
        if conn is not None:
            conn.close()
            print('Database connection closed.')

if __name__ == '__main__':
    connect()