python - 如何在Scrapy中更新DjangoItem

Question

我一直在使用 Scrapy，但遇到了一些问题。

DjangoItem 有一个 save 方法来使用 Django ORM 保存项目。这很棒，但如果我多次运行抓取工具，即使我可能只想更新以前的值，也会在数据库中创建新项目。

查看文档和源代码后，我没有看到任何更新现有项目的方法。

我知道我可以调用 ORM 来查看某个项目是否存在并更新它，但这意味着为每个对象调用数据库，然后再次保存该项目。

如果项目已经存在，我如何更新它们？

Answer 1

不幸的是，我发现实现此目的的最佳方法是严格执行所述操作:使用 django_model.objects.get 检查数据库中是否存在该项目，如果存在则更新它.

在我的设置文件中，我添加了新管道:

ITEM_PIPELINES = {
    # ...
    # Last pipeline, because further changes won't be saved.
    'apps.scrapy.pipelines.ItemPersistencePipeline': 999
}

我创建了一些辅助方法来处理创建项目模型的工作，并在必要时创建一个新模型:

def item_to_model(item):
    model_class = getattr(item, 'django_model')
    if not model_class:
        raise TypeError("Item is not a `DjangoItem` or is misconfigured")

    return item.instance


def get_or_create(model):
    model_class = type(model)
    created = False

    # Normally, we would use `get_or_create`. However, `get_or_create` would
    # match all properties of an object (i.e. create a new object
    # anytime it changed) rather than update an existing object.
    #
    # Instead, we do the two steps separately
    try:
        # We have no unique identifier at the moment; use the name for now.
        obj = model_class.objects.get(name=model.name)
    except model_class.DoesNotExist:
        created = True
        obj = model  # DjangoItem created a model for us.

    return (obj, created)


def update_model(destination, source, commit=True):
    pk = destination.pk

    source_dict = model_to_dict(source)
    for (key, value) in source_dict.items():
        setattr(destination, key, value)

    setattr(destination, 'pk', pk)

    if commit:
        destination.save()

    return destination

然后，最终的管道相当简单:

class ItemPersistencePipeline(object):
    def process_item(self, item, spider):
        try:
             item_model = item_to_model(item)
        except TypeError:
            return item

        model, created = get_or_create(item_model)

        update_model(model, item_model)

        return item