def run_program(args: argparse.Namespace, parser: argparse.ArgumentParser) -> None:
# logging
log_level = log.logging.INFO
if args.verbose:
log_level = log.logging.DEBUG
log.setLevel(log_level)
handler = log.logging.StreamHandler(sys.stderr)
handler.setLevel(log_level)
log.addHandler(handler)
log.debug('args: %s', str(args))
if args.subcommand in [ 'download', 'd', 'dl' ]:
download(args)
elif args.subcommand in [ 'login', 'l' ]:
login(args)
elif args.subcommand in [ 'submit', 's' ]:
submit(args)
elif args.subcommand in [ 'test', 't' ]:
test(args)
elif args.subcommand in [ 'test-reactive', 't/r' ]:
test_reactive(args)
elif args.subcommand in [ 'generate-scanner', 'g/s' ]:
generate_scanner(args)
elif args.subcommand in [ 'generate-output', 'g/o' ]:
generate_output(args)
elif args.subcommand in [ 'split-input', 's/i' ]:
split_input(args)
elif args.subcommand in [ 'code-statistics', 'c/s' ]:
code_statistics(args)
elif args.subcommand == 'get-standings':
get_standings(args)
else:
parser.print_help(file=sys.stderr)
sys.exit(1)
def setUpClass(cls):
cwd = os.getcwd()
cls.ojtools = os.path.join(cwd, 'oj')
# loggin
log_level = log.logging.INFO
log.setLevel(log_level)
handler = log.logging.StreamHandler(sys.stdout)
handler.setLevel(log_level)
log.addHandler(handler)
def main(args: Optional[List[str]] = None) -> None:
log.addHandler(log.logging.StreamHandler(sys.stderr))
log.setLevel(log.logging.INFO)
version_check()
parser = get_parser()
namespace = parser.parse_args(args=args)
try:
run_program(namespace, parser=parser)
except NotImplementedError as e:
log.debug('\n' + traceback.format_exc())
log.error('NotImplementedError')
log.info(
'The operation you specified is not supported yet. Pull requests are welcome.'
)
log.info(
'see: https://github.com/kmyk/online-judge-tools/blob/master/CONTRIBUTING.md'
)
def main(args=None):
# argparse
parser = argparse.ArgumentParser(
description="Tools for online judge services")
parser.add_argument("-v", "--verbose", action="store_true")
parser.add_argument(
"-c",
"--cookie",
help="path to cookie. (default: {})".format(utils.default_cookie_path),
)
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
AtCoder
Codeforces
HackerRank
Yukicoder
CS Academy
supported services with --system:
Aizu Online Judge
Yukicoder
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
""",
)
subparser.add_argument("url")
subparser.add_argument(
"-f",
"--format",
help=
'a format string to specify paths of cases (defaut: "sample-%%i.%%e" if not --system)',
) # default must be None for --system
subparser.add_argument(
"-d",
"--directory",
default="test",
help="a directory name for test cases (default: test/)",
)
subparser.add_argument("--overwrite", action="store_true")
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")
# login
subparser = subparsers.add_parser(
"login",
aliases=["l"],
help="login to a service",
formatter_class=argparse.RawTextHelpFormatter,
epilog="""\
supported services:
AtCoder
Codeforces
HackerRank
Yukicoder
TopCoder
strings for --method:
github for yukicoder, login via github (default)
twitter for yukicoder, login via twitter (not implementated yet)
""",
)
subparser.add_argument("url")
subparser.add_argument("-u", "--username")
subparser.add_argument("-p", "--password")
subparser.add_argument("--method")
# submit
subparser = subparsers.add_parser(
"submit",
aliases=["s"],
help="submit your solution",
formatter_class=argparse.RawTextHelpFormatter,
epilog="""\
supported services:
AtCoder
Yukicoder
TopCoder (Marathon Match)
""",
)
subparser.add_argument("url")
subparser.add_argument("file")
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=
'shebang or modelines are used by default. write something like "#!/usr/bin/env python3". (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(
"--format-dos2unix",
action="store_true",
help="replace CRLF with LF for given file",
)
subparser.add_argument(
"--format-rstrip",
action="store_true",
help="remove trailing newlines from given file",
)
subparser.add_argument(
"-G",
"--golf",
action="store_true",
help="now equivalent to --format-dos2unix --format-rstrip",
)
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("--open-browser")
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")
subparser.add_argument(
"--full-submission",
action="store_true",
help=
'for TopCoder Marathon Match. use this to do "Submit", the default behavier is "Test Examples".',
)
# 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:
You can do similar things with shell: e.g. `for f in test/*.in ; do echo $f ; diff <(./a.out < $f) ${f/.in/.out} ; done`
""",
)
subparser.add_argument(
"-c",
"--command",
default="./a.out",
help='your solution to be tested. (default: "./a.out")',
)
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",
default="test",
help="a directory name for test cases (default: test/)",
)
subparser.add_argument(
"-m",
"--mode",
choices=["all", "line"],
default="all",
help="mode to check an output with the correct answer. (default: all)",
)
subparser.add_argument(
"-1",
"--line",
dest="mode",
action="store_const",
const="line",
help="equivalent to --mode line",
)
subparser.add_argument("--no-rstrip", action="store_false", dest="rstrip")
subparser.add_argument("--rstrip",
action="store_true",
help="rstrip output before compare (default)")
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)
subparser.add_argument("-i",
"--print-input",
action="store_true",
help="print input cases if not AC")
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(
"test",
nargs="*",
help="paths of test cases. (if empty: globbed from --format)")
# generate scanner
subparser = subparsers.add_parser(
"generate-scanner",
aliases=["g/s"],
help="generate input scanner (experimental)",
formatter_class=argparse.RawTextHelpFormatter,
epilog="""\
supported services:
AtCoder
HackerRank
(Yukicoder has been removed)
example:
http://agc001.contest.atcoder.jp/tasks/agc001_a
input format:
N
L_1 L_2 \dots L_{2N}
generated code:
int N; cin >> N;
vector<int> L(2*N); REPEAT (i,2*N) cin >> L[i];
tips:
in Vim, the command "r! oj g/s -s http://agc001.contest.atcoder.jp/tasks/agc001_a" inserts above generated code.
in Emacs, the command "C-u M-! oj g/s -s http://agc001.contest.atcoder.jp/tasks/agc001_a" does.
I recommend that map it to some command, like "nnoremap <space>gs :r! oj generate-scanner --silent --repeat-macro=repeat ",
and use some plugin for templates, e.g. https://github.com/thinca/vim-template.
""",
)
subparser.add_argument("--repeat-macro",
help="use repeat macro with given name")
subparser.add_argument("--scanf",
action="store_true",
help="use scanf instead of cin")
subparser.add_argument("-s", "--silent", action="store_true")
subparser.add_argument("url")
# generate output
subparser = subparsers.add_parser(
"generate-output",
aliases=["g/o"],
help="generate output files form 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="./a.out",
help='your solution to be tested. (default: "./a.out")',
)
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",
default="test",
help="a directory name for test cases (default: test/)",
)
subparser.add_argument(
"test",
nargs="*",
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)',
)
# split input
subparser = subparsers.add_parser(
"split-input",
aliases=["s/i"],
help=
"split a input file which contains many cases, using your implementation",
formatter_class=argparse.RawTextHelpFormatter,
epilog="""\
format string for --output:
%i index
command for --command:
Specify a command which outputs something after reading each case.
It can be easily made from your solution for the problem.
example:
#include <iostream>
using namespace std;
int main() {
while (true) {
int v, e; cin >> v >> e;
if (v == 0 and e == 0) break;
for (int i = 0; i < e; ++ i) {
int v, w; char c; cin >> v >> w >> c;
}
cout << "foo" << endl;
}
return 0;
}
example:
if original input is below, it consists of two cases:
4 2
0 1 A
1 2 B
6 6
0 1 A
0 2 A
0 3 B
0 4 A
1 2 B
4 5 C
0 0
then the result with appropriate options:
4 2
0 1 A
1 2 B
0 0
ans
6 6
0 1 A
0 2 A
0 3 B
0 4 A
1 2 B
4 5 C
0 0
""",
)
subparser.add_argument(
"-c",
"--command",
default="./a.out",
help='your solution to be tested. (default: "./a.out")',
)
subparser.add_argument("-i",
"--input",
metavar="PATH",
required=True,
help="input file (required)")
subparser.add_argument(
"-o",
"--output",
metavar="FORMAT",
required=True,
help="output path (required)",
)
subparser.add_argument(
"-t",
"--time",
metavar="SECOND",
default=0.1,
type=float,
help="the interval between two cases",
)
subparser.add_argument(
"--ignore",
metavar="N",
default=0,
type=int,
help="ignore initial N lines of input",
)
subparser.add_argument("--header",
help="put a header string to the output")
subparser.add_argument("--footer",
help="put a footer string to the output")
subparser.add_argument(
"--auto-footer",
action="store_const",
const=split_input_auto_footer,
dest="footer",
help="use the original last line as a footer",
)
# test reactive
subparser = subparsers.add_parser(
"test-reactive",
aliases=["t/r"],
help="test for reactive problem",
formatter_class=argparse.RawTextHelpFormatter,
epilog="""\
""",
)
subparser.add_argument(
"-c",
"--command",
default="./a.out",
help='your solution to be tested. (default: "./a.out")',
)
subparser.add_argument("judge", help="judge program using standard I/O")
# code statistics
subparser = subparsers.add_parser(
"code-statistics",
aliases=["c/s"],
help="print the code statistics used in Anarchy Golf",
formatter_class=argparse.RawTextHelpFormatter,
epilog="""\
""",
)
subparser.add_argument("file")
# get standings
subparser = subparsers.add_parser(
"get-standings",
help="get and print the standings",
formatter_class=argparse.RawTextHelpFormatter,
epilog="""\
supported services:
TopCoder (Marathon Match)
""",
)
subparser.add_argument("url")
subparser.add_argument(
"-f",
"--format",
choices=["csv", "tsv", "json"],
default="tsv",
help="default: tsv",
)
args = parser.parse_args(args=args)
# logging
log_level = log.logging.INFO
if args.verbose:
log_level = log.logging.DEBUG
log.setLevel(log_level)
handler = log.logging.StreamHandler(sys.stderr)
handler.setLevel(log_level)
log.addHandler(handler)
log.debug("args: %s", str(args))
if args.subcommand in ["download", "d", "dl"]:
download(args)
elif args.subcommand in ["login", "l"]:
login(args)
elif args.subcommand in ["submit", "s"]:
submit(args)
elif args.subcommand in ["test", "t"]:
test(args)
elif args.subcommand in ["test-reactive", "t/r"]:
test_reactive(args)
elif args.subcommand in ["generate-scanner", "g/s"]:
generate_scanner(args)
elif args.subcommand in ["generate-output", "g/o"]:
generate_output(args)
elif args.subcommand in ["split-input", "s/i"]:
split_input(args)
elif args.subcommand in ["code-statistics", "c/s"]:
code_statistics(args)
elif args.subcommand == "get-standings":
get_standings(args)
else:
parser.print_help(file=sys.stderr)
def main(args=None):
# argparse
parser = argparse.ArgumentParser(
description='Tools for online judge services')
parser.add_argument('-v', '--verbose', action='store_true')
default_cookie_path = os.path.join(default_data_dir, 'cookie.jar')
parser.add_argument(
'-c',
'--cookie',
default=default_cookie_path,
help='path for cookie. (default: {})'.format(default_cookie_path))
parser.add_argument('-x', '--extra-option', action='append', default=[])
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
AtCoder
Codeforces
HackerRank
Yukicoder
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
extra opitons via -x:
-x all for yukicoder, use "テストケース一括ダウンロード"
''')
subparser.add_argument('url')
subparser.add_argument('-f',
'--format',
help='a format string to specify paths of cases')
subparser.add_argument('--overwrite', action='store_true')
subparser.add_argument('-n',
'--dry-run',
action='store_true',
help='don\'t write to files')
# login
subparser = subparsers.add_parser(
'login',
aliases=['l'],
help='login to a service',
formatter_class=argparse.RawTextHelpFormatter,
epilog='''\
supported services:
AtCoder
Codeforces
HackerRank
Yukicoder
extra opitons via -x:
-x method=github for yukicoder, login via github
-x method=twitter for yukicoder, login via github (not implementated yet)
''')
subparser.add_argument('url')
subparser.add_argument('-u', '--username')
subparser.add_argument('-p', '--password')
# submit
subparser = subparsers.add_parser(
'submit',
aliases=['s'],
help='submit your solution',
formatter_class=argparse.RawTextHelpFormatter,
epilog='''\
supported services:
AtCoder
Yukicoder
''')
subparser.add_argument('url')
subparser.add_argument('file')
subparser.add_argument('-l', '--language')
subparser.add_argument('--open',
nargs='?',
const=True,
help='open the result page after submission')
subparser.add_argument('-w',
'--wait',
metavar='SECCOND',
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:
You can do similar things with shell: e.g. `for f in test/*.in ; do echo $f ; diff <(./a.out < $f) ${f/.in/.out} ; done`
''')
subparser.add_argument(
'-c',
'--command',
default='./a.out',
help='your solution to be tested. (default: "./a.out")')
subparser.add_argument(
'--shell',
action='store_true',
help='use the --command as a shellscript instead of a path')
subparser.add_argument(
'-f',
'--format',
default='test/%s.%e',
help=
'a format string to recognize the relationship of test cases. (default: "test/%%s.%%e")'
)
subparser.add_argument(
'-m',
'--mode',
choices=['all', 'line'],
default='all',
help='mode to check an output with the correct answer. (default: all)')
subparser.add_argument('-1',
'--line',
dest='mode',
action='store_const',
const='line',
help='equivalent to --mode line')
subparser.add_argument('--rstrip',
action='store_true',
help='rstrip output before comapre')
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)
subparser.add_argument(
'test',
nargs='*',
help='paths of test cases. (if empty: globbed from --format)')
# generate scanner
subparser = subparsers.add_parser(
'generate-scanner',
aliases=['g/s'],
help='generate input scanner (experimental)',
formatter_class=argparse.RawTextHelpFormatter,
epilog='''\
supported services:
AtCoder
Yukicoder
HackerRank
example:
http://agc001.contest.atcoder.jp/tasks/agc001_a
input format:
N
L_1 L_2 \dots L_{2N}
generated code:
int N; cin >> N;
vector<int> L(2*N); REPEAT (i,2*N) cin >> L[i];
tips:
in Vim, the command "r! oj g/s -s http://agc001.contest.atcoder.jp/tasks/agc001_a" inserts above generated code.
in Emacs, the command "C-u M-! oj g/s -s http://agc001.contest.atcoder.jp/tasks/agc001_a" does.
I recommend that map it to some command, like "nnoremap <space>gs :r! oj generate-scanner --silent --repeat-macro=repeat ",
and use some plugin for templates, e.g. https://github.com/thinca/vim-template.
''')
subparser.add_argument('--repeat-macro',
help='use repeat macro with given name')
subparser.add_argument('--scanf',
action='store_true',
help='use scanf instead of cin')
subparser.add_argument('-s', '--silent', action='store_true')
subparser.add_argument('url')
# generate output
subparser = subparsers.add_parser(
'generate-output',
aliases=['g/o'],
help='generate output files form 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 ./aout < $f > ${f/.in/.out} ; done`
''')
subparser.add_argument(
'-c',
'--command',
default='./a.out',
help='your solution to be tested. (default: "./a.out")')
subparser.add_argument(
'--shell',
action='store_true',
help='use the --command as a shellscript instead of a path')
subparser.add_argument(
'-f',
'--format',
default='test/%s.%e',
help=
'a format string to recognize the relationship of test cases. (default: "test/%%s.%%e")'
)
subparser.add_argument(
'test',
nargs='*',
help='paths of input cases. (if empty: globbed from --format)')
# split input
subparser = subparsers.add_parser(
'split-input',
aliases=['s/i'],
help=
'split a input file which contains many cases, using your implementation',
formatter_class=argparse.RawTextHelpFormatter,
epilog='''\
format string for --output:
%i index
command for --command:
Specify a command which outputs something after reading each case.
It can be easily made from your solution for the problem.
exampe:
#include <iostream>
using namespace std;
int main() {
while (true) {
int v, e; cin >> v >> e;
if (v == 0 and e == 0) break;
for (int i = 0; i < e; ++ i) {
int v, w; char c; cin >> v >> w >> c;
}
cout << "foo" << endl;
}
return 0;
}
example:
if original input is below, it consists of two cases:
4 2
0 1 A
1 2 B
6 6
0 1 A
0 2 A
0 3 B
0 4 A
1 2 B
4 5 C
0 0
then the result with appropriate options:
4 2
0 1 A
1 2 B
0 0
ans
6 6
0 1 A
0 2 A
0 3 B
0 4 A
1 2 B
4 5 C
0 0
''')
subparser.add_argument(
'-c',
'--command',
default='./a.out',
help='your solution to be tested. (default: "./a.out")')
subparser.add_argument(
'--shell',
action='store_true',
help='use the --command as a shellscript instead of a path')
subparser.add_argument('-i',
'--input',
metavar='PATH',
required=True,
help='input file (required)')
subparser.add_argument('-o',
'--output',
metavar='FORMAT',
required=True,
help='output path (required)')
subparser.add_argument('-t',
'--time',
metavar='SECOND',
default=0.1,
type=float,
help='the interval between two cases')
subparser.add_argument('--ignore',
metavar='N',
default=0,
type=int,
help='ignore initial N lines of input')
subparser.add_argument('--header',
help='put a header string to the output')
subparser.add_argument('--footer',
help='put a footer string to the output')
subparser.add_argument('--auto-footer',
action='store_const',
const=split_input_auto_footer,
dest='footer',
help='use the original last line as a footer')
# test reactive
subparser = subparsers.add_parser(
'test-reactive',
aliases=['t/r'],
help='test for reactive problem',
formatter_class=argparse.RawTextHelpFormatter,
epilog='''\
''')
subparser.add_argument(
'-c',
'--command',
default='./a.out',
help='your solution to be tested. (default: "./a.out")')
subparser.add_argument(
'--shell',
action='store_true',
help='use the judge and --command as a shellscript instead of a path')
subparser.add_argument('judge', help='judge program using standard I/O')
args = parser.parse_args(args=args)
# logging
log_level = log.logging.INFO
if args.verbose:
log_level = log.logging.DEBUG
log.setLevel(log_level)
handler = log.logging.StreamHandler(sys.stderr)
handler.setLevel(log_level)
log.addHandler(handler)
log.debug('args: %s', str(args))
if args.subcommand in ['download', 'd', 'dl']:
download(args)
elif args.subcommand in ['login', 'l']:
login(args)
elif args.subcommand in ['submit', 's']:
submit(args)
elif args.subcommand in ['test', 't']:
test(args)
elif args.subcommand in ['test-reactiv', 't/r']:
test_reactive(args)
elif args.subcommand in ['generate-scanner', 'g/s']:
generate_scanner(args)
elif args.subcommand in ['generate-output', 'g/o']:
generate_output(args)
elif args.subcommand in ['split-input', 's/i']:
split_input(args)
else:
parser.print_help(file=sys.stderr)
sys.exit(1)