我有一个与某些 Resource 对象有 M2M 关系的 Story 模型。一些 Resource 对象缺少名称,因此我想将 Story 的标题复制到分配的 Resource 对象。
这是我的代码:
from collector import models
from django.core.paginator import Paginator
paginator = Paginator(models.Story.objects.all(), 1000)
def fix_issues():
for page in range(1, paginator.num_pages + 1):
for story in paginator.page(page).object_list:
name_story = story.title
for r in story.resources.select_subclasses():
if r.name != name_story:
r.name = name_story
r.save()
if len(r.name) == 0:
print("Something went wrong: " + name_story)
print("done processing page %s out of %s" % (page, paginator.num_pages))
fix_issues()
我需要使用分页器,因为我正在处理一百万个对象。奇怪的是,在调用 fix_issues() 之后,我的大约一半没有名称的资源现在有了正确的名称,而另一半仍然没有名称。我可以一次又一次地调用 fix_issues() 并且每次有更多对象收到一个名称。这对我来说似乎很奇怪,为什么第一次不更新对象而只更新第二次?
附加信息:
- 永远不会打印“出现问题:”消息。
- 我正在使用 django-model-utils 中的 select_subclasses 来迭代所有资源(任何类型)。
- story.title永远不会是空的。
- 当我运行这些命令时,不会打印任何错误消息。
- 我没有重写 Resource 模型的 save 方法(只有 Story 模型的 save 方法)。
- 我尝试使用@transaction.atomic,但结果是一样的。
我的模型:
class Resource(models.Model):
name = models.CharField(max_length=200)
# Important for retrieving the correct subtype.
objects = InheritanceManager()
def __str__(self):
return str(self.name)
class CustomResource(Resource):
homepage = models.CharField(max_length=3000, default="", blank=True, null=True)
class Story(models.Model):
url = models.URLField(max_length=3000)
resources = models.ManyToManyField(Resource)
popularity = models.FloatField()
def _update_popularity(self):
self.popularity = 3
def save(self, *args, **kwargs):
super(Story, self).save(*args, **kwargs)
self._update_popularity()
super(Story, self).save(*args, **kwargs)
select_subclasses 的 文档:http: //django-model-utils.readthedocs.io/en/latest/managers.html#inheritancemanager
进一步调查: 我认为也许 select_subclasses 没有返回所有对象。现在每个故事都只有一个资源。因此很容易检查 select_subclasses 是否总是返回一项。这是我使用的功能:
def find_issues():
for page in range(1, paginator.num_pages + 1):
for story in paginator.page(page).object_list:
assert(len(story.resources.select_subclasses()) == 1)
print("done processing page %s out of %s" % (page, paginator.num_pages))
但同样,这执行没有任何问题。所以我不认为 select_subclasses 是罪魁祸首。我还检查了 paginator.num_pages 是否正确。如果我除以 1000(每页项目),我得到的正是我数据库中的故事数量。