if conf.get(Config.QUOTA_TYPE) == 'space' and query_yes_no("Are you sure you want to clear %s's space quota?" % (conf.get(Config.USER)), default="no"):
                userUtil.clearUserSpaceQuota(conf.get(Config.USER))
                userUtil.printUserSpaceQuota(conf.get(Config.USER))
            elif conf.get(Config.QUOTA_TYPE) == 'inode' and query_yes_no("Are you sure you want to clear %s's inode quota?" % (conf.get(Config.USER)), default="no"):
                userUtil.clearUserINodeQuota(conf.get(Config.USER))
                userUtil.printUserINodeQuota(conf.get(Config.USER))
        else:
            printError("Unknown action %s" % (conf.get(Config.ACTION)))

# postgres option
    elif sys.argv[1] == "pg-util": 
        parser = OptionParser()
        parser.add_option("-c", "--config", dest="configFile", help="Configuration file (default phd-metrics.ini)", default="phd-metrics.ini")
        parser.add_option("-s", "--sqlfile", dest="sqlFile", help="Filename to write SQL statements to (default none)", default=None)
            
        conf = Config(parser, sys.argv[2:])      

        try:
            conf.get(Config.SQL_FILE)
        except KeyError:
            printError("Must specify --sqlfile option for pg-util tool")

        pgutil = PostgresUtil(conf)
        pgutil.open()
        pgutil.writeCreates()
        pgutil.close()
    else:
        printError("Unknown tool")
        
    sys.exit(0)
Ejemplo n.º 2
0
# postgres option
    elif sys.argv[1] == "pg-util":
        parser = OptionParser()
        parser.add_option("-c",
                          "--config",
                          dest="configFile",
                          help="Configuration file (default phd-metrics.ini)",
                          default="phd-metrics.ini")
        parser.add_option(
            "-s",
            "--sqlfile",
            dest="sqlFile",
            help="Filename to write SQL statements to (default none)",
            default=None)

        conf = Config(parser, sys.argv[2:])

        try:
            conf.get(Config.SQL_FILE)
        except KeyError:
            printError("Must specify --sqlfile option for pg-util tool")

        pgutil = PostgresUtil(conf)
        pgutil.open()
        pgutil.writeCreates()
        pgutil.close()
    else:
        printError("Unknown tool")

    sys.exit(0)