connection option for MongoCache

1 files changed, 4 insertions, 2 deletions

diff --git a/requests_cache/backends/mongo.py b/requests_cache/backends/mongo.py
index 5e223b5..77dfd73 100644
--- a/requests_cache/backends/mongo.py
+++ b/requests_cache/backends/mongo.py
@@ -13,11 +13,13 @@ from requests_cache.backends.mongodict import MongoDict, MongoPickleDict
 class MongoCache(BaseCache):
     """ ``mongo`` cache backend.
     """
-    def __init__(self, db_name='requests-cache'):
+    def __init__(self, db_name='requests-cache', **options):
         """
         :param db_name: database name (default: ``'requests-cache'``)
+        :param connection: (optional) ``pymongo.Connection``
         """
         super(MongoCache, self).__init__()
-        self.responses = MongoPickleDict(db_name, 'responses')
+        self.responses = MongoPickleDict(db_name, 'responses',
+                                         options.get('connection'))
         self.url_map = MongoDict(db_name, 'urls', self.responses.connection)