通过 shell 在 Django 中插入大量数据

Question

我目前有一个使用 postgres 设置的 Django 环境。我试图在 shell 中运行一个简单的脚本，该脚本需要很长时间才能完成。

有没有一种有效的方法来更新 Django 中的 postgres 表中的所有记录？

我的表名为 City，它包含大约 200,000 个城市。

这是我在 python manage.py shell 中运行的脚本：

from locations.models import City
from django.template.defaultfilters import slugify
counter = 0
for obj in City.objects.all():
    counter = counter + 1
    if counter % 1000 == 0:
        print counter
    obj.city_name_slug = slugify(obj.city_name) + "-" + slugify(obj.region)
    obj.save()

Answer 1

使用事务，它将加快您的查询速度。还将obj.save() 调用替换为filter().update() 组合，这是更改数据库中单个字段的最快方法：

from locations.models import City
from django.db import transaction
from django.template.defaultfilters import slugify
counter = 0
with transaction.atomic():
    for obj in City.objects.all():
        counter = counter + 1
        if counter % 1000 == 0:
            print counter
        slug = slugify(obj.city_name) + "-" + slugify(obj.region)
        City.objects.filter(pk=obj.pk).update(city_name_slug=slug)