Completed tool chain

Author: AlfredChester

Lynx/__init__.py

@@ -19,10 +19,26 @@
 # SOFTWARE.
 
 import click
+from Lynx.utils.constants import VERSION
+
+
+def print_version(ctx, param, value):
+    if not value or ctx.resilient_parsing:
+        return
+    click.echo(VERSION)
+    ctx.exit()
 
 
 # register CLI group, must be placed before importing submodules
 @click.group()
+@click.option(
+    "--version",
+    is_flag=True,
+    is_eager=True,
+    expose_value=False,
+    callback=print_version,
+    help="Show the version of lynx.",
+)
 def cli():
     """lynx: Next generation testcase tool set for OI / ACM contests."""
     pass

Lynx/modules/init.py

@@ -28,7 +28,20 @@
     default="custom",
     help="Use a validator. You can also use the validator provided by testlib in a similar way as the checker.",
 )
-def init(problem, language, checker, validator):
+@click.option("--testcase", type=int, default=20, help="The number of testcases.")
+@click.option(
+    "--time-limit",
+    type=int,
+    default=1000,
+    help="The time limit of the problem, uses ms as unit.",
+)
+@click.option(
+    "--memory-limit",
+    type=int,
+    default=512,
+    help="The memory limit of the problem, uses MB as unit.",
+)
+def init(problem, language, checker, validator, testcase, time_limit, memory_limit):
     """This command initializes a lynx problem directory."""
     # Turn the problem path into an absolute path
     if not os.path.isabs(problem):
@@ -37,4 +50,8 @@ def init(problem, language, checker, validator):
     if os.path.exists(problem):
         error_and_exit(f"Problem directory {problem} already exists.")
     if language in CPP_STANDARDS:
-        cpp.init_problem(problem, language, checker, validator)
+        cpp.init_problem(
+            problem, language, checker, validator, testcase, time_limit, memory_limit
+        )
+    else:  # Other languages will be supported in the future
+        assert False  # This should never happen

Lynx/utils/config.py

@@ -1,9 +1,18 @@
 import yaml
+import os
 
 
 # Used to manage the range of variables more easily
 # It will be replaced by some features of cyaron in the future.
 class SubtaskManager: ...
 
 
-def read_config(file_path: str) -> dict: ...
+class CppProblemConfig:
+    def __init__(self, problem_path: str):
+        self.problem_path = problem_path
+        self.problem_config_path = os.path.join(problem_path, "problem.yaml")
+        self.testdata_config_path = os.path.join(
+            problem_path, "testdata", "config.yaml"
+        )
+
+    def init_configs(self, standard, testcase, time_limit, memory_limit): ...

Lynx/utils/constants.py

@@ -1,5 +1,7 @@
 import os
 
+VERSION = "2.0.1b2"
+
 # Some directories
 LYNX_ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
 TEMPLATES_ROOT = os.path.join(LYNX_ROOT, "templates")

Lynx/utils/language/cpp.py

@@ -1,8 +1,9 @@
 import os
 import shutil
 
-from Lynx.utils.constants import *
 from Lynx.utils.log import *
+from Lynx.utils.constants import *
+from Lynx.utils.config import CppProblemConfig
 
 """The structure of a problem directory is as follows:
 .
@@ -20,10 +21,24 @@
 """
 
 
-def init_problem(problem_path: str, standard: str, checker: str, validator: str):
+def init_problem(
+    problem_path: str,
+    standard: str,
+    checker: str,
+    validator: str,
+    testcase: int,
+    time_limit: int,
+    memory_limit: int,
+):
     shutil.copytree(CPP_TEMPLATES_ROOT, problem_path)
     shutil.copyfile(
         os.path.join(TEMPLATES_ROOT, "generator.py"),
         os.path.join(problem_path, "programs", "generator.py"),
     )
     os.remove(os.path.join(problem_path, "additional_file", ".gitkeep"))
+    if checker != "custom":
+        ...
+    if validator != "custom":
+        ...
+    config = CppProblemConfig(problem_path)
+    config.init_configs(standard, testcase, time_limit, memory_limit)

Lynx/utils/log.py

@@ -1,5 +1,4 @@
 import click
-from Lynx.utils.constants import VERBOSE
 
 
 def error(message: str) -> None:
@@ -15,5 +14,4 @@ def error_and_exit(message: str, code: int = 1) -> None:
 
 def debug(message: str) -> None:
     """Print a debug message."""
-    if VERBOSE:
-        click.echo(click.style(f"DEBUG: {message}", fg="blue"))
+    click.echo(click.style(f"DEBUG: {message}", fg="blue"))

Makefile

@@ -34,4 +34,7 @@ check: deps _build _check
 _build:
 	rm -fR dist/
 	python setup.py sdist build
-build: deps _build
+build: deps _build
+
+publish: build
+	twine upload dist/*

setup.py

@@ -1,11 +1,24 @@
 # !/usr/bin/env python
 from setuptools import setup, find_packages
+from Lynx.utils.constants import VERSION
 
 packages = find_packages()
 
+# CPP templates
+packages += [
+    "Lynx.templates",
+    "Lynx.templates.cpp",
+    "Lynx.templates.cpp.programs",
+    "Lynx.templates.cpp.testdata",
+    "Lynx.templates.cpp.additional_file",
+]
+
+# testlib
+packages += ["testlib", "testlib.checkers", "testlib.validators"]
+
 setup(
-    name="lynx",
-    version="2.0.1b0",
+    name="pyLynx",
+    version=VERSION,
     keywords="olympic informatics testcase testcase-tools tools",
     description="lynx: Next generation high performance testcase tool set for OI / ACM contests. - 新一代信息学竞赛数据生成工具集",
     license="MIT",
@@ -15,7 +28,7 @@
     packages=packages,
     include_package_data=True,
     platforms="any",
-    install_requires=["Click", "cyaron", "tqdm"],
+    install_requires=["Click", "cyaron", "tqdm", "PyYAML"],
     entry_points="""
         [console_scripts]
         lynx=Lynx:cli