python - 在 Django 中一次更新所有模型

Question

我正在尝试在 Django（python）中以特定顺序一次更新所有对象的位置字段。

这就是我现在所做的，但问题是它会产生大量查询。

    servers = frontend_models.Server.objects.all().order_by('-vote_count')

    i = 1
    for server in servers:
        server.last_rank = i
        server.save()
        i += 1

有没有办法更新

Model.objects.all().order_by('some_field').update(position=some_number_that_changes_for each_object)

谢谢！

Answer 1

你可以使用F()表达式 fromdjango.db.models来做同样的事情：

Model.objects.all().order_by('some_field').update(position=F(some_field)+1)

这将生成单个 SQL 查询来更新所有列值，因此对数据库也很有效。

Answer 2

据我所知，Django 的对象关系映射系统并没有提供表达这种更新操作的方法。但是，如果您知道如何用 SQL 来表达它，那么您可以通过自定义 SQL 查询来运行它：

from django.db import connection
cursor = connection.cursor()
cursor.execute('''UPDATE myapp_server ...''')

不同的数据库引擎以不同的方式表达此操作。在 MySQL 中，您将运行以下查询：

SET @rownum=0;
UPDATE myapp_server A,
       (SELECT id, @rownum := @rownum + 1 AS rank
        FROM myapp_server
        ORDER BY vote_count DESCENDING) B
SET A.rank = B.rank
WHERE A.id = B.id

在 PostgreSQL 中，我认为你会使用

UPDATE myapp_server A,
       (SELECT id, rownumber() AS rank
        OVER (ORDER BY vote_count DESCENDING)
        FROM myapp_server) B
SET A.rank = B.rank
WHERE A.id = B.id

（但这是未经测试的，所以要小心！）。