Jelajahi Sumber

MongoDB: Pickle url with backend. Closes #2347

Ask Solem 10 tahun lalu
induk
melakukan
9e0eae688b
1 mengubah file dengan 8 tambahan dan 8 penghapusan
  1. 8 8
      celery/backends/mongodb.py

+ 8 - 8
celery/backends/mongodb.py

@@ -59,7 +59,7 @@ class MongoBackend(BaseBackend):
 
     _connection = None
 
-    def __init__(self, **kwargs):
+    def __init__(self, app=None, url=None, **kwargs):
         """Initialize MongoDB backend instance.
 
         :raises celery.exceptions.ImproperlyConfigured: if
@@ -68,7 +68,7 @@ class MongoBackend(BaseBackend):
         """
         self.options = {}
 
-        super(MongoBackend, self).__init__(**kwargs)
+        super(MongoBackend, self).__init__(app, **kwargs)
 
         if not pymongo:
             raise ImproperlyConfigured(
@@ -100,10 +100,10 @@ class MongoBackend(BaseBackend):
             self.options.setdefault('max_pool_size', self.max_pool_size)
             self.options.setdefault('auto_start_request', False)
 
-        url = kwargs.get('url')
-        if url:
+        self.url = url
+        if self.url:
             # Specifying backend as an URL
-            self.host = url
+            self.host = self.url
 
     def _get_connection(self):
         """Connect to the MongoDB server."""
@@ -232,9 +232,9 @@ class MongoBackend(BaseBackend):
         )
 
     def __reduce__(self, args=(), kwargs={}):
-        kwargs.update(
-            dict(expires=self.expires))
-        return super(MongoBackend, self).__reduce__(args, kwargs)
+        return super(MongoBackend, self).__reduce__(
+            args, dict(kwargs, expires=self.expires, url=self.url),
+        )
 
     def _get_database(self):
         conn = self._get_connection()