74 lines
3.2 KiB
Python
74 lines
3.2 KiB
Python
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, storage_metadata):
|
|
repo = _basequery.get_existing_repository(namespace, repo_name)
|
|
location = ImageStorageLocation.get(name=location_name)
|
|
return BlobUpload.create(repository=repo, location=location, uuid=uuid,
|
|
storage_metadata=storage_metadata)
|