From 018bd917453389aff2b551103ff40e0569c67580 Mon Sep 17 00:00:00 2001 From: Cristian Date: Tue, 8 Sep 2020 10:39:01 -0500 Subject: [PATCH] refactor: Remove get_iter lambda from archive_links --- archivebox/extractors/__init__.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/archivebox/extractors/__init__.py b/archivebox/extractors/__init__.py index 7820c4a3..cc310570 100644 --- a/archivebox/extractors/__init__.py +++ b/archivebox/extractors/__init__.py @@ -135,11 +135,10 @@ def archive_links(all_links: any, overwrite: bool=False, methods: Optional[Itera if type(all_links) is QuerySet: num_links: int = all_links.count() get_link = lambda x: x.as_link() - get_iter = lambda x: x.iterator() + all_links = all_links.iterator() else: num_links: int = len(all_links) get_link = lambda x: x - get_iter = lambda x: x if num_links == 0: return [] @@ -147,7 +146,7 @@ def archive_links(all_links: any, overwrite: bool=False, methods: Optional[Itera log_archiving_started(num_links) idx: int = 0 try: - for link in get_iter(all_links): + for link in all_links: idx += 1 to_archive = get_link(link) archive_link(to_archive, overwrite=overwrite, methods=methods, out_dir=link.link_dir)