From f9dd051ec90c9fc17d2cc1221d43e6549427b915 Mon Sep 17 00:00:00 2001 From: Ross Lawley Date: Tue, 27 Nov 2012 14:02:50 +0000 Subject: [PATCH] Merged get_document fix --- mongoengine/base/common.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/mongoengine/base/common.py b/mongoengine/base/common.py index 82728d1e..dc43d405 100644 --- a/mongoengine/base/common.py +++ b/mongoengine/base/common.py @@ -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: