Simpler db connection pattern

This commit is contained in:
Yohan Boniface 2016-05-13 21:22:05 +02:00
parent 71719e849f
commit 65dd8fad24

View file

@ -10,16 +10,11 @@ import psycopg2.extras
def db_connect(): def db_connect():
try: return psycopg2.connect(
db = psycopg2.connect(dbname="oedb") dbname=os.getenv("DB_NAME", "oedb"),
except: host=os.getenv("DB_HOST", None),
db_host = os.getenv("DB_HOST", "localhost") password=os.getenv("POSTGRES_PASSWORD", None),
db_user = os.getenv("DB_USER", "oedb") user=os.getenv("DB_USER", None))
db_password = os.getenv("POSTGRES_PASSWORD", "")
db = psycopg2.connect(dbname="oedb", host=db_host,
password=db_password, user=db_user)
return db
def standard_headers(resp): def standard_headers(resp):