def start(self): """ """ parser = CustomParser(self) parser.prog = "tftp" parser.add_argument("hostname", nargs='?', default=None) parser.add_argument("-c", nargs=2) parser.add_argument("-l") parser.add_argument("-g") parser.add_argument("-p") parser.add_argument("-r") try: args = parser.parse_args(self.args) if args.c: if len(args.c) > 1: command = args.c[0] self.file_to_get = args.c[1] if args.hostname is None: raise OptionNotFound("Hostname is invalid") self.hostname = args.hostname elif args.r: self.file_to_get = args.r self.hostname = args.g else: parser.print_usage() raise OptionNotFound("Missing!!") if self.hostname is None: raise OptionNotFound("Hostname is invalid") self.makeTftpRetrieval() except OptionNotFound: self.exit() return except ExitException: self.exit() return except Exception: self.exit() return self.exit()
def start(self): """ """ parser = CustomParser(self) parser.prog = "tftp" parser.add_argument("hostname", nargs='?', default=None) parser.add_argument("-c", nargs=2) parser.add_argument("-l") parser.add_argument("-g") parser.add_argument("-p") parser.add_argument("-r") try: args = parser.parse_args(self.args) if args.c: if len(args.c) > 1: command = args.c[0] self.file_to_get = args.c[1] if args.hostname is None: raise OptionNotFound("Hostname is invalid") self.hostname = args.hostname elif args.r: self.file_to_get = args.r self.hostname = args.g else: parser.print_usage() raise OptionNotFound("Missing!!") if self.hostname is None: raise OptionNotFound("Hostname is invalid") if self.hostname.find(':') != -1: host, port = self.hostname.split(':') self.hostname = host self.port = int(port) self.makeTftpRetrieval() except Exception as err: log.err(str(err)) self.exit()