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

Fix validate_model_fields(), caused errors when metaclass raises errors

Browse Source
This commit is contained in:
Diederik van der Boor
2014-10-29 18:58:54 +01:00
parent aead57c037
commit 4bddac7c70
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
polymorphic/base.py
+4 -3
View File
@@ -200,12 +200,13 @@ class PolymorphicModelBase(ModelBase):
del(meta.app_label) del(meta.app_label)
return new_class return new_class
def validate_model_fields(self): @classmethod
def validate_model_fields(self, new_class):
"check if all fields names are allowed (i.e. not in POLYMORPHIC_SPECIAL_Q_KWORDS)" "check if all fields names are allowed (i.e. not in POLYMORPHIC_SPECIAL_Q_KWORDS)"
for f in self._meta.fields: for f in new_class._meta.fields:
if f.name in POLYMORPHIC_SPECIAL_Q_KWORDS: if f.name in POLYMORPHIC_SPECIAL_Q_KWORDS:
e = 'PolymorphicModel: "%s" - field name "%s" is not allowed in polymorphic models' e = 'PolymorphicModel: "%s" - field name "%s" is not allowed in polymorphic models'
raise AssertionError(e % (self.__name__, f.name)) raise AssertionError(e % (new_class.__name__, f.name))
@classmethod @classmethod
def validate_model_manager(self, manager, model_name, manager_name): def validate_model_manager(self, manager, model_name, manager_name):