wip

django_q/exceptions.py

@@ -0,0 +1,23 @@
+import signal
+
+class TimeoutException(SystemExit):
+    """Exception for when a worker takes too long to complete a task"""
+    pass
+
+
+class TimeoutHandler:
+    def __init__(self, timeout: int):
+        self._timeout = timeout
+
+    def raise_timeout_exception(self, signum, frame):
+        raise TimeoutException('Task exceeded maximum timeout value '
+                              '({0} seconds)'.format(self._timeout))
+
+    def __enter__(self):
+        signal.signal(signal.SIGALRM, self.raise_timeout_exception)
+        signal.alarm(self._timeout)
+
+    def __exit__(self, exc_type, exc_value, traceback):
+        """When getting out of the timeout, reset the alarm, so it won't trigger"""
+        signal.alarm(0)
+        signal.signal(signal.SIGALRM, signal.SIG_DFL)