from uuid import uuid4 from data.model import tag, _basequery, BlobDoesNotExist, InvalidBlobUpload, db_transaction from data.database import (Repository, Namespace, ImageStorage, Image, ImageStorageLocation, ImageStoragePlacement, BlobUpload) def get_repo_blob_by_digest(namespace, repo_name, blob_digest): """ Find the content-addressable blob linked to the specified repository. """ placements = list(ImageStoragePlacement .select(ImageStoragePlacement, ImageStorage, ImageStorageLocation) .join(ImageStorageLocation) .switch(ImageStoragePlacement) .join(ImageStorage) .join(Image) .join(Repository) .join(Namespace, on=(Namespace.id == Repository.namespace_user)) .where(Repository.name == repo_name, Namespace.username == namespace, ImageStorage.checksum == blob_digest, ImageStorage.uploading == False)) if not placements: raise BlobDoesNotExist('Blob does not exist with digest: {0}'.format(blob_digest)) found = placements[0].storage found.locations = {placement.location.name for placement in placements} return found def store_blob_record_and_temp_link(namespace, repo_name, blob_digest, location_obj, byte_count, link_expiration_s): """ Store a record of the blob and temporarily link it to the specified repository. """ random_image_name = str(uuid4()) with db_transaction(): repo = _basequery.get_existing_repository(namespace, repo_name) try: storage = ImageStorage.get(checksum=blob_digest) storage.image_size = byte_count storage.save() ImageStoragePlacement.get(storage=storage, location=location_obj) except ImageStorage.DoesNotExist: storage = ImageStorage.create(checksum=blob_digest, uploading=False, image_size=byte_count) ImageStoragePlacement.create(storage=storage, location=location_obj) except ImageStoragePlacement.DoesNotExist: ImageStoragePlacement.create(storage=storage, location=location_obj) # Create a temporary link into the repository, to be replaced by the v1 metadata later # and create a temporary tag to reference it image = Image.create(storage=storage, docker_image_id=random_image_name, repository=repo) tag.create_temporary_hidden_tag(repo, image, link_expiration_s) def get_blob_upload(namespace, repo_name, upload_uuid): """ Load the upload which is already in progress. """ try: return (BlobUpload .select() .join(Repository) .join(Namespace, on=(Namespace.id == Repository.namespace_user)) .where(Repository.name == repo_name, Namespace.username == namespace, BlobUpload.uuid == upload_uuid) .get()) except BlobUpload.DoesNotExist: raise InvalidBlobUpload() def initiate_upload(namespace, repo_name, uuid, location_name): repo = _basequery.get_existing_repository(namespace, repo_name) location = ImageStorageLocation.get(name=location_name) return BlobUpload.create(repository=repo, location=location, uuid=uuid)