예제 #1
0
def main():
    try:
        opts, args = getopt.getopt(sys.argv[1:], 'c:', ['config='])
    except getopt.GetoptError as err:
        error(str(err), 2, usage=usage)

    dbConfigFile = 'configs/terrain/database.cfg'
    for o, a in opts:
        if o in ('-c', '--config'):
            dbConfigFile = a

    if len(args) < 1:
        error('you must specify a command', 3, usage=usage)

    db = DB(dbConfigFile)

    command = args[0]
    if command == 'console':
        db.console()
    elif command == 'create':
        db.create()
    elif command == 'createuser':
        db.createUser()
    elif command == 'createdb':
        db.createDB()
    elif command == 'setupfunctions':
        db.setupFunctions()
    elif command == 'populate':
        db.populate()
    elif command == 'populatelakes':
        db.populateLakes()
    elif command == 'dropuser':
        db.dropUser()
    elif command == 'dropdb':
        db.dropDatabase()
    elif command == 'destroy':
        db.destroy()
    else:
        error("unknown command '%(command)s'" % {'command': command},
              4,
              usage=usage)
예제 #2
0
def main():
    try:
        opts, args = getopt.getopt(sys.argv[1:], 'c:', ['config='])
    except getopt.GetoptError as err:
        error(str(err), 2, usage=usage)

    dbConfigFile = 'configs/terrain/database.cfg'
    for o, a in opts:
        if o in ('-c', '--config'):
            dbConfigFile = a

    if len(args) < 1:
        error('you must specify a command', 3, usage=usage)

    db = DB(dbConfigFile)

    command = args[0]
    if command == 'console':
        db.console()
    elif command == 'create':
        db.create()
    elif command == 'createuser':
        db.createUser()
    elif command == 'createdb':
        db.createDB()
    elif command == 'setupfunctions':
        db.setupFunctions()
    elif command == 'populate':
        db.populate()
    elif command == 'populatelakes':
        db.populateLakes()
    elif command == 'dropuser':
        db.dropUser()
    elif command == 'dropdb':
        db.dropDatabase()
    elif command == 'destroy':
        db.destroy()
    else:
        error("unknown command '%(command)s'" % {'command': command}, 4, usage=usage)