db.py

import psycopg2
import os

def init_connection():
    return psycopg2.connect(os.environ["DB_CONNECT"])