diff --git a/swh/loader/package/loader.py b/swh/loader/package/loader.py --- a/swh/loader/package/loader.py +++ b/swh/loader/package/loader.py @@ -46,6 +46,7 @@ ) from swh.model.identifiers import SWHID from swh.storage import get_storage +from swh.storage.interface import StorageInterface from swh.storage.utils import now from swh.storage.algos.snapshot import snapshot_get_latest @@ -126,7 +127,7 @@ # This expects to use the environment variable SWH_CONFIG_FILENAME self.config = SWHConfig.parse_config_file() self._check_configuration() - self.storage = get_storage(**self.config["storage"]) + self.storage: StorageInterface = get_storage(**self.config["storage"]) self.url = url self.visit_date = datetime.datetime.now(tz=datetime.timezone.utc) self.max_content_size = self.config["max_content_size"] @@ -686,11 +687,11 @@ (authority.type, authority.url): authority for authority in authorities } if authorities: - self.storage.metadata_authority_add(deduplicated_authorities.values()) + self.storage.metadata_authority_add(list(deduplicated_authorities.values())) def _create_fetchers(self, fetchers: Iterable[MetadataFetcher]) -> None: deduplicated_fetchers = { (fetcher.name, fetcher.version): fetcher for fetcher in fetchers } if fetchers: - self.storage.metadata_fetcher_add(deduplicated_fetchers.values()) + self.storage.metadata_fetcher_add(list(deduplicated_fetchers.values()))