我有以下型号:
class Comp(models.Model):
....
is_active = models.BooleanField(default=False)
class Item(models.Model):
comp = models.ForeignKey(Comp, blank=True, null=True, related_name='items', on_delete=models.CASCADE)
romp = models.ForeignKey(Romp, blank=True, null=True, related_name='items', on_delete=models.CASCADE)
is_active = models.BooleanField(default=False)
我想编写一个通用的模型clean
方法,以检查模型的non Null
外键,如果它们的is_active=True
我想在多个模型(使用抽象模型)上继承此方法
在下面的代码中,我检查了FK的常规列表中的每个模型属性
def clean(self, *args, **kwargs):
....
for attr in FK_LIST:
if hasattr(self, attr):
fk_obj = getattr(self, attr)
if not fk_obj.active:
raise ValidationError({'is_active': 'The {} {} needs to be active first'
.format(type(fk_obj).__name__, fk_obj.name)})
我的代码有2个问题:
答案 0 :(得分:1)
您可以使用self._meta.fields
获取列列表。这将产生带有字段的元组,值是列定义。
我们可以迭代此列,并检查类型是否为ForeignKey
。在这种情况下,我们执行检查:
def clean(self, *args, **kwargs):
# ...
for field in self._meta.fields():
if isinstance(field, ForeignKey):
fk_obj = getattr(self, field.name)
if not fk_obj.active:
raise ValidationError({'is_active': 'The {} {} needs to be active first'
.format(type(fk_obj).__name__, fk_obj.name)})
或者我们可以过滤掉非继承的
def clean(self, *args, **kwargs):
# ...
for field in self._meta.get_fields(include_parents=False):
if isinstance(field, ForeignKey):
fk_obj = getattr(self, field.name)
if not fk_obj.active:
raise ValidationError({'is_active': 'The {} {} needs to be active first'
.format(type(fk_obj).__name__, fk_obj.name)})