28 lines
929 B
Python
28 lines
929 B
Python
from mongoengine.errors import NotRegistered
|
|
|
|
__all__ = ('get_document', '_document_registry')
|
|
|
|
ALLOW_INHERITANCE = False # TODO is this really necessary?
|
|
|
|
_document_registry = {}
|
|
|
|
|
|
def get_document(name):
|
|
"""Get a document class by 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
|