Empire/empire

#!/usr/bin/python

import sqlite3, argparse

# Empire imports
from lib.common import empire
from lib.common import listeners
from lib.common import http
from lib.common import packets
from lib.common import messages


if __name__ == '__main__':

    parser = argparse.ArgumentParser()
    parser.add_argument('--debug', action='store_true', help='Debug mode for output.')

    parser.add_argument('-s', '--stager', nargs='?', const="list", help='Specify a stager to generate. Lists all stagers if none is specified.')
    parser.add_argument('-o', '--stager-options', nargs='*', help="Supply options to set for a stager in OPTION=VALUE format. Lists options if nothing is specified.")
    parser.add_argument('-l', '--listeners', nargs='?', const="list", help='Display listener options. Displays all listeners if nothing is specified.')

    args = parser.parse_args()

    main = empire.MainMenu(args=args)
    main.cmdloop()