def setParserOptions(): parser = argparse.ArgumentParser(prog="trex_console.py") parser.add_argument("-s", "--server", help = "TRex Server [default is localhost]", default = "localhost", type = str) parser.add_argument("-p", "--port", help = "TRex Server Port [default is 4501]\n", default = 4501, type = int) parser.add_argument("--async_port", help = "TRex ASync Publisher Port [default is 4500]\n", default = 4500, dest='pub', type = int) parser.add_argument("-u", "--user", help = "User Name [default is currently logged in user]\n", default = get_current_user(), type = str) parser.add_argument("-v", "--verbose", dest="verbose", action="store_true", help="Switch ON verbose option. Default is: OFF.", default = False) parser.add_argument("--no_acquire", dest="acquire", action="store_false", help="Acquire all ports on connect. Default is: ON.", default = True) parser.add_argument("--batch", dest="batch", nargs = 1, type = is_valid_file, help = "Run the console in a batch mode with file", default = None) parser.add_argument("-t", "--tui", dest="tui", action="store_true", help="Starts with TUI mode", default = False) parser.add_argument("-x", "--xtui", dest="xtui", action="store_true", help="Starts with XTERM TUI mode", default = False) parser.add_argument("--top", dest="top", action="store_true", help="Set the window as always on top", default = False) parser.add_argument("-q", "--quiet", dest="quiet", action="store_true", help="Starts with all outputs suppressed", default = False) return parser
def __init__(self, trex_client=None, server=None): self.ssl_ctx = None if not (bool(server) ^ bool(trex_client)): raise STLError( 'Please specify either trex_client or server argument.') if not server: server = trex_client.get_connection_info()['server'] self.bg_client = STLClient('AP Manager', server, verbose_level=0) self.trex_client = trex_client or self.bg_client self.aps = [] self.clients = [] self.ap_by_id = {} self.client_by_id = {} self.bg_lock = threading.RLock() self.service_ctx = {} self.base_file_path = '/tmp/trex/console/%s_%s.wlc_base' % ( get_current_user(), server) base_file_dir = os.path.dirname(self.base_file_path) if not os.path.exists(base_file_dir): os.makedirs(base_file_dir, mode=0o777) self._ap_name_re = re.compile(r'(.*?)(\d+)') self._init_base_vals()
def setParserOptions(): parser = argparse.ArgumentParser(prog="trex_console.py") parser.add_argument("-s", "--server", help = "TRex Server [default is localhost]", default = "localhost", type = str) parser.add_argument("-p", "--port", help = "TRex Server Port [default is 4501]\n", default = 4501, type = int) parser.add_argument("--async_port", help = "TRex ASync Publisher Port [default is 4500]\n", default = 4500, dest='pub', type = int) parser.add_argument("-u", "--user", help = "User Name [default is currently logged in user]\n", default = get_current_user(), type = str) parser.add_argument("-v", "--verbose", dest="verbose", action="store_true", help="Switch ON verbose option. Default is: OFF.", default = False) group = parser.add_mutually_exclusive_group() group.add_argument("-a", "--acquire", dest="acquire", nargs = '+', type = int, help="Acquire ports on connect. default is all available ports", default = None) group.add_argument("-r", "--readonly", dest="readonly", action="store_true", help="Starts console in a read only mode", default = False) parser.add_argument("-f", "--force", dest="force", action="store_true", help="Force acquire the requested ports", default = False) parser.add_argument("--batch", dest="batch", nargs = 1, type = is_valid_file, help = "Run the console in a batch mode with file", default = None) parser.add_argument("-t", "--tui", dest="tui", action="store_true", help="Starts with TUI mode", default = False) parser.add_argument("-x", "--xtui", dest="xtui", action="store_true", help="Starts with XTERM TUI mode", default = False) parser.add_argument("--top", dest="top", action="store_true", help="Set the window as always on top", default = False) parser.add_argument("-q", "--quiet", dest="quiet", action="store_true", help="Starts with all outputs suppressed", default = False) return parser
def get_console_identifier(self): return "{context}_{server}".format(context=get_current_user(), server=self.stateless_client.get_connection_info()['server'])