Organize and remove unused imports

Florents-Tselai · Oct 20, 2023 · 89d8a75 · 89d8a75
1 parent ab3af9a
commit 89d8a75
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/tests/test_warcdb.py b/tests/test_warcdb.py
@@ -1,11 +1,11 @@
-from click.testing import CliRunner
-from warcdb import warcdb_cli
 import os
-import re
 import pathlib
+import re
+
 import pytest
 import sqlite_utils
-from unittest import TestCase
+from click.testing import CliRunner
+from warcdb import warcdb_cli
 
 db_file = "test_warc.db"
 tests_dir = pathlib.Path(__file__).parent

diff --git a/warcdb/__init__.py b/warcdb/__init__.py
@@ -264,6 +264,7 @@ def to_import():
             if f.startswith('http'):
                 yield from tqdm(ArchiveIterator(req.get(f, stream=True).raw, arc2warc=True), desc=f)
             elif f.endswith('.wacz'):
+                # TODO: can we support loading WACZ files by URL?
                 wacz = zipfile.ZipFile(f)
                 warcs = filter(lambda f: f.filename.endswith('warc.gz'), wacz.infolist())
                 for warc in warcs: