def add_subparser(subparsers: argparse.Action) -> None:
subparsers_add_parser: Callable[..., argparse.ArgumentParser] = subparsers.add_parser # type: ignore
subparser = subparsers_add_parser('test-reactive', aliases=['t/r'], help='test for reactive problem', formatter_class=argparse.RawTextHelpFormatter, epilog='''\
tips:
You can do similar things with shell
e.g. $ mkfifo a.pipe && ./a.out < a.pipe | python3 judge.py > a.pipe
''')
subparser.add_argument('-c', '--command', default=utils.get_default_command(), help='your solution to be tested. (default: "{}")'.format(utils.get_default_command()))
subparser.add_argument('judge', help='judge program using standard I/O')
def add_subparser(subparsers: argparse.Action) -> None:
subparsers_add_parser: Callable[..., argparse.ArgumentParser] = subparsers.add_parser # type: ignore
subparser = subparsers_add_parser('generate-output', aliases=['g/o'], help='generate output files from input and reference implementation', formatter_class=argparse.RawTextHelpFormatter, epilog='''\
format string for --format:
%s name
%e extension: "in" or "out"
(both %s and %e are required.)
tips:
You can do similar things with shell
e.g. $ for f in test/*.in ; do ./a.out < $f > ${f%.in}.out ; done
''')
subparser.add_argument('-c', '--command', default=utils.get_default_command(), help='your solution to be tested. (default: "{}")'.format(utils.get_default_command()))
subparser.add_argument('-f', '--format', default='%s.%e', help='a format string to recognize the relationship of test cases. (default: "%%s.%%e")')
subparser.add_argument('-d', '--directory', type=pathlib.Path, default=pathlib.Path('test'), help='a directory name for test cases (default: test/)')
subparser.add_argument('-t', '--tle', type=float, help='set the time limit (in second) (default: inf)')
subparser.add_argument('-j', '--jobs', type=int, help='run tests in parallel')
subparser.add_argument('test', nargs='*', type=pathlib.Path, help='paths of input cases. (if empty: globbed from --format)')
subparser.add_argument('--no-ignore-backup', action='store_false', dest='ignore_backup')
subparser.add_argument('--ignore-backup', action='store_true', help='ignore backup files and hidden files (i.e. files like "*~", "\\#*\\#" and ".*") (default)')
def add_subparser(subparsers: argparse.Action) -> None:
subparsers_add_parser: Callable[..., argparse.ArgumentParser] = subparsers.add_parser # type: ignore
subparser = subparsers_add_parser('test', aliases=['t'], help='test your code', formatter_class=argparse.RawTextHelpFormatter, epilog='''\
format string for --format:
%s name
%e extension: "in" or "out"
(both %s and %e are required.)
tips:
There is a feature to use special judges. See https://github.com/online-judge-tools/oj/blob/master/docs/getting-started.md#test-for-problems-with-special-judge for details.
You can do similar things with shell
e.g. $ for f in test/*.in ; do echo $f ; ./a.out < $f | diff - ${f%.in}.out ; done
''')
subparser.add_argument('-c', '--command', default=utils.get_default_command(), help='your solution to be tested. (default: "{}")'.format(utils.get_default_command()))
subparser.add_argument('-f', '--format', default='%s.%e', help='a format string to recognize the relationship of test cases. (default: "%%s.%%e")')
subparser.add_argument('-d', '--directory', type=pathlib.Path, default=pathlib.Path('test'), help='a directory name for test cases (default: test/)')
subparser.add_argument('-m', '--compare-mode', choices=[mode.value for mode in CompareMode], default=CompareMode.CRLF_INSENSITIVE_EXACT_MATCH.value, help='mode to compare outputs. The default behavoir is exact-match to ensure that you always get AC on remote judge servers when you got AC on local tests for the same cases. (default: crlf-insensitive-exact-match)')
subparser.add_argument('-M', '--display-mode', choices=[mode.value for mode in DisplayMode], default=DisplayMode.SUMMARY.value, help='mode to display outputs (default: summary)')
subparser.add_argument('-S', '--ignore-spaces', dest='compare_mode', action='store_const', const=CompareMode.IGNORE_SPACES.value, help="ignore spaces to compare outputs, but doesn't ignore newlines (equivalent to --compare-mode=ignore-spaces")
subparser.add_argument('-N', '--ignore-spaces-and-newlines', dest='compare_mode', action='store_const', const=CompareMode.IGNORE_SPACES_AND_NEWLINES.value, help='ignore spaces and newlines to compare outputs (equivalent to --compare-mode=ignore-spaces-and-newlines')
subparser.add_argument('-D', '--diff', dest='display_mode', action='store_const', const=DisplayMode.DIFF.value, help='display the diff (equivalent to --display-mode=diff)')
subparser.add_argument('-s', '--silent', action='store_true', help='don\'t report output and correct answer even if not AC (for --mode all)')
subparser.add_argument('-e', '--error', type=float, help='check as floating point number: correct if its absolute or relative error doesn\'t exceed it')
subparser.add_argument('-t', '--tle', type=float, help='set the time limit (in second) (default: inf)')
subparser.add_argument('--mle', type=float, help='set the memory limit (in megabyte) (default: inf)')
subparser.add_argument('-i', '--print-input', action='store_true', default=True, help='print input cases if not AC (default)')
subparser.add_argument('--no-print-input', action='store_false', dest='print_input')
subparser.add_argument('-j', '--jobs', metavar='N', type=int, help='specifies the number of jobs to run simultaneously (default: no parallelization)')
subparser.add_argument('--print-memory', action='store_true', help='print the amount of memory which your program used, even if it is small enough')
subparser.add_argument('--gnu-time', help='used to measure memory consumption (default: "time")', default='time')
subparser.add_argument('--no-ignore-backup', action='store_false', dest='ignore_backup')
subparser.add_argument('--ignore-backup', action='store_true', help='ignore backup files and hidden files (i.e. files like "*~", "\\#*\\#" and ".*") (default)')
subparser.add_argument('--log-file', type=pathlib.Path, help=argparse.SUPPRESS)
subparser.add_argument('--judge-command', dest='judge', default=None, help='specify judge command instead of default diff judge. The given command (e.g. `./judge`) will be called as `$ ./judge input.txt actual-output.txt expected-output.txt` and should return the result with the exit code of its `main` function.')
subparser.add_argument('test', nargs='*', type=pathlib.Path, help='paths of test cases. (if empty: globbed from --format)')
def get_parser() -> argparse.ArgumentParser:
parser = argparse.ArgumentParser(
description='Tools for online judge services')
parser.add_argument('-v', '--verbose', action='store_true')
parser.add_argument('-c',
'--cookie',
type=pathlib.Path,
default=utils.default_cookie_path,
help='path to cookie. (default: {})'.format(
utils.default_cookie_path))
parser.add_argument('--version',
action='store_true',
help='print the online-judge-tools version number')
# TODO: configure subparsers in each module for the subcommand, not this module for the entry point
subparsers = parser.add_subparsers(
dest='subcommand',
help='for details, see "{} COMMAND --help"'.format(sys.argv[0]))
# download
subparser = subparsers.add_parser(
'download',
aliases=['d', 'dl'],
help='download sample cases',
formatter_class=argparse.RawTextHelpFormatter,
epilog='''\
supported services:
Anarchy Golf
Aizu Online Judge (including the Arena)
AtCoder
Codeforces
yukicoder
CS Academy
HackerRank
PKU JudgeOnline
Kattis
Toph (Problem Archive)
CodeChef
Facebook Hacker Cup
Google Code Jam
Library Checker (https://judge.yosupo.jp/)
supported services with --system:
Aizu Online Judge
yukicoder
Library Checker (https://judge.yosupo.jp/)
format string for --format:
%i index: 1, 2, 3, ...
%e extension: "in" or "out"
%n name: e.g. "Sample Input 1", "system_test3.txt", ...
%b os.path.basename(name)
%d os.path.dirname(name)
%% '%' itself
tips:
This subcommand doesn't have the feature to download all test cases for all problems in a contest at once. If you want to do this, please use `oj-prepare` command at https://github.com/online-judge-tools/template-generator instead.
You can do similar things with shell and oj-api command. see https://github.com/online-judge-tools/api-client
e.g. $ oj-api get-problem https://atcoder.jp/contests/agc001/tasks/agc001_a | jq -cr '.result.tests | to_entries[] | [{path: "test/sample-\\(.key).in", data: .value.input}, {path: "test/sample-\\(.key).out", data: .value.output}][] | {path, data: @sh "\\(.data)"} | "mkdir -p test; echo -n \\(.data) > \\(.path)"' | sh
''')
subparser.add_argument('url')
subparser.add_argument(
'-f',
'--format',
help=
'a format string to specify paths of cases (default: "sample-%%i.%%e" if not --system)'
) # default must be None for --system
subparser.add_argument(
'-d',
'--directory',
type=pathlib.Path,
help='a directory name for test cases (default: test/)'
) # default must be None for guessing in submit command
subparser.add_argument('-n',
'--dry-run',
action='store_true',
help='don\'t write to files')
subparser.add_argument('-a',
'--system',
action='store_true',
help='download system testcases')
subparser.add_argument('-s', '--silent', action='store_true')
subparser.add_argument('--yukicoder-token', type=str)
subparser.add_argument('--log-file',
type=pathlib.Path,
help=argparse.SUPPRESS)
# login
subparser = subparsers.add_parser(
'login',
aliases=['l'],
help='login to a service',
formatter_class=argparse.RawTextHelpFormatter,
epilog='''\
supported services:
AtCoder
Codeforces
yukicoder
HackerRank
Toph
tips:
You can do similar things with shell and oj-api command. see https://github.com/online-judge-tools/api-client
e.g. $ USERNAME=foo PASSWORD=bar oj-api login-service https://atcoder.jp/
''')
subparser.add_argument('url')
subparser.add_argument('-u', '--username')
subparser.add_argument('-p', '--password')
subparser.add_argument('--check',
action='store_true',
help='check whether you are logged in or not')
subparser.add_argument(
'--use-browser',
choices=('always', 'auto', 'never'),
default='auto',
help=
'specify whether it uses a GUI web browser to login or not (default: auto)'
)
# submit
subparser = subparsers.add_parser(
'submit',
aliases=['s'],
help='submit your solution',
formatter_class=argparse.RawTextHelpFormatter,
epilog='''\
supported services:
AtCoder
Codeforces
yukicoder
HackerRank
Toph (Problem Archive)
tips:
This subcommand has the feature to guess the problem to submit to. To guess the problem, run `oj download https://...` in the same directory without `--directory` option before using `oj submit ...`.
you can do similar things with shell and oj-api command. see https://github.com/online-judge-tools/api-client
e.g. $ oj-api submit-code --file main.cpp --language $(oj-api guess-language-id --file main.cpp https://atcoder.jp/contests/agc001/tasks/agc001_a | jq -r .result.id) https://atcoder.jp/contests/agc001/tasks/agc001_a
''')
subparser.add_argument(
'url',
nargs='?',
help=
'the URL of the problem to submit. if not given, guessed from history of download command.'
)
subparser.add_argument('file', type=pathlib.Path)
subparser.add_argument('-l',
'--language',
help='narrow down language choices if ambiguous')
subparser.add_argument('--no-guess', action='store_false', dest='guess')
subparser.add_argument('-g',
'--guess',
action='store_true',
help='guess the language for your file (default)')
subparser.add_argument('--no-guess-latest',
action='store_false',
dest='guess_cxx_latest')
subparser.add_argument('--guess-cxx-latest',
action='store_true',
help='use the lasest version for C++ (default)')
subparser.add_argument(
'--guess-cxx-compiler',
choices=('gcc', 'clang', 'all'),
default='gcc',
help=
'use the specified C++ compiler if both of GCC and Clang are available (default: gcc)'
)
subparser.add_argument('--guess-python-version',
choices=('2', '3', 'auto', 'all'),
default='auto',
help='default: auto')
subparser.add_argument(
'--guess-python-interpreter',
choices=('cpython', 'pypy', 'all'),
default='cpython',
help=
'use the specified Python interpreter if both of CPython and PyPy are available (default: cpython)'
)
subparser.add_argument('--no-open', action='store_false', dest='open')
subparser.add_argument(
'--open',
action='store_true',
default=True,
help='open the result page after submission (default)')
subparser.add_argument('-w',
'--wait',
metavar='SECOND',
type=float,
default=3,
help='sleep before submitting')
subparser.add_argument('-y',
'--yes',
action='store_true',
help='don\'t confirm')
# test
subparser = subparsers.add_parser(
'test',
aliases=['t'],
help='test your code',
formatter_class=argparse.RawTextHelpFormatter,
epilog='''\
format string for --format:
%s name
%e extension: "in" or "out"
(both %s and %e are required.)
tips:
There is a feature to use special judges. See https://online-judge-tools.readthedocs.io/en/master/introduction.en.html#test-for-problems-with-special-judge for details.
You can do similar things with shell
e.g. $ for f in test/*.in ; do echo $f ; ./a.out < $f | diff - ${f%.in}.out ; done
''')
subparser.add_argument(
'-c',
'--command',
default=utils.get_default_command(),
help='your solution to be tested. (default: "{}")'.format(
utils.get_default_command()))
subparser.add_argument(
'-f',
'--format',
default='%s.%e',
help=
'a format string to recognize the relationship of test cases. (default: "%%s.%%e")'
)
subparser.add_argument(
'-d',
'--directory',
type=pathlib.Path,
default=pathlib.Path('test'),
help='a directory name for test cases (default: test/)')
subparser.add_argument(
'-m',
'--compare-mode',
choices=[mode.value for mode in CompareMode],
default=CompareMode.CRLF_INSENSITIVE_EXACT_MATCH.value,
help=
'mode to compare outputs. The default behavoir is exact-match to ensure that you always get AC on remote judge servers when you got AC on local tests for the same cases. (default: crlf-insensitive-exact-match)'
)
subparser.add_argument('-M',
'--display-mode',
choices=[mode.value for mode in DisplayMode],
default=DisplayMode.SUMMARY.value,
help='mode to display outputs (default: summary)')
subparser.add_argument(
'-S',
'--ignore-spaces',
dest='compare_mode',
action='store_const',
const=CompareMode.IGNORE_SPACES.value,
help=
"ignore spaces to compare outputs, but doesn't ignore newlines (equivalent to --compare-mode=ignore-spaces"
)
subparser.add_argument(
'-N',
'--ignore-spaces-and-newlines',
dest='compare_mode',
action='store_const',
const=CompareMode.IGNORE_SPACES_AND_NEWLINES.value,
help=
'ignore spaces and newlines to compare outputs (equivalent to --compare-mode=ignore-spaces-and-newlines'
)
subparser.add_argument(
'-D',
'--diff',
dest='display_mode',
action='store_const',
const=DisplayMode.DIFF.value,
help='display the diff (equivalent to --display-mode=diff)')
subparser.add_argument(
'-s',
'--silent',
action='store_true',
help=
'don\'t report output and correct answer even if not AC (for --mode all)'
)
subparser.add_argument(
'-e',
'--error',
type=float,
help=
'check as floating point number: correct if its absolute or relative error doesn\'t exceed it'
)
subparser.add_argument(
'-t',
'--tle',
type=float,
help='set the time limit (in second) (default: inf)')
subparser.add_argument(
'--mle',
type=float,
help='set the memory limit (in megabyte) (default: inf)')
subparser.add_argument('-i',
'--print-input',
action='store_true',
default=True,
help='print input cases if not AC (default)')
subparser.add_argument('--no-print-input',
action='store_false',
dest='print_input')
subparser.add_argument(
'-j',
'--jobs',
metavar='N',
type=int,
help=
'specifies the number of jobs to run simultaneously (default: no parallelization)'
)
subparser.add_argument(
'--print-memory',
action='store_true',
help=
'print the amount of memory which your program used, even if it is small enough'
)
subparser.add_argument(
'--gnu-time',
help='used to measure memory consumption (default: "time")',
default='time')
subparser.add_argument('--no-ignore-backup',
action='store_false',
dest='ignore_backup')
subparser.add_argument(
'--ignore-backup',
action='store_true',
help=
'ignore backup files and hidden files (i.e. files like "*~", "\\#*\\#" and ".*") (default)'
)
subparser.add_argument('--log-file',
type=pathlib.Path,
help=argparse.SUPPRESS)
subparser.add_argument(
'--judge-command',
dest='judge',
default=None,
help=
'specify judge command instead of default diff judge. The given command (e.g. `./judge`) will be called as `$ ./judge input.txt actual-output.txt expected-output.txt` and should return the result with the exit code of its `main` function.'
)
subparser.add_argument(
'test',
nargs='*',
type=pathlib.Path,
help='paths of test cases. (if empty: globbed from --format)')
# generate output
subparser = subparsers.add_parser(
'generate-output',
aliases=['g/o'],
help='generate output files from input and reference implementation',
formatter_class=argparse.RawTextHelpFormatter,
epilog='''\
format string for --format:
%s name
%e extension: "in" or "out"
(both %s and %e are required.)
tips:
You can do similar things with shell
e.g. $ for f in test/*.in ; do ./a.out < $f > ${f%.in}.out ; done
''')
subparser.add_argument(
'-c',
'--command',
default=utils.get_default_command(),
help='your solution to be tested. (default: "{}")'.format(
utils.get_default_command()))
subparser.add_argument(
'-f',
'--format',
default='%s.%e',
help=
'a format string to recognize the relationship of test cases. (default: "%%s.%%e")'
)
subparser.add_argument(
'-d',
'--directory',
type=pathlib.Path,
default=pathlib.Path('test'),
help='a directory name for test cases (default: test/)')
subparser.add_argument(
'-t',
'--tle',
type=float,
help='set the time limit (in second) (default: inf)')
subparser.add_argument('-j',
'--jobs',
type=int,
help='run tests in parallel')
subparser.add_argument(
'test',
nargs='*',
type=pathlib.Path,
help='paths of input cases. (if empty: globbed from --format)')
subparser.add_argument('--no-ignore-backup',
action='store_false',
dest='ignore_backup')
subparser.add_argument(
'--ignore-backup',
action='store_true',
help=
'ignore backup files and hidden files (i.e. files like "*~", "\\#*\\#" and ".*") (default)'
)
# generate input
subparser = subparsers.add_parser(
'generate-input',
aliases=['g/i'],
help='generate input files from given generator',
formatter_class=argparse.RawTextHelpFormatter,
epilog='''\
format string for --format:
%s name
%e extension: "in" or "out"
(both %d and %e are required.)
tips:
There is a command to automatically generate a input generator, `oj-template` command. See https://github.com/online-judge-tools/template-generator .
This subcommand has also the feature to find a hack case.
e.g. for a target program `a.out`, a correct (but possibly slow) program `naive`, and a random input-case generator `generate.py`, run $ oj g/i --hack-actual ./a.out --hack-expected ./naive 'python3 generate.py'
You can do similar things with shell
e.g. $ for i in `seq 100` ; do python3 generate.py > test/random-$i.in ; done
''')
subparser.add_argument(
'-f',
'--format',
default='%s.%e',
help=
'a format string to recognize the relationship of test cases. (default: "%%s.%%e")'
)
subparser.add_argument(
'-d',
'--directory',
type=pathlib.Path,
default=pathlib.Path('test'),
help='a directory name for test cases (default: test/)')
subparser.add_argument(
'-t',
'--tle',
type=float,
help='set the time limit (in second) (default: inf)')
subparser.add_argument('-j',
'--jobs',
type=int,
help='run tests in parallel')
subparser.add_argument(
'--width',
type=int,
default=3,
help='specify the width of indices of cases. (default: 3)')
subparser.add_argument(
'--name', help='specify the base name of cases. (default: "random")')
subparser.add_argument('-c',
'--command',
help='specify your solution to generate output')
subparser.add_argument('--hack-expected',
dest='command',
help='alias of --command')
subparser.add_argument(
'--hack',
'--hack-actual',
dest='hack',
help=
'specify your wrong solution to be compared with the reference solution given by --hack-expected'
)
subparser.add_argument('generator',
type=str,
help='your program to generate test cases')
subparser.add_argument(
'count',
nargs='?',
type=int,
help='the number of cases to generate (default: 100)')
# test reactive
subparser = subparsers.add_parser(
'test-reactive',
aliases=['t/r'],
help='test for reactive problem',
formatter_class=argparse.RawTextHelpFormatter,
epilog='''\
tips:
You can do similar things with shell
e.g. $ mkfifo a.pipe && ./a.out < a.pipe | python3 judge.py > a.pipe
''')
subparser.add_argument(
'-c',
'--command',
default=utils.get_default_command(),
help='your solution to be tested. (default: "{}")'.format(
utils.get_default_command()))
subparser.add_argument('judge', help='judge program using standard I/O')
return parser