A convenience script to run psql with some variables set based on the esgf.properties

esgf_psql.py

#!/usr/bin/env python

import os
import re

def read_properties(file="/esg/config/esgf.properties"):
    "returns the esgf.properties as a dictionary"
    d = {}
    with open(file) as f:
        for line in f.readlines():
            m = re.match("([^#=]+)\s*=\s*(.*)$", line)
            if m:
                d[m.group(1)] = m.group(2)
    return d

def set_pg_env_vars():
    "set certain env vars based on values in the esgf.properties file"
    mappings = {
        "db.database": "PGDATABASE",
        "db.host": "PGHOST",
        "db.port": "PGPORT",
        "db.user": "PGUSER"
        }
    for key, val in read_properties().items():
        if key in mappings:
            os.environ[mappings[key]] = val

if __name__ == '__main__':
    set_pg_env_vars()
    os.system("psql")