File size: 567 Bytes
7a88b43
 
 
 
 
 
 
 
 
a8c0833
 
 
 
7a88b43
 
 
 
a8c0833
 
 
 
 
7a88b43
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24

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)