fixing thread instead of fork

Author: leodisarli

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "PDFSentinel"
-version = "3.0.0"
+version = "3.0.1"
 authors = [
   { name="Not Empty Foundation", email="dev@not-empty.org" },
 ]

src/pdfsentinel/sentinel.py

@@ -1,5 +1,6 @@
 import json
 import multiprocessing as mp
+import threading
 from queue import Empty
 from pathlib import Path
 
@@ -245,16 +246,15 @@ def _run_with_timeout(self, worker, worker_args, timeout_seconds):
         if float(timeout_seconds) <= 0:
             raise ValueError("timeout_seconds must be greater than 0")
 
-        start_method = "fork" if "fork" in mp.get_all_start_methods() else "spawn"
+        if threading.active_count() > 1:
+            start_method = "spawn"
+        else:
+            start_method = "fork" if "fork" in mp.get_all_start_methods() else "spawn"
         ctx = mp.get_context(start_method)
         result_queue = ctx.Queue()
         process = ctx.Process(target=worker, args=(*worker_args, result_queue))
-        process.start()
-        process.join(float(timeout_seconds))
 
-        if process.is_alive():
-            process.terminate()
-            process.join()
+        if not self._finish_process_with_timeout(process, timeout_seconds):
             return None
 
         try:
@@ -267,6 +267,18 @@ def _run_with_timeout(self, worker, worker_args, timeout_seconds):
 
         return payload["result"]
 
+    @staticmethod
+    def _finish_process_with_timeout(process, timeout_seconds):
+        process.start()
+        process.join(float(timeout_seconds))
+
+        if not process.is_alive():
+            return True
+
+        process.terminate()
+        process.join()
+        return False
+
     def _file_analysis_sync(self, file_path, config=None):
         cfg = self._merge_config(self.base_config, config or {})
         doc = pymupdf.open_document(file_path)