connect_database_url.py

''' Method to connect to a database url(URI) with psycopg2 '''

import os
import psycopg2
import urlparse # import urllib.parse for python 3

result = urlparse.urlparse(os.environ["DATABASE_URL"])
username = result.username
password = result.password
database = result.path[1:]
hostname = result.hostname
connection = psycopg2.connect(
    database = database,
    user = username,
    password = password,
    host = hostname
)