fork/django-q2
1
0
Fork 0
mirror of https://github.com/django-q2/django-q2.git synced 2026-10-01 12:28:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

More refactoring

Browse Source
This commit is contained in:
GDay
2023-03-31 02:09:54 +02:00
parent a694a9f53c
commit e3517bd4c6
16 changed files with 715 additions and 750 deletions
Download Patch File Download Diff File Expand all files Collapse all files
django_q/exceptions.py
+4 -1
View File
@@ -1,4 +1,5 @@
import signal
from typing import Optional
class TimeoutException(SystemExit):
"""Exception for when a worker takes too long to complete a task"""
@@ -6,7 +7,7 @@ class TimeoutException(SystemExit):
class TimeoutHandler:
def __init__(self, timeout: int):
def __init__(self, timeout: Optional[int] = None):
self._timeout = timeout
def raise_timeout_exception(self, signum, frame):
@@ -14,6 +15,8 @@ class TimeoutHandler:
'({0} seconds)'.format(self._timeout))
def __enter__(self):
if self._timeout is None:
return
signal.signal(signal.SIGALRM, self.raise_timeout_exception)
signal.alarm(self._timeout)