3425264077
Conflicts: AUTHORS docs/changelog.rst mongoengine/__init__.py mongoengine/base.py mongoengine/fields.py python-mongoengine.spec tests/test_document.py tests/test_fields.py tests/test_queryset.py
27 lines
876 B
Python
27 lines
876 B
Python
from mongoengine.errors import NotRegistered
|
|
|
|
__all__ = ('ALLOW_INHERITANCE', 'get_document', '_document_registry')
|
|
|
|
ALLOW_INHERITANCE = False
|
|
|
|
_document_registry = {}
|
|
|
|
|
|
def get_document(name):
|
|
doc = _document_registry.get(name, None)
|
|
if not doc:
|
|
# Possible old style name
|
|
single_end = name.split('.')[-1]
|
|
compound_end = '.%s' % single_end
|
|
possible_match = [k for k in _document_registry.keys()
|
|
if k.endswith(compound_end) or k == single_end]
|
|
if len(possible_match) == 1:
|
|
doc = _document_registry.get(possible_match.pop(), None)
|
|
if not doc:
|
|
raise NotRegistered("""
|
|
`%s` has not been registered in the document registry.
|
|
Importing the document class automatically registers it, has it
|
|
been imported?
|
|
""".strip() % name)
|
|
return doc
|