From f292cface27e6de0a552d2fc1e78fd99f6aa9219 Mon Sep 17 00:00:00 2001 From: Cristian Date: Wed, 4 Nov 2020 14:40:44 -0500 Subject: [PATCH] fix: Add condition for oneshot when archiving links --- archivebox/extractors/__init__.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/archivebox/extractors/__init__.py b/archivebox/extractors/__init__.py index d5d8832f..23a4f5ef 100644 --- a/archivebox/extractors/__init__.py +++ b/archivebox/extractors/__init__.py @@ -66,8 +66,9 @@ def archive_link(link: Link, overwrite: bool=False, methods: Optional[Iterable[s """download the DOM, PDF, and a screenshot into a folder named after the link's timestamp""" # TODO: Remove when the input is changed to be a snapshot. Suboptimal approach. - from core.models import Snapshot, ArchiveResult - snapshot = Snapshot.objects.get(url=link.url) + if not skip_index: + from core.models import Snapshot, ArchiveResult + snapshot = Snapshot.objects.get(url=link.url) ARCHIVE_METHODS = get_default_archive_methods() @@ -103,7 +104,8 @@ def archive_link(link: Link, overwrite: bool=False, methods: Optional[Iterable[s stats[result.status] += 1 log_archive_method_finished(result) - ArchiveResult.objects.create(snapshot=snapshot, extractor=method_name, cmd=result.cmd, cmd_version=result.cmd_version, + if not skip_index: + ArchiveResult.objects.create(snapshot=snapshot, extractor=method_name, cmd=result.cmd, cmd_version=result.cmd_version, output=result.output, pwd=result.pwd, start_ts=result.start_ts, end_ts=result.end_ts, status=result.status) else: