run to the db hills

run-with-db.py

# Run something in a db transaction, handle commit etc.
# f is a function which takes db cursor as a parameter for callback
def run_with_db(f):
    connection = None

    try:
        connection = psycopg2.connect(__get_connection_string(config))
        cursor = connection.cursor()
        rv = f(cursor) # run something in db transaction
        cursor.close()
        return rv
    except psycopg2.DatabaseError, exception:
        print exception
        sys.exit(1)

    finally:
        if connection:
            connection.close()