pub/mongoengine
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merged get_document fix

Browse Source
This commit is contained in:
Ross Lawley 2012-11-27 14:02:50 +00:00
parent 68e4a27aaf
commit f9dd051ec9
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mongoengine/base/common.py
View File

@ -9,11 +9,10 @@ _document_registry = {}
def get_document(name):
doc = _document_registry.get(name, None)
if not doc:
# Possible old style names
end = ".%s" % name
possible_match = [k for k in _document_registry.keys()
if k.endswith(end)]
if not doc and '.' in name:
# Possible old style name
end = name.split('.')[-1]
possible_match = [k for k in _document_registry.keys() if k == end]
if len(possible_match) == 1:
doc = _document_registry.get(possible_match.pop(), None)
if not doc: