TaskPool now supports initargs

celery/concurrency/processes/__init__.py

@@ -38,7 +38,7 @@ 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
@@ -46,6 +46,7 @@ class TaskPool(object):
         self.timeout = timeout
         self.soft_timeout = soft_timeout
         self.putlocks = putlocks
+        self.initargs = None
         self._pool = None
 
     def start(self):
@@ -56,6 +57,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)