upgrade.py

import sys

from sqltools import dbUpgradeExecute, dbRollback

try:
    from getpass import getpass
except ImportError:
    print "Missing Python module 'getpass'. You can continue but password will echo on console\n"

sys.argv.pop(0)

print "Please make sure you have backed up your database before proceeding"
username = raw_input('Enter DB admin username: ')
try:
    password = getpass('Enter DB admin password: ')
except NameError:
    password = raw_input('Enter DB admin password: ')

try:
    if len(sys.argv) == 1:
        cmd = None
        try:
            upgradeFile = sys.argv[0]
            with open(upgradeFile) as fp:
                for line in fp:
                    cmd = line.strip('\n')
                    if len(cmd) > 0:
                        # make sure we skip empty lines
                        if cmd[:2] == '2.':
                            print "Running patch: ", cmd[2:]
                            exec cmd[2:]
                        else:
                            print "Running patch: ", cmd
                            dbUpgradeExecute(username, password, cmd)
            print "Patches Applied Successfully"

        except Exception, e:
            print str(e) + " problem applying command: " + str(cmd) + " \nPossibly already applied?"
            dbRollback()

    else:
        print "Usage Guidelines: python upgrade.py <patchfile>"
except Exception, e:
    print "Something failed trying to upgrade " + str(e)
    dbRollback()