def file_input_main(file_path: str, output: Output, connection_id_sink: ConnectionIDSink, command_sink: CommandSink, ui_state: UIState, input_func: Callable[[str], str]) -> None: ui = TerminalUI(command_sink, ui_state, input_func) logger.info('Opening ' + file_path) try: input_file = open(file_path) parse.into_sink(input_file, output, connection_id_sink) input_file.close() except FileNotFoundError: output.error(file_path + ' not found') ui.run_until_stopped() logger.info('Done with file')
def get(self): args = self.reqparse.parse_args() image_url = args['image_url'] image_file = glob(image_url) if len(image_file) < 1: print("") return Output.error("文件不存在!请先上传文件再进行识别!") image = np.array(Image.open(image_file).convert('RGB')) t = time.time() result, image_framed = ocr.model(image) print("Mission complete, it took {:.3f}s".format(time.time() - t)) print("\nRecognition Result:\n") data = '' for key in result: print(result[key][1]) # data.append(result[key][1]) data += result[key][1] # val.val() return Output.success(data)
def main(out_stream: stream.Base, err_stream: stream.Base, argv: List[str], input_func: Callable[[str], str]) -> None: ''' Parse arguments and run wayland-debug out_stream: An instance of stream.Base to use for output err_stream: An instance of stream.Base to use for logging and errors argv: A list of str arguments (should include the program name like sys.argv) input_func: the input builtin, or a mock function that behaves the same ''' # If we want to run inside GDB, the rest of main does not get called in this instance of the script # Instead GDB is run, an instance of wayland-debug is run inside it and main() is run in that # gdb_plugin.runner.parse_args() will check if this needs to happen, and gdb_plugin.run_gdb() will do it gdb_runner_args = gdb_plugin.runner.parse_args(argv) if gdb_runner_args: gdb_plugin.run_gdb(gdb_runner_args, False) return import argparse parser = argparse.ArgumentParser( description= 'Debug Wayland protocol messages, see https://github.com/wmww/wayland-debug for additional info' ) parser.add_argument('--matcher-help', action='store_true', help='show how to write matchers and exit') parser.add_argument( '-v', '--verbose', action='store_true', help='verbose output, mostly used for debugging this program') parser.add_argument('-l', '--load', dest='path', type=str, help='load WAYLAND_DEBUG=1 messages from a file') parser.add_argument( '-p', '--pipe', action='store_true', help= 'receive WAYLAND_DEBUG=1 messages from stdin (note: messages are printed to stderr so you may want to redirect using 2>&1 before piping)' ) parser.add_argument('-s', '--supress', action='store_true', help='supress non-wayland output of the program') parser.add_argument( '-c', '--color', action='store_true', help='force color output (default for interactive sessions)') parser.add_argument( '-C', '--no-color', action='store_true', help='disable color output (default for non-interactive sessions)') parser.add_argument( '-f', '--filter', dest='f', type=str, help='only show these objects/messages (see --matcher-help for syntax)' ) parser.add_argument( '-b', '--break', dest='b', type=str, help='stop on these objects/messages (see --matcher-help for syntax)') parser.add_argument( '-g', '--gdb', action='store_true', help= 'run inside gdb, all subsequent arguments are sent to gdb, when inside gdb start commands with \'wl\'' ) # NOTE: -g/--gdb is here only for the help text, it is processed without argparse in gdb_runner.main() args = parser.parse_args( args=argv[1:]) # chop off the first argument (program name) assert not args.gdb, 'GDB argument should have been intercepted by gdb_plugin.runner.parse_args()' if args.no_color: set_color_output(False) elif args.color: set_color_output(True) verbose = bool(args.verbose) unprocessed_output = not bool(args.supress) output = Output(verbose, unprocessed_output, out_stream, err_stream) if verbose: set_verbose(True) logger.info('Verbose output enabled') if args.no_color: if args.color: output.warn( 'Ignoring --color, since --no-color was also specified') logger.info('Color output disabled') elif args.color: # Print message in rainbow colors s = '' for i, c in enumerate('Color output enabled'): s += color('1;' + str(i % 6 + 31), c) logger.info(s) if unprocessed_output: logger.info('Showing unparsable output') if args.matcher_help: matcher.show_help(output) return filter_matcher = matcher.always if args.f: try: filter_matcher = matcher.parse(args.f).simplify() logger.info('Filter matcher: ' + str(filter_matcher)) except RuntimeError as e: output.error(e) stop_matcher = matcher.never if args.b: try: stop_matcher = matcher.parse(args.b).simplify() logger.info('Break matcher: ' + str(stop_matcher)) except RuntimeError as e: output.error(e) protocol.load_all(output) connection_list = ConnectionManager() ui_controller = Controller(output, connection_list, filter_matcher, stop_matcher) file_path = args.path input_from_pipe = args.pipe if check_gdb(): try: if file_path is not None or input_from_pipe: output.warn( 'Ignoring load/pipe argument because we\'re inside GDB') gdb_plugin.plugin.Plugin(output, connection_list, ui_controller, ui_controller) except: import traceback traceback.print_exc() elif file_path is not None: if input_from_pipe: output.warn('Ignoring piped input because load file is specified') file_input_main(file_path, output, connection_list, ui_controller, ui_controller, input_func) elif input_from_pipe: if args.b: output.warn( 'Ignoring stop matcher when stdin is used for messages') piped_input_main(output, connection_list) else: output.warn('No action specified') parser.print_help()
def test_error(self): o = Output(True, True, self.out, self.err) o.error('abc') self.assertIn('Error:', self.err.buffer) self.assertIn('abc', self.err.buffer) self.assertEqual(self.out.buffer, '')