在其他模型中重用 Django 字段定义

Question

我在不同的模型中有几个领域。 例如

class A(models.Model):
    field_1 = models.CharField(max_length=5, blank=False)
    field_2 = models.CharField(max_length=100, blank=False)
    field_3 = models.CharField(max_length=200, blank=False)

class B(models.Model):
    field_4 = models.CharField(max_length=5, blank=False)
    field_5 = models.CharField(max_length=100, blank=False)
    field_6 = models.CharField(max_length=200, blank=False)

除了这些类之外，我还想创建一个潜在客户 class。 此 class 包含将来可能会复制到其他模型中的数据。 因此，数据约束是相同的，但由于这是可选数据，因此可能会丢失一些数据。

我想使用 DRY，但我也想更新空白字段。 关于如何使用这些原始类中的确切字段定义的任何提示？

class Lead(models.Model):
    field_1 = A.field_1  # but with updated blank field
    field_3 = B.field_3  # but with updated blank field

我考虑过 class 依赖项，但实际上我正在使用来自 5 个类的字段，如果前导 class 依赖于这些，我担心这些字段会变得难以理解。

Answer 1

Django ORM

由于 Django 在 Python 之上添加了一层，因此 class 依赖项的工作方式不同。 仅引用另一个非抽象 class 的字段定义并且不允许覆盖字段，因为它会破坏数据库或引入令人讨厌的错误。

解决方案

不要引用 function，而是使用内置的 function clone()复制它。

# Create a new function to gather the field definition from another model
def get_field_from_model(model, field):
    return model._meta.get_field(field)


def modify_field(base_model, base_field, **kwargs):
    field = get_field_from_model(base_model, base_field)
    field_copy = field.clone()
    field_copy.__dict__.update(kwargs)
    return field_copy

官方 _meta 隐藏了 class 但它被认为是稳定的并且是官方的 API 。 但是，我将它放在单独的 function 中，以便在这种情况发生变化时快速访问。 这种方法基于此处更详细的答案（解决方案类似，原始问题侧重于从变量中复制字段，而不是在 class 定义中）。

执行

现在这个 function 可以在 class 定义中使用。 它也在迁移步骤中被识别，并将基于引用的字段创建新字段。

class Lead(models.Model):
    field_1 = modify_field(A, 'field_1', blank = True)
    field_3 = modify_field(B, 'field_3', blank = True)