diff --git a/archivebox/index/html.py b/archivebox/index/html.py index d44137e3..41fbe35e 100644 --- a/archivebox/index/html.py +++ b/archivebox/index/html.py @@ -49,12 +49,13 @@ def parse_html_main_index(out_dir: Path=OUTPUT_DIR) -> Iterator[str]: yield line.split('"')[1] return () - +@enforce_types def generate_index_from_links(links: List[Link], with_headers: bool): if with_headers: output = main_index_template(links) else: output = main_index_template(links, template=MINIMAL_INDEX_TEMPLATE) + return output @enforce_types def main_index_template(links: List[Link], template: str=MAIN_INDEX_TEMPLATE) -> str: diff --git a/archivebox/index/json.py b/archivebox/index/json.py index 05521e9d..f24b969f 100644 --- a/archivebox/index/json.py +++ b/archivebox/index/json.py @@ -8,7 +8,7 @@ from pathlib import Path from datetime import datetime from typing import List, Optional, Iterator, Any, Union -from .schema import Link, ArchiveResult +from .schema import Link from ..system import atomic_write from ..util import enforce_types from ..config import ( diff --git a/archivebox/main.py b/archivebox/main.py index db589841..2d36e1f2 100644 --- a/archivebox/main.py +++ b/archivebox/main.py @@ -59,6 +59,7 @@ from .index.sql import ( from .index.html import ( generate_index_from_links, ) +from .index.csv import links_to_csv from .extractors import archive_links, archive_link, ignore_methods from .config import ( stderr, @@ -770,7 +771,6 @@ def list_all(filter_patterns_str: Optional[str]=None, elif html: output = generate_index_from_links(folders.values(), with_headers) elif csv: - from .index.csv import links_to_csv output = links_to_csv(folders.values(), cols=csv.split(','), header=with_headers) else: output = printable_folders(folders, with_headers=with_headers)