27 lines
1 KiB
Python
27 lines
1 KiB
Python
|
from data.database import ImageStorage, ManifestBlob
|
||
|
|
from data.model import BlobDoesNotExist
|
||
|
|
from data.model.storage import get_storage_by_uuid, InvalidImageException
|
||
|
|
from data.model.blob import get_repository_blob_by_digest as legacy_get
|
||
|
|
|
||
|
|
def get_repository_blob_by_digest(repository, blob_digest):
|
||
|
|
""" Find the content-addressable blob linked to the specified repository and
|
||
|
|
returns it or None if none.
|
||
|
|
"""
|
||
|
|
try:
|
||
|
|
storage = (ImageStorage
|
||
|
|
.select(ImageStorage.uuid)
|
||
|
|
.join(ManifestBlob)
|
||
|
|
.where(ManifestBlob.repository == repository,
|
||
|
|
ImageStorage.content_checksum == blob_digest,
|
||
|
|
ImageStorage.uploading == False)
|
||
|
|
.get())
|
||
|
|
|
||
|
|
return get_storage_by_uuid(storage.uuid)
|
||
|
|
except (ImageStorage.DoesNotExist, InvalidImageException):
|
||
|
|
# TODO: Remove once we are no longer using the legacy tables.
|
||
|
|
# Try the legacy call.
|
||
|
|
try:
|
||
|
|
return legacy_get(repository, blob_digest)
|
||
|
|
except BlobDoesNotExist:
|
||
|
|
return None
|