Changeset View

Standalone View

				# Copyright (C) 2022 The Software Heritage developers
				# See the AUTHORS file at the top-level directory of this distribution
				# License: GNU General Public License version 3, or any later version
				# See top-level LICENSE file for more information

				from typing import Iterable, List

				from swh.model.model import ObjectType, Sha1Git
				from swh.storage.interface import StorageInterface


				def identifiers_missing(
				storage: StorageInterface, obj_type: ObjectType, obj_ids: List[Sha1Git]
				) -> Iterable[Sha1Git]:
				"""Lookup missing Software Heritage hashes, choosing the correct
				underlying method based on the type.

				The idea is that conceptually we are looking up a list of Core
				SWHIDs with the same objec type part, and to enforce that we pash
				the type and the hashes separately.

				Args:
				storage (swh.storage.interface.StorageInterface): the storage
				instance

				obj_type: What type of object are the hashes supposed to point
				to.

				obj_ids: list of hashes to look up

				Yields:
				missing hashes for the given object type
				"""

				if obj_type == ObjectType.CONTENT:
				return storage.content_missing_per_sha1_git(obj_ids)
				elif obj_type == ObjectType.DIRECTORY:
				return storage.directory_missing(obj_ids)
				elif obj_type == ObjectType.REVISION:
				return storage.revision_missing(obj_ids)
				elif obj_type == ObjectType.RELEASE:
				return storage.release_missing(obj_ids)
				elif obj_type == ObjectType.SNAPSHOT:
				return storage.snapshot_missing(obj_ids)