如果传递了模型,则获取模型的FK属性

时间:2018-09-01 09:52:31

标签: django django-models

我有以下型号:

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个问题:

  1. 我有两个人手动维护FK名称列表
  2. 我需要遍历所有属性,而不是仅检查FK
  3. 我不想使用隐式FK_list进行签入,而是准备使用排除列表(易于维护): 检查模型上所有非NULL或Exclude_list中的FK

1 个答案:

答案 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)})