Merge pull request #356 from online-judge-tools/fix/timeout-for-python

Set timeout for the dependency analysis of Python files

Author: kmyk

onlinejudge_verify/languages/python.py

@@ -1,6 +1,8 @@
 # Python Version: 3.x
+import concurrent.futures
 import functools
 import pathlib
+import platform
 import sys
 import textwrap
 from logging import getLogger
@@ -53,7 +55,16 @@ def _python_list_depending_files(path: pathlib.Path, basedir: pathlib.Path) -> L
         importlab.fs.Path([importlab.fs.OSFileSystem(str(basedir.resolve()))]),
         (sys.version_info.major, sys.version_info.minor),
     )
-    res_graph = importlab.graph.ImportGraph.create(env, [str(path)])
+    try:
+        executor = concurrent.futures.ThreadPoolExecutor()
+        future = executor.submit(importlab.graph.ImportGraph.create, env, [str(path)])
+        if platform.uname().system == 'Windows':
+            timeout = 5.0  # 1.0 sec causes timeout on CI using Windows
+        else:
+            timeout = 1.0
+        res_graph = future.result(timeout=timeout)
+    except concurrent.futures.TimeoutError as e:
+        raise RuntimeError(f"Failed to analyze the dependency graph (timeout): {path}") from e
     try:
         node_deps_pairs = res_graph.deps_list()  # type: List[Tuple[str, List[str]]]
     except Exception as e: