Merge pull request #441 from Karmak23/patch-1

Fix the ._get_db() attribute after a Document.switch_db()
This commit is contained in:
Ross Lawley 2013-08-20 01:19:26 -07:00
commit 3622ebfabd

View File

@ -400,7 +400,7 @@ class Document(BaseDocument):
""" """
with switch_db(self.__class__, db_alias) as cls: with switch_db(self.__class__, db_alias) as cls:
collection = cls._get_collection() collection = cls._get_collection()
db = cls._get_db db = cls._get_db()
self._get_collection = lambda: collection self._get_collection = lambda: collection
self._get_db = lambda: db self._get_db = lambda: db
self._collection = collection self._collection = collection