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

Also introduce a new exception type for invalid database IDs

Browse Source
This commit is contained in:
Diederik van der Boor
2017-08-01 11:44:21 +02:00
parent fb8eed78ad
commit 4835cd244e
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
polymorphic/models.py
+5 -1
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