Hopiu/django-ddp
1
0
Fork 0
mirror of https://github.com/jazzband/django-ddp.git synced 2026-03-16 22:40:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

get dddp to get correct collection and not throw an error

Browse source
This commit is contained in:
craig 2015-12-20 00:41:05 +11:00
parent 8cbd3f3389
commit d517e28e8e
1 changed files with 11 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
dddp/api.py
View file

@ -230,6 +230,8 @@ class CollectionMeta(APIMeta):
"""DDP Collection metaclass."""
model_name_map = {}
def __new__(mcs, name, bases, attrs):
"""Create a new Collection class."""
attrs.update(
@ -240,6 +242,10 @@ class CollectionMeta(APIMeta):
attrs.update(
name=model_name(model),
)
if model and attrs.get('name'):
if model._meta.label_lower:
model = model._meta.label_lower
CollectionMeta.model_name_map.update({model: attrs.get('name')})
return super(CollectionMeta, mcs).__new__(mcs, name, bases, attrs)
@ -605,7 +611,11 @@ class DDP(APIMixin):
def get_col_by_name(self, name):
"""Return collection instance for given name."""
return self._registry[COLLECTION_PATH_FORMAT.format(name=name)]
try:
return self._registry[COLLECTION_PATH_FORMAT.format(name=name)]
except:
name = Collection.model_name_map[str(name)]
return self._registry[COLLECTION_PATH_FORMAT.format(name=name)]
def get_pub_by_name(self, name):
"""Return publication instance for given name."""