检查 Django 中是否存在 model 字段

Question

这是我在models.py中提出的解决方案：

from django.db import models

@classmethod
def model_field_exists(cls, field):
    try:
        cls._meta.get_field(field)
        return True
    except models.FieldDoesNotExist:
        return False

models.Model.field_exists = model_field_exists

并像这样使用它：

Post.field_exists('title') # > True or False

问题出在外键上，我的帖子 model 属于一个类别，此检查有效：

Post.field_exists('category') # > True

但是这个没有：

Post.field_exists('category_id') # > False

这是 db 中的实际字段名称，我需要像这样检查它。 如何在 django 中做到这一点？

Answer 1

hasattr（cls，field）在所有情况下都不起作用，例如使用mixins的情况。 安全测试是：

            try:
                model._meta.get_field(field)
                .. do stuff here
            except FieldDoesNotExist:
                pass

要么

            field = None
            try:
                field = model._meta.get_field(field)
            except FieldDoesNotExist:
                pass

            if field:
              .. do stuff here

Answer 2

如果该字段不存在，您可以使用getattr返回默认值。如下所示

getattr(table, field, False)

Answer 3

您的“category_id”字段是https://docs.djangoproject.com/en/1.10/ref/models/meta/术语中的隐藏字段。 get_field无法检索它。

也许循环遍历所有字段并检查其名称; get_fields可以返回隐藏字段：

for field in cls._meta.get_fields(include_hidden=True):
    if field.name == field:
        return True
return False

如果您需要关心其db_column已从其默认值更改的字段，您也可以测试field.db_column（但如果未明确设置则未设置）。

Answer 4

但是这个没有：

Post.field_exists（'category_id'）＃> False

这是db中的实际字段名称，我需要像这样检查它。 我怎么能在django做到这一点？

除了_id部分之外的所有事情的情况在另一个答案中得到了充分的回答。 为了处理_id案例， 除了适当处理这些案例的答案之外 ，你基本上还需要一个特殊的条件。 类似这样的东西：

if field_name.endswith('_id'):
    try:
        field = model._meta.get_field(field_name.strip('_id'))
    except FieldDoesNotExist:
        return False
    if not f.is_relation:
        return False
    return True

需要进行额外的检查以确保字段减去_id返回ForeignKey，而不是恰好具有相同名称的其他类型的字段，这意味着模型上的name_id字段无效。

Answer 5

你为什么需要这个？ 对象关系映射器的概念是用于描述/定义数据库的类。 因此，如果您想测试模型的类型，您应该使用isinstance如下所示：

if isinstance(model, Post):
    # model is a Post, Post has a publish()
    model.publish()
elif isinstance(model, Comment):
    # model is a Comment, Comment has a foo()
    model.foo()

您的模型定义定义了您的架构。 没有理由必须测试场的存在。 如果有，您可能没有通过使用setattr分配字段来遵循最佳实践？

Answer 6

YourModelName.objects.filter(YourModelField=TheFieldToCheck).exists()