db.py

import os
import psycopg2
from dotenv import load_dotenv

load_dotenv()

CONN = None
TESTING_MODE = False
# def connect():
# global CONN

DATABASE_URL = os.getenv("DATABASE_URL")
try:
    CONN = psycopg2.connect(DATABASE_URL, sslmode="allow")
    print("PostgreSQL connection successful")
except (Exception, psycopg2.DatabaseError) as error:
    print(error)


def query(sql, args=()):
    """query the database and get back rows selected/modified"""
    cur = CONN.cursor()
    try:
        cur.execute(sql, args)
    except Exception as e:
        print(type(e).__name__, e)
        CONN.rollback()
        raise e
    if cur.description is None:
        rows = []
    else:
        rows = cur.fetchall()
    if not TESTING_MODE:
        CONN.commit()
    cur.close()
    return rows