mirror of https://github.com/scrapy/scrapy.git
Call `finish_exporting` even when itemcount == 0
This commit is contained in:
parent
fe60c1224e
commit
087334009c
|
|
@ -350,11 +350,11 @@ class FeedExporter:
|
|||
return defer.DeferredList(deferred_list) if deferred_list else None
|
||||
|
||||
def _close_slot(self, slot, spider):
|
||||
slot.finish_exporting()
|
||||
if not slot.itemcount and not slot.store_empty:
|
||||
# We need to call slot.storage.store nonetheless to get the file
|
||||
# properly closed.
|
||||
return defer.maybeDeferred(slot.storage.store, slot.file)
|
||||
slot.finish_exporting()
|
||||
logmsg = f"{slot.format} feed ({slot.itemcount} items) in: {slot.uri}"
|
||||
d = defer.maybeDeferred(slot.storage.store, slot.file)
|
||||
|
||||
|
|
|
|||
|
|
@ -33,8 +33,9 @@ from zope.interface.verify import verifyObject
|
|||
|
||||
import scrapy
|
||||
from scrapy.exceptions import NotConfigured, ScrapyDeprecationWarning
|
||||
from scrapy.exporters import CsvItemExporter
|
||||
from scrapy.exporters import CsvItemExporter, JsonItemExporter
|
||||
from scrapy.extensions.feedexport import (
|
||||
_FeedSlot,
|
||||
BlockingFeedStorage,
|
||||
FeedExporter,
|
||||
FileFeedStorage,
|
||||
|
|
@ -890,6 +891,60 @@ class FeedExportTest(FeedExportTestBase):
|
|||
data = yield self.exported_no_data(settings)
|
||||
self.assertEqual(b'', data[fmt])
|
||||
|
||||
@defer.inlineCallbacks
|
||||
def test_finish_exporting_is_called(self):
|
||||
# for each format, keep track of when start_exporting
|
||||
# has been called but finish_exporting hasn't been called
|
||||
startRecordingTracker = {}
|
||||
# we expect finish_recording to be called, setting this to false
|
||||
expected = {'json': False}
|
||||
|
||||
items = [
|
||||
self.MyItem({'foo': 'bar1', 'egg': 'spam1'}),
|
||||
]
|
||||
settings = {
|
||||
'FEEDS': {
|
||||
self._random_temp_filename(): {'format': 'json'},
|
||||
},
|
||||
'FEED_EXPORT_INDENT': None,
|
||||
}
|
||||
|
||||
# override export_item to raise exception
|
||||
class FakeJsonItemExporter(JsonItemExporter):
|
||||
def export_item(self, item):
|
||||
raise Exception('foo')
|
||||
|
||||
# override start/stop_exporting to modify startRecordingTracker
|
||||
class FakeFeedSlot(_FeedSlot):
|
||||
def start_exporting(self):
|
||||
startRecordingTracker[self.format] = True
|
||||
if not self._exporting:
|
||||
self.exporter.start_exporting()
|
||||
self._exporting = True
|
||||
|
||||
def finish_exporting(self):
|
||||
print('finish export called')
|
||||
startRecordingTracker[self.format] = False
|
||||
if self._exporting:
|
||||
self.exporter.finish_exporting()
|
||||
self._exporting = False
|
||||
|
||||
|
||||
with ExitStack() as stack:
|
||||
stack.enter_context(
|
||||
mock.patch(
|
||||
'scrapy.exporters.JsonItemExporter', FakeJsonItemExporter
|
||||
)
|
||||
)
|
||||
stack.enter_context(
|
||||
mock.patch(
|
||||
'scrapy.extensions.feedexport._FeedSlot', FakeFeedSlot
|
||||
)
|
||||
)
|
||||
_ = yield self.exported_data(items, settings)
|
||||
self.assertDictEqual(startRecordingTracker, expected)
|
||||
|
||||
|
||||
@defer.inlineCallbacks
|
||||
def test_export_no_items_store_empty(self):
|
||||
formats = (
|
||||
|
|
|
|||
Loading…
Reference in New Issue