Add WORKER_FUNC_MODIFIER feature

JereMalinen · Mar 18, 2019 · ab5a3da · ab5a3da
1 parent c84fd11
commit ab5a3da
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/django_q/cluster.py b/django_q/cluster.py
@@ -368,6 +368,16 @@ def worker(task_queue, result_queue, timer, timeout=Conf.TIMEOUT):
                 result = (e, False)
                 if error_reporter:
                     error_reporter.report()
+        if Conf.WORKER_FUNC_MODIFIER:
+            try:
+                module, func = f.rsplit('.', 1)
+                m = importlib.import_module(module)
+                modifier = getattr(m, func)
+                f = modifier(f)
+            except (ValueError, ImportError, AttributeError) as e:
+                result = (e, False)
+                if error_reporter:
+                    error_reporter.report()
         # We're still going
         if not result:
             db.close_old_connections()

diff --git a/django_q/conf.py b/django_q/conf.py
@@ -98,6 +98,8 @@ class Conf(object):
     # Option to undaemonize the workers and allow them to spawn child processes
     DAEMONIZE_WORKERS = conf.get('daemonize_workers', True)
 
+    WORKER_FUNC_MODIFIER = conf.get('worker_func_modifier', None)
+
     # Maximum number of tasks that each cluster can work on
     QUEUE_LIMIT = conf.get('queue_limit', int(WORKERS) ** 2)