From 7008f9b735c58d5189ea77f611b5d9c39950b109 Mon Sep 17 00:00:00 2001
From: Cristian <cristianvargasvalencia@gmail.com>
Date: Sat, 28 Nov 2020 13:11:15 -0500
Subject: [PATCH] feat: move import

---
 archivebox/index/html.py | 3 ++-
 archivebox/index/json.py | 2 +-
 archivebox/main.py       | 2 +-
 3 files changed, 4 insertions(+), 3 deletions(-)

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)