database_cleaner.py

import dj_database_url
import psycopg2
import sys

D = dj_database_url.config()

#set databae variables
print """export PGUSER=%s
export PGPASSWORD=%s
export PGHOST=%s
export PGDATABASE=%s
"""%(D['USER'],
     D['PASSWORD'],
     D['HOST'],
     D['NAME'])


# Drop all tables from a given database
if False:
	try:
	    conn = psycopg2.connect("""dbname='%s'
		                         user='%s'
		                     password='%s'
		                         host='%s'"""%(D['NAME'],
		                                       D['USER'],
		                                       D['PASSWORD'],
		                                       D['HOST']
		                                       )
		                    )
	    conn.set_isolation_level(0)
	except:
	    print "Unable to connect to the database."

	cur = conn.cursor()

	try:
	    cur.execute("SELECT table_schema,table_name FROM information_schema.tables WHERE table_schema = 'public' ORDER BY table_schema,table_name")
	    rows = cur.fetchall()
	    for row in rows:
		print "dropping table: ", row[1]
		cur.execute("drop table " + row[1] + " cascade")
	    cur.close()
	    conn.close()
	except:
	    print "Error: ", sys.exc_info()[1]