Verify to get Postgresql connection

test_connection.py

import logging
import psycopg2
import sys
import traceback

logging.basicConfig(level=logging.INFO)
log = logging.getLogger(__file__)
log.setLevel(logging.INFO)


def main():
    conn = psycopg2.connect(
        host="127.0.0.1",
        port=5432,
        database="some_db",
        user="postgres",
        password="password"
    )
    cursor = conn.cursor()
    cursor.execute("""SELECT * from roles""")
    rows = cursor.fetchall()
    log.info(rows)


if __name__ == "__main__":
    try:
        main()
        sys.exit()
    except Exception as ex:
        log.error(ex)
        print(traceback.format_exc())
        sys.exit(1)