drop_table.py

Go to the documentation of this file.

#!/bin/env python

import sys, getopt, psycopg2

Usage = """
python drop_table.py [options] <database name> <table_name> 
options:
    -h <host>
    -p <port>
    -U <user>
    -w <password>
    -d <detector>
    
"""

host = None
port = None
user = None
password = None
detector = None

dbcon = []

opts, args = getopt.getopt(sys.argv[1:], 'h:U:w:p:d:')

if len(args) < 2 or args[0] == 'help':
    print Usage
    sys.exit(0)

for opt, val in opts:
    if opt == '-h':         dbcon.append("host=%s" % (val,))
    elif opt == '-p':       dbcon.append("port=%s" % (int(val),))
    elif opt == '-U':       dbcon.append("user=%s" % (val,))
    elif opt == '-w':       dbcon.append("password=%s" % (val,))    
    elif opt == '-d':       detector=val
        
dbcon.append("dbname=%s" % (args[0],))
tname = args[1]

if detector == None:
    print Usage
    sys.exit(0)
    
try:
#    conn = psycopg2.connect(dbcon)
    conn = psycopg2.connect("dbname=nova_dev host=ifdbdev.fnal.gov port=5433")
except:
    print "I am unable to connect to the database"
    exit(0)
    
cur = conn.cursor()

cur.execute("set role nova_admin")

SQL = "drop table %s.%s_tag_snapshot; drop table %s.%s_tag cascade; drop table %s.%s_snapshot_data; drop table %s.%s_update; drop table %s.%s_snapshot cascade;" % (detector,tname,detector,tname,detector,tname,detector,tname,detector,tname)

try:
    cur.execute(SQL)
except psycopg2.ProgrammingError:
    print "Validity tables for %s.%s not found" % (detector,tname)
    exit(0)
    
SQL = "commit"

cur.execute(SQL)

print "Validity tables for %s.%s dropped" % (detector,tname)