Merge pull request #624 from kmyk/feature/special-yosupo

add an internal method LibraryCheckerProblem.download_checker_cpp()

onlinejudge/service/library_checker.py

@@ -4,6 +4,7 @@
 """
 
 import os
+import pathlib
 import re
 import subprocess
 import sys
@@ -57,10 +58,10 @@ def download_system_cases(self, *, session: Optional[requests.Session] = None) -
         files += [(file.name, file.read_bytes()) for file in path.glob('out/*.out')]
         return onlinejudge._implementation.testcase_zipper.extract_from_files(iter(files))
 
-    def _get_cloned_repository_path(self):
+    def _get_cloned_repository_path(self) -> pathlib.Path:
         return utils.cache_dir / 'library-checker-problems'
 
-    def _generate_test_cases_in_cloned_repository(self):
+    def _generate_test_cases_in_cloned_repository(self) -> None:
         path = self._get_cloned_repository_path()
 
         try:
@@ -93,7 +94,7 @@ def _generate_test_cases_in_cloned_repository(self):
                 log.error("the generate.py failed: check https://github.com/yosupo06/library-checker-problems/issues")
                 raise
 
-    def _get_problem_directory_path(self):
+    def _get_problem_directory_path(self) -> pathlib.Path:
         path = self._get_cloned_repository_path()
         problems = toml.load(path / 'problems.toml')
         return path / problems['problems'][self.problem_id]['dir']
@@ -115,6 +116,12 @@ def from_url(cls, url: str) -> Optional['LibraryCheckerProblem']:
                 return cls(problem_id=m.group(1))
         return None
 
+    def download_checker_cpp(self) -> bytes:
+        self._generate_test_cases_in_cloned_repository()
+        path = self._get_problem_directory_path()
+        with open(str(path / "checker.cpp"), "rb") as fh:
+            return fh.read()
+
 
 onlinejudge.dispatch.services += [LibraryCheckerService]
 onlinejudge.dispatch.problems += [LibraryCheckerProblem]