def cli_main(): import logging from dmoj import judgeenv, contrib, executors judgeenv.load_env(cli=True) executors.load_executors() contrib.load_contrib_modules() print('Running local judge...') logging.basicConfig(filename=judgeenv.log_file, level=logging.INFO, format='%(levelname)s %(asctime)s %(module)s %(message)s') judge = LocalJudge() for warning in judgeenv.startup_warnings: print_ansi('#ansi[Warning: %s](yellow)' % warning) del judgeenv.startup_warnings print() from dmoj.commands import all_commands, commands, register_command for command in all_commands: register_command(command(judge)) def run_command(line): if not line: return 127 if line[0] in commands: cmd = commands[line[0]] try: return cmd.execute(line[1:]) except InvalidCommandException as e: if e.message: print_ansi("#ansi[%s](red|bold)\n" % e.message) print() return 1 else: print_ansi('#ansi[Unrecognized command %s](red|bold)' % line[0]) print() return 127 with judge: try: judge.listen() if judgeenv.cli_command: return run_command(judgeenv.cli_command) else: while True: command = input(ansi_style("#ansi[dmoj](magenta)#ansi[>](green) ")).strip() run_command(shlex.split(command)) except (EOFError, KeyboardInterrupt): print() finally: judge.murder()
def main(): judgeenv.load_env(cli=True, testsuite=True) logging.basicConfig(filename=judgeenv.log_file, level=logging.INFO, format='%(levelname)s %(asctime)s %(module)s %(message)s') executors.load_executors() contrib.load_contrib_modules() tester = Tester(judgeenv.problem_regex, judgeenv.case_regex) fails = tester.test_all() print() print('Test complete') if fails: print_ansi('#ansi[A total of %d case(s) failed](red|bold).' % fails) else: print_ansi('#ansi[All cases passed.](green|bold)') raise SystemExit(int(fails != 0))
def main(): # pragma: no cover unicode_stdout_stderr() if not sanity_check(): return 1 from dmoj import judgeenv, contrib, executors judgeenv.load_env() executors.load_executors() contrib.load_contrib_modules() if hasattr(signal, 'SIGUSR2'): signal.signal(signal.SIGUSR2, signal.SIG_IGN) print('Running live judge...') for warning in judgeenv.startup_warnings: print_ansi('#ansi[Warning: %s](yellow)' % warning) del judgeenv.startup_warnings if 'judges' in env: logfile = judgeenv.log_file try: logfile = logfile % 'master' except TypeError: pass logging.basicConfig( filename=logfile, level=logging.INFO, format='%(levelname)s %(asctime)s %(process)d %(name)s %(message)s' ) if env.pidfile: with open(env.pidfile) as f: f.write(str(os.getpid())) manager = JudgeManager(env.judges) manager.run() else: return judge_proc(need_monitor=True)
def main(): # pragma: no cover unicode_stdout_stderr() if not sanity_check(): return 1 from dmoj import judgeenv, contrib, executors judgeenv.load_env() executors.load_executors() contrib.load_contrib_modules() print('Running live judge...') for warning in judgeenv.startup_warnings: print_ansi('#ansi[Warning: %s](yellow)' % warning) del judgeenv.startup_warnings logfile = judgeenv.log_file try: logfile = logfile % env['id'] except TypeError: pass logging.basicConfig( filename=logfile, level=logging.INFO, format='%(levelname)s %(asctime)s %(process)d %(module)s %(message)s') setproctitle('DMOJ Judge %s on %s' % (env['id'], make_host_port(judgeenv))) judge = ClassicJudge( judgeenv.server_host, judgeenv.server_port, secure=judgeenv.secure, no_cert_check=judgeenv.no_cert_check, cert_store=judgeenv.cert_store, ) monitor = Monitor() monitor.callback = judge.update_problems if hasattr(signal, 'SIGUSR2'): def update_problem_signal(signum, frame): judge.update_problems() signal.signal(signal.SIGUSR2, update_problem_signal) if judgeenv.api_listen: judge_instance = judge class Handler(JudgeControlRequestHandler): judge = judge_instance api_server = HTTPServer(judgeenv.api_listen, Handler) thread = threading.Thread(target=api_server.serve_forever) thread.daemon = True thread.start() else: api_server = None print() with monitor: try: judge.listen() except KeyboardInterrupt: pass except Exception: traceback.print_exc() finally: judge.murder() if api_server: api_server.shutdown()
def ci_test(executors_to_test, overrides, allow_fail=frozenset()): result = {} failed = False failed_executors = [] for name in executors_to_test: executor = import_module('dmoj.executors.' + name) print_ansi('%-34s%s' % ('Testing #ansi[%s](|underline):' % name, ''), end=' ') if not hasattr(executor, 'Executor'): failed = True print_ansi( '#ansi[Does not export](red|bold) #ansi[Executor](red|underline)' ) continue if not hasattr(executor.Executor, 'autoconfig'): print_ansi('#ansi[Could not autoconfig](magenta|bold)') continue try: if name in overrides: if not overrides[name]: print_ansi('#ansi[Environment not found on Travis](red)') continue print_ansi('#ansi[(manual config)](yellow)', end=' ') data = executor.Executor.autoconfig_run_test(overrides[name]) else: data = executor.Executor.autoconfig() config = data[0] success = data[1] feedback = data[2] errors = '' if len(data) < 4 else data[3] except Exception: print_ansi('#ansi[Autoconfig broken](red|bold)') traceback.print_exc() if name not in allow_fail: failed = True failed_executors.append(name) else: print_ansi( ['#ansi[%s](red|bold)', '#ansi[%s](green|bold)'][success] % (feedback or ['Failed', 'Success'][success])) if success: result.update(config) executor.Executor.runtime_dict = config executors[name] = executor for runtime, ver in executor.Executor.get_runtime_versions(): print_ansi(' #ansi[%s](cyan): %s' % (runtime, '.'.join(map(str, ver))) if ver else 'unknown') else: if feedback == 'Could not find JVM': continue if config: print(' Attempted:') print( ' ', yaml.safe_dump( config, default_flow_style=False).rstrip().replace( '\n', '\n' + ' ' * 4)) if errors: print(' Errors:') print(' ', errors.replace('\n', '\n' + ' ' * 4)) if name not in allow_fail: failed = True failed_executors.append(name) print() print_ansi('#ansi[Configuration result](green|bold|underline):') print( yaml.safe_dump({ 'runtime': result }, default_flow_style=False).rstrip()) print() if failed: print_ansi('#ansi[Executor configuration failed.](red|bold)') print_ansi('#ansi[Failed executors:](|bold)', ', '.join(failed_executors)) else: print_ansi('#ansi[Executor configuration succeeded.](green|bold)') load_contrib_modules() print() print() print('Running test cases...') judgeenv.problem_dirs = [ os.path.normpath( os.path.join(os.path.dirname(__file__), '..', 'testsuite')) ] tester = Tester() fails = tester.test_all() print() print('Test complete.') if fails: print_ansi('#ansi[A total of %d case(s) failed](red|bold).' % fails) else: print_ansi('#ansi[All cases passed.](green|bold)') failed |= fails != 0 raise SystemExit(int(failed))