Skip to content

Commit b189a0e

Browse files
Fix imports
1 parent 4d98a06 commit b189a0e

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

archive_query_log/cli/captures.py

+3-5
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
from archive_query_log.cli.util import pass_config
1919
from archive_query_log.config import Config
2020
from archive_query_log.legacy.model import ArchivedUrl
21+
from archive_query_log.legacy.urls.iterable import ArchivedUrls
2122
from archive_query_log.namespaces import NAMESPACE_CAPTURE
2223
from archive_query_log.orm import Source, Capture, Archive, Provider, \
2324
InnerProvider, InnerArchive
@@ -240,9 +241,6 @@ def _import_aql22_path(
240241
importable_path: _Aql22ImportablePath,
241242
check_memento: bool = True,
242243
) -> None:
243-
from archive_query_log.legacy.model import ArchivedUrl
244-
from archive_query_log.legacy.urls.iterable import ArchivedUrls
245-
246244
echo(f"Importing captures from {importable_path.path} to "
247245
f"archive {importable_path.archive.id} and "
248246
f"provider {importable_path.provider.id}.")
@@ -276,7 +274,7 @@ def _import_aql22_path(
276274
desc="Importing captures",
277275
unit="capture",
278276
)
279-
captures = _iter_aql22_captures(
277+
captures_iter = _iter_aql22_captures(
280278
config=config,
281279
importable_path=importable_path,
282280
last_modified=oldest_modification_time,
@@ -288,7 +286,7 @@ def _import_aql22_path(
288286
**capture.to_dict(include_meta=True),
289287
"_op_type": "create",
290288
}
291-
for capture in captures
289+
for capture in captures_iter
292290
)
293291
try:
294292
responses: Iterable[tuple[bool, Any]] = config.es.streaming_bulk(

0 commit comments

Comments
 (0)