#! /usr/bin/env python

import os, sys

print(os.environ['WL_PG_URI'])

url = os.environ['WL_PG_URI'].split('//')[1]

userpass, rest = url.split('@')

hostport, dbname = rest.split('/')

host, port = hostport.split(":")

u, p = userpass.split(':')

passstring = "PGPASSWORD=" + p + " " if p else ""

command = passstring + "psql -h " + host + " -p " + port + " -U " + u + " -d " + dbname + " " + " ".join(["'" + a + "'" for a in sys.argv[1:]])
print(command)


os.system(command)