Issue
I'm trying to create a custom Scrapy Item Exporter based off JsonLinesItemExporter so I can slightly alter the structure it produces.
I have read the documentation here http://doc.scrapy.org/en/latest/topics/exporters.html but it doesn't state how to create a custom exporter, where to store it or how to link it to your Pipeline.
I have identified how to go custom with the Feed Exporters but this is not going to suit my requirements, as I want to call this exporter from my Pipeline.
Here is the code I've come up with which has been stored in a file in the root of the project called exporters.py
from scrapy.contrib.exporter import JsonLinesItemExporter
class FanItemExporter(JsonLinesItemExporter):
def __init__(self, file, **kwargs):
self._configure(kwargs, dont_fail=True)
self.file = file
self.encoder = ScrapyJSONEncoder(**kwargs)
self.first_item = True
def start_exporting(self):
self.file.write("""{
'product': [""")
def finish_exporting(self):
self.file.write("]}")
def export_item(self, item):
if self.first_item:
self.first_item = False
else:
self.file.write(',\n')
itemdict = dict(self._get_serialized_fields(item))
self.file.write(self.encoder.encode(itemdict))
I have simply tried calling this from my pipeline by using FanItemExporter and trying variations of the import but it's not resulting in anything.
Solution
This answer is now outdated See Max's answer for an easier approach
It is true that the Scrapy documentation does not clearly state where to place an Item Exporter. To use an Item Exporter, these are the steps to follow.
- Choose an Item Exporter class and import it to
pipeline.py
in the project directory. It could be a pre-defined Item Exporter (ex.XmlItemExporter
) or user-defined (likeFanItemExporter
defined in the question) - Create an Item Pipeline class in
pipeline.py
. Instantiate the imported Item Exporter in this class. Details will be explained in the later part of the answer. - Now, register this pipeline class in
settings.py
file.
Following is a detailed explanation of each step. Solution to the question is included in each step.
Step 1
If using a pre-defined Item Exporter class, import it from
scrapy.exporters
module.
Ex:from scrapy.exporters import XmlItemExporter
If you need a custom exporter, define a custom class in a file. I suggest placing the class in
exporters.py
file. Place this file in the project folder (wheresettings.py
,items.py
reside).While creating a new sub-class, it is always a good idea to import
BaseItemExporter
. It would be apt if we intend to change the functionality entirely. However, in this question, most of the functionality is close toJsonLinesItemExporter
.
Hence, I am attaching two versions of the same ItemExporter. One version extends BaseItemExporter
class and the other extends JsonLinesItemExporter
class
Version 1: Extending BaseItemExporter
Since BaseItemExporter
is the parent class, start_exporting()
, finish_exporting()
, export_item()
must be overrided to suit our needs.
from scrapy.exporters import BaseItemExporter
from scrapy.utils.serialize import ScrapyJSONEncoder
from scrapy.utils.python import to_bytes
class FanItemExporter(BaseItemExporter):
def __init__(self, file, **kwargs):
self._configure(kwargs, dont_fail=True)
self.file = file
self.encoder = ScrapyJSONEncoder(**kwargs)
self.first_item = True
def start_exporting(self):
self.file.write(b'{\'product\': [')
def finish_exporting(self):
self.file.write(b'\n]}')
def export_item(self, item):
if self.first_item:
self.first_item = False
else:
self.file.write(b',\n')
itemdict = dict(self._get_serialized_fields(item))
self.file.write(to_bytes(self.encoder.encode(itemdict)))
Version 2: Extending JsonLinesItemExporter
JsonLinesItemExporter
provides the exact same implementation of export_item()
method. Therefore only start_exporting()
and finish_exporting()
methods are overrided.
Implementation of JsonLinesItemExporter
can be seen in the folder python_dir\pkgs\scrapy-1.1.0-py35_0\Lib\site-packages\scrapy\exporters.py
from scrapy.exporters import JsonItemExporter
class FanItemExporter(JsonItemExporter):
def __init__(self, file, **kwargs):
# To initialize the object using JsonItemExporter's constructor
super().__init__(file)
def start_exporting(self):
self.file.write(b'{\'product\': [')
def finish_exporting(self):
self.file.write(b'\n]}')
Note: When writing data to file, it is important to note that the standard Item Exporter classes expect binary files. Hence, the file must be opened in binary mode (b
). For the same reason, write()
method in both the version write bytes
to file.
Step 2
Creating an Item Pipeline class.
from project_name.exporters import FanItemExporter
class FanExportPipeline(object):
def __init__(self, file_name):
# Storing output filename
self.file_name = file_name
# Creating a file handle and setting it to None
self.file_handle = None
@classmethod
def from_crawler(cls, crawler):
# getting the value of FILE_NAME field from settings.py
output_file_name = crawler.settings.get('FILE_NAME')
# cls() calls FanExportPipeline's constructor
# Returning a FanExportPipeline object
return cls(output_file_name)
def open_spider(self, spider):
print('Custom export opened')
# Opening file in binary-write mode
file = open(self.file_name, 'wb')
self.file_handle = file
# Creating a FanItemExporter object and initiating export
self.exporter = FanItemExporter(file)
self.exporter.start_exporting()
def close_spider(self, spider):
print('Custom Exporter closed')
# Ending the export to file from FanItemExport object
self.exporter.finish_exporting()
# Closing the opened output file
self.file_handle.close()
def process_item(self, item, spider):
# passing the item to FanItemExporter object for expoting to file
self.exporter.export_item(item)
return item
Step 3
Since the Item Export Pipeline is defined, register this pipeline in settings.py
file. Also add the field FILE_NAME
to settings.py
file. This field contains the filename of the output file.
Add the following lines to settings.py
file.
FILE_NAME = 'path/outputfile.ext'
ITEM_PIPELINES = {
'project_name.pipelines.FanExportPipeline' : 600,
}
If ITEM_PIPELINES
is already uncommented, then add the following line to the ITEM_PIPELINES
dictionary.
'project_name.pipelines.FanExportPipeline' : 600,
This is one way to create a custom Item Export pipeline.
Answered By - pbskumar
0 comments:
Post a Comment
Note: Only a member of this blog may post a comment.