Scrapy文件下载如何使用自定义文件名

时间:2017-10-31 08:58:20

标签: python scrapy scrapy-spider scrapy-pipeline

对于我scrapy项目,我目前正在使用FilesPipeline。下载的文件以其URL的SHA1哈希值存储为文件名。

[(True,
  {'checksum': '2b00042f7481c7b056c4b410d28f33cf',
   'path': 'full/0a79c461a4062ac383dc4fade7bc09f1384a3910.jpg',
   'url': 'http://www.example.com/files/product1.pdf'}),
 (False,
  Failure(...))]

如何使用自定义文件名存储文件?

在上面的示例中,我希望文件名为“product1_0a79c461a4062ac383dc4fade7bc09f1384a3910.pdf”,因此我保持唯一性,但使文件名可见。

作为一个起点,我探索了我项目的pipelines.py但没有取得多大成功。

import scrapy
from scrapy.pipelines.images import FilesPipeline
from scrapy.exceptions import DropItem

class MyFilesPipeline(FilesPipeline):

    def file_path(self, request, response=None, info=None):
        return request.meta.get('filename','')

    def get_media_requests(self, item, info):
        file_url = item['file_url']
        meta = {'filename': item['name']}
        yield Request(url=file_url, meta=meta)

在我的settings.py

中包含此参数
ITEM_PIPELINES = {
    #'scrapy.pipelines.files.FilesPipeline': 300
    'io_spider.pipelines.MyFilesPipeline': 200
}

已经询问了similar question,但它确实定位了图片,而不是文件。

任何帮助将不胜感激。

2 个答案:

答案 0 :(得分:4)

file_path应返回文件的路径。在您的代码中,file_path会返回item['name'],这将是您文件的路径。请注意,默认情况下file_path calculates SHA1 hashes。所以你的方法应该是这样的:

def file_path(self, request, response=None, info=None):
    original_path = super(MyFilesPipeline, self).file_path(request, response=None, info=None)
    sha1_and_extension = original_path.split('/')[1] # delete 'full/' from the path
    return request.meta.get('filename','') + "_" + sha1_and_extension

答案 1 :(得分:1)

尝试使用此file_path方法:

def file_path(self, request, response=None, info=None):
    url = request if not isinstance(request, Request) else request.url
    media_guid = hashlib.sha1(to_bytes(url)).hexdigest()
    path, media_ext = os.path.splitext(url)
    media_name = os.path.split(path)[1]
    return '%s_%s%s' % (media_name, media_guid, media_ext)

(注意:这是未经测试的代码)

相关问题