Explorar o código

pass options to route_for_task

Ali Bozorgkhan %!s(int64=10) %!d(string=hai) anos
pai
achega
1b7a9f6187
Modificáronse 1 ficheiros con 3 adicións e 3 borrados
  1. 3 3
      celery/app/routes.py

+ 3 - 3
celery/app/routes.py

@@ -74,7 +74,7 @@ class Router(object):
     def route(self, options, task, args=(), kwargs={}):
         options = self.expand_destination(options)  # expands 'queue'
         if self.routes:
-            route = self.lookup_route(task, args, kwargs)
+            route = self.lookup_route(task, args, kwargs, options)
             if route:  # expands 'queue' in route.
                 return lpmerge(self.expand_destination(route), options)
         if 'queue' not in options:
@@ -102,8 +102,8 @@ class Router(object):
                         'Queue {0!r} missing from task_queues'.format(queue))
         return route
 
-    def lookup_route(self, task, args=None, kwargs=None):
-        return _first_route(self.routes, task, args, kwargs)
+    def lookup_route(self, task, args=None, kwargs=None, options=None):
+        return _first_route(self.routes, task, args, kwargs, options)
 
 
 def prepare(routes):