Briq
/
django-polymorphic
6
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Also introduce a new exception type for invalid database IDs

fix_request_path_info
Diederik van der Boor 2017-08-01 11:40:55 +02:00
parent fb8eed78ad
commit 4835cd244e
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
polymorphic/models.py
View File

@ -21,6 +21,10 @@ class PolymorphicTypeUndefined(LookupError):
pass pass
class PolymorphicTypeInvalid(RuntimeError):
pass
class PolymorphicModel(six.with_metaclass(PolymorphicModelBase, models.Model)): class PolymorphicModel(six.with_metaclass(PolymorphicModelBase, models.Model)):
""" """
Abstract base class that provides polymorphic behaviour Abstract base class that provides polymorphic behaviour
@ -105,7 +109,7 @@ class PolymorphicModel(six.with_metaclass(PolymorphicModelBase, models.Model)):
if model is not None \ if model is not None \
and not issubclass(model, self.__class__) \ and not issubclass(model, self.__class__) \
and not issubclass(model, self.__class__._meta.proxy_for_model): and not issubclass(model, self.__class__._meta.proxy_for_model):
raise RuntimeError("ContentType {0} for {1} #{2} does not point to a subclass!".format( raise PolymorphicTypeInvalid("ContentType {0} for {1} #{2} does not point to a subclass!".format(
self.polymorphic_ctype_id, model, self.pk, self.polymorphic_ctype_id, model, self.pk,
)) ))
return model return model