使用查询数据库的默认字段函数迁移时出现 Django 错误

Question

我正在尝试创建一个可调用的默认字段，如下所示：

from datetime import date, timedelta
from django.db import models

class Preferences(models.Model):
    expiration_days = models.IntegerField(default=30)
    ... other db settings here ...

class ImportantBusinessObject(models.Model):
    expiration_date = models.DateField(default=get_default_expiration_date)

def get_default_expiration_date():
    expiration_days = Preferences.objects.first().expiration_days
    return date.today() + timedelta(days=expiration_days)

这个想法是在数据库中存储了一组全局首选项，这些首选项会影响其他模型对象的默认值。

在向Preferences 类添加另一个字段之前，一切都很好。然后，在构建数据库时（例如，在运行测试时），添加 expiration_date 字段的迁移最终会在添加另一个首选项字段的新迁移之前执行。在调用get_default_expiration_date() 期间，Preferences 对象是最新版本，但数据库尚未获得最新迁移，我收到column preferences.new_preference does not exist 错误。

解决这个问题的最佳方法是什么？

现在我正在测试期间禁用迁移作为一种解决方法，但我想知道是否还有其他方法。

我尝试从get_default_expiration_date() 中的应用注册表中获取模型，但它仍然是最新的，而不是您在migrations.RunPython 函数中提供的版本。

我还尝试将获取 Preferences 对象的行包装在 try/except 块中，但数据库最终处于回滚状态，我无法继续。

其他位：

• 我使用的是 Django 1.8
• 在我的实际应用中，Preferences 类实际上是使用 django-solo 的 db 单例。

UPDATE：明确操作顺序

以下是正在发生的事情的时间表：

1. preferences/migrations/initial_0001.py：初始偏好，包括expiration_days
2. business/migrations/initial_0001.py：初始业务对象，依赖于preferences 0001
3. preferences/migrations/add_pref_0002.py：给Preferences添加一个新字段

一旦我在第 3 步中定义了迁移，事情就开始中断，因为在运行以从第 2 步添加迁移时，会调用get_default_expiration_date，这会引发错误。从下面关于更新依赖项的评论线程中，我的观点是，为了做到这一点，每次我添加新的首选项迁移时，我都需要更新先前定义的 business/migrations/initial_0001.py 的依赖项（以及任何其他行为类似的）。希望这可以稍微澄清一下情况。

Answer 1

您似乎遇到了文档here 中讨论的问题：

因为不可能序列化任意 Python 代码，所以这些历史模型不会有您定义的任何自定义方法。

这通常出现在创建模型实例的数据迁移的上下文中。如果您在迁移期间没有创建任何实例，那么get_default_expiration_date() 将永远不会被执行，并且您不会有任何问题。 （我不确定您的情况是什么，可能与 django-solo 或您正在使用的其他软件包有关？）

我认为在这种情况下最简单的解决方案是确保您的默认函数仅引用它们需要的字段，这样它们就不会生成对可能尚不存在的 SQL 列的引用。当您像现在这样获取整个对象时，它会尝试获取所有字段，并查看当前（而非历史）模型定义以查看这些字段是什么。

如果您使用values()（或only()），生成的 SQL 将仅引用您指定的字段：

def get_default_expiration_date():
    expiration_days = Preferences.objects.values("expiration_days")[0]["expiration_days"]