Browse Source

TaskPool: Support taking the initargs argument to Pool()

Ask Solem 14 years ago
parent
commit
d60d544092
1 changed files with 3 additions and 1 deletions
  1. 3 1
      celery/concurrency/processes/__init__.py

+ 3 - 1
celery/concurrency/processes/__init__.py

@@ -36,10 +36,11 @@ class TaskPool(object):
 
     def __init__(self, limit, logger=None, initializer=None,
             maxtasksperchild=None, timeout=None, soft_timeout=None,
-            putlocks=True):
+            putlocks=True, initargs=None):
         self.limit = limit
         self.logger = logger or log.get_default_logger()
         self.initializer = initializer
+        self.initargs = initargs
         self.maxtasksperchild = maxtasksperchild
         self.timeout = timeout
         self.soft_timeout = soft_timeout
@@ -54,6 +55,7 @@ class TaskPool(object):
         """
         self._pool = self.Pool(processes=self.limit,
                                initializer=self.initializer,
+                               initargs=self.initargs,
                                timeout=self.timeout,
                                soft_timeout=self.soft_timeout,
                                maxtasksperchild=self.maxtasksperchild)