Removed wrapt decorator

buriy · buriy · commit efebb4cdd90f · 2025-01-13T03:00:49.000+07:00
diff --git a/setup.py b/setup.py
@@ -9,14 +9,8 @@
      "cchardet",
 ]
 
-test_deps = [
-    # Test timeouts
-    "wrapt-timeout-decorator",
-]
-
 extras = {
     'speed': speed_deps,
-    'test': test_deps,
 }
 
 # Adapted from https://github.com/pypa/pip/blob/master/setup.py
@@ -54,7 +48,6 @@ def find_version(*file_paths):
         "lxml-html-clean; python_version < '3.11'",
         "cssselect"
     ],
-    tests_require=test_deps,
     extras_require=extras,
     classifiers=[
         "Environment :: Web Environment",
diff --git a/tests/test_article_only.py b/tests/test_article_only.py
@@ -1,8 +1,34 @@
 import os
+import time
 import unittest
 
 from readability import Document
-from wrapt_timeout_decorator import *
+from functools import wraps
+
+
+class TimeoutException(Exception):
+    """Exception raised when a function exceeds its time limit."""
+    pass
+
+
+def timeout(seconds):
+    """Decorator to enforce a timeout on function execution."""
+    def decorator(func):
+        @wraps(func)
+        def wrapper(*args, **kwargs):
+            start_time = time.perf_counter()
+            result = func(*args, **kwargs)
+            end_time = time.perf_counter()
+            elapsed_time = end_time - start_time
+            if elapsed_time > seconds:
+                raise TimeoutException(
+                    f"Function '{func.__name__}' exceeded time limit of {seconds} seconds "
+                    f"with an execution time of {elapsed_time:.4f} seconds"
+                )
+            return result
+        return wrapper
+    return decorator
+
 
 SAMPLES = os.path.join(os.path.dirname(__file__), "samples")
 
@@ -100,7 +126,7 @@ def test_correct_cleanup(self):
         assert not "aside" in s
 
     # Many spaces make some regexes run forever
-    @timeout(3, use_signals=False)
+    @timeout(3)
     def test_many_repeated_spaces(self):
         long_space = " " * 1000000
         sample = "<html><body><p>foo" + long_space + "</p></body></html>"
