From 09bfcb70e64fc50241fcedbebf775d91c18b7f76 Mon Sep 17 00:00:00 2001 From: Roman Date: Wed, 11 Apr 2012 21:21:11 +0400 Subject: connection option for MongoCache --- requests_cache/backends/mongo.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) (limited to 'requests_cache/backends/mongo.py') 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) -- cgit v1.2.1