from os import environ as env
from urllib.parse import quote_plus

import psycopg2

config = {
    "host": env.get("DATABASE_HOSTNAME"),
    "port": env.get("DATABASE_PORT", 5432),
    "user": env.get("DATABASE_USER"),
    "password": env.get("DATABASE_PASSWORD"),
    "database": env.get("DATABASE_DB"),
    "sslmode": "require"
}

if __name__ == "__main__":
    try:
        conn = psycopg2.connect(**config)
        conn.close()
        exit(0)
    except (Exception, psycopg2.DatabaseError) as e:
        print(f"Database connection error: {e}")
        exit(1)