Initial commit

Author: infmagic2047

.coveragerc

@@ -0,0 +1,2 @@
+[run]
+source = limejudge

.gitignore

@@ -0,0 +1,88 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+env/
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+*.egg-info/
+.installed.cfg
+*.egg
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*,cover
+.hypothesis/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+
+# Flask instance folder
+instance/
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+target/
+
+# IPython Notebook
+.ipynb_checkpoints
+
+# pyenv
+.python-version
+
+# celery beat schedule file
+celerybeat-schedule
+
+# dotenv
+.env
+
+# virtualenv
+venv/
+ENV/
+
+# Spyder project settings
+.spyderproject
+
+# Rope project settings
+.ropeproject

LICENSE

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2015 Yutao Yuan
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

MANIFEST.in

@@ -0,0 +1,2 @@
+include LICENSE
+recursive-include tests *.py

README.rst

@@ -0,0 +1,13 @@
+Simple judge tool for OI contests
+=================================
+
+LimeJudge is a simple OI judge for Linux.
+
+You will need Python 3.4+, PyYAML, docopt and termcolor to run this
+program.
+
+Warning: This program doesn't have any sandbox. You should only run it
+with trusted code.
+
+It doesn't work well with programs that use multiple threads or spawn
+other processes.

limejudge/__init__.py

@@ -0,0 +1,6 @@
+"""Simple OI judge tool."""
+
+
+# Version information. The following line must be in this format. It is
+# read in setup.py to get the version string.
+__version__ = '0.0.9'

limejudge/__main__.py

@@ -0,0 +1,9 @@
+"""__main__ wrapper for limejudge."""
+
+import sys
+
+from limejudge.cmdline import main
+
+
+if __name__ == '__main__':
+    sys.exit(main())

limejudge/anscompare.py

@@ -0,0 +1,66 @@
+"""Compare program output and provided answer."""
+
+import subprocess
+
+
+def compare(comp_type, infile, outfile, ansfile, full_score):
+    """Compare answers using comparison type comp_type.
+
+    infile is the input file.
+    outfile is the contestant's output file.
+    ansfile is the standard output file.
+    full_score is the maximum score.
+    """
+    if comp_type == 'no-extra-ws':
+        if _compare_no_extra_ws(outfile, ansfile):
+            return full_score
+        else:
+            return 0
+    elif comp_type.startswith('special-judge:'):
+        spj_name = comp_type[len('special-judge:'):]
+        return _compare_spj(spj_name, infile, outfile, ansfile,
+                            full_score)
+    else:
+        raise ValueError('Unknown comparison type: ' + repr(comp_type))
+
+
+def _compare_no_extra_ws(out1, out2):
+    """Compare file out1 and out2. Return true if two files match.
+
+    This function ignores trailing whitespace in every line and trailing
+    blank lines.
+    """
+    try:
+        with open(out1) as f1, open(out2) as f2:
+            return _compare_files_no_extra_ws(f1, f2)
+    except FileNotFoundError:
+        return False
+
+
+def _compare_files_no_extra_ws(f1, f2):
+    while True:
+        l1 = f1.readline()
+        l2 = f2.readline()
+        if l1 == '' or l2 == '':
+            # One of the files ends
+            break
+        if l1.rstrip() != l2.rstrip():
+            return False
+    while l1 != '' or l2 != '':
+        if l1.rstrip() != '' or l2.rstrip() != '':
+            # One of the files has a non-empty line
+            return False
+        l1 = f1.readline()
+        l2 = f2.readline()
+    return True
+
+
+def _compare_spj(spj_name, infile, outfile, ansfile, full_score):
+    proc = subprocess.Popen([spj_name, infile, outfile, ansfile,
+                             str(full_score)],
+                            stdout=subprocess.PIPE)
+    proc_out, _ = proc.communicate()
+    retcode = proc.poll()
+    if retcode:
+        return 0
+    return int(proc_out)

limejudge/cmdline.py

@@ -0,0 +1,37 @@
+"""Simple judge tool for OI contests.
+
+Usage:
+    limejudge <command> [<args>...]
+    limejudge --help
+    limejudge --version
+
+Available commands:
+    autocreate    Create a contest configuration file automatically.
+    resourcetest  Test the time and memory a program uses.
+    runjudge      Judge the contest.
+
+Options:
+    -h, --help  Print this help message and exit.
+    --version   Print version information and exit.
+"""
+
+import sys
+
+from docopt import docopt
+
+from limejudge import __version__
+from limejudge.commands import UnknownCommandError, run_command_by_name
+
+
+def main(argv=None):
+    if argv is None:
+        argv = sys.argv[1:]
+    args = docopt(__doc__, argv=argv,
+                  version='limejudge ' + __version__,
+                  options_first=True)
+    cmd_argv = [args['<command>']] + args['<args>']
+    try:
+        return run_command_by_name(args['<command>'], cmd_argv)
+    except UnknownCommandError as err:
+        print(str(err), file=sys.stderr)
+        return 1

limejudge/commands/__init__.py

@@ -0,0 +1,17 @@
+"""Subcommands for limejudge."""
+
+
+class UnknownCommandError(ValueError):
+    """Command name is unknown."""
+
+
+def run_command_by_name(name, argv=None):
+    if name == 'autocreate':
+        from limejudge.commands.autocreate import main
+    elif name == 'resourcetest':
+        from limejudge.commands.resourcetest import main
+    elif name == 'runjudge':
+        from limejudge.commands.runjudge import main
+    else:
+        raise UnknownCommandError('Unknown command name: ' + repr(name))
+    return main(argv)

limejudge/commands/autocreate.py

@@ -0,0 +1,121 @@
+"""Create a contest configuration file automatically.
+
+Usage:
+    limejudge autocreate [options]
+    limejudge autocreate --help
+
+Options:
+    -c, --contest=<file>  Write contest configuration to this file.
+                          [default: contest.yaml]
+    -f, --force           Force writing contest configuration even if
+                          the file already exists.
+    --cflags=<flags>      Use extra C compiler flags for this contest.
+    --cxxflags=<flags>    Use extra C++ compiler flags for this contest.
+    -h, --help            Print this help message and exit.
+"""
+
+import os
+import sys
+from glob import glob
+
+import yaml
+from docopt import docopt
+
+
+def ask_resource_limits(probname):
+    time_limit = input(
+        'What is the time limit (in s) for "' + probname + '"? ')
+    time_limit = float(time_limit)
+    if int(time_limit) == time_limit:
+        time_limit = int(time_limit)
+    memory_limit = input(
+        'What is the memory limit (in MB) for "' + probname + '"? ')
+    memory_limit = int(memory_limit) * 1024 * 1024
+    return {
+        'time-limit': time_limit,
+        'memory-limit': memory_limit,
+    }
+
+
+def find_contestants(basedir):
+    source_dir = os.path.join(basedir, 'source')
+    contestants = []
+    for name in sorted(os.listdir(source_dir)):
+        if os.path.isdir(os.path.join(source_dir, name)):
+            contestants.append({
+                'name': name,
+                'path': os.path.join('source', name),
+            })
+    return contestants
+
+
+def find_testcases(basedir, probname):
+    resource_limits = ask_resource_limits(probname)
+    prob_dir = os.path.join(basedir, 'data', probname)
+    has_in = {os.path.splitext(os.path.basename(x))[0]
+              for x in glob(os.path.join(prob_dir, '*.in'))
+              if os.path.isfile(x)}
+    has_out = {os.path.splitext(os.path.basename(x))[0]
+               for x in glob(os.path.join(prob_dir, '*.out'))
+               if os.path.isfile(x)}
+    testcases = []
+    prob_reldir = os.path.join('data', probname)
+    for testdata in sorted(has_in & has_out):
+        testcases.append({
+            'input-file': os.path.join(prob_reldir, testdata + '.in'),
+            'output-file': os.path.join(prob_reldir, testdata + '.out'),
+            'resource-limits': resource_limits,
+        })
+    if not testcases:
+        return []
+    testcase_cnt = len(testcases)
+    full_score_each = 100 // testcase_cnt
+    for testcase in testcases:
+        testcase['full-score'] = full_score_each
+    return testcases
+
+
+def find_problems(basedir):
+    data_dir = os.path.join(basedir, 'data')
+    problems = []
+    for name in sorted(os.listdir(data_dir)):
+        if os.path.isdir(os.path.join(data_dir, name)):
+            testcases = find_testcases(basedir, name)
+            problems.append({
+                'type': 'normal',
+                'name': name,
+                'input-file': name + '.in',
+                'output-file': name + '.out',
+                'source-path': name,
+                'testcases': testcases,
+            })
+    return problems
+
+
+def main(argv=None):
+    if argv is None:
+        argv = sys.argv[1:]
+    args = docopt(__doc__, argv=argv)
+    if os.path.exists(args['--contest']):
+        if os.path.isdir(args['--contest']):
+            print('Not overwriting directory.', file=sys.stderr)
+            return 1
+        elif not args['--force']:
+            print('Not overwriting existing file.', file=sys.stderr)
+            return 1
+    cwd = os.getcwd()
+    contest_data = {
+        'metadata': {'title': os.path.basename(cwd)},
+        'contestants': find_contestants(cwd),
+        'problems': find_problems(cwd),
+    }
+    compiler_flags = {}
+    if args['--cflags'] is not None:
+        compiler_flags['c'] = args['--cflags'].strip()
+    if args['--cxxflags'] is not None:
+        compiler_flags['cpp'] = args['--cxxflags'].strip()
+    if compiler_flags:
+        for problem in contest_data['problems']:
+            problem['compiler-flags'] = compiler_flags
+    with open(args['--contest'], 'w') as fout:
+        fout.write(yaml.safe_dump(contest_data))