class ImageStoreImageView(APIView):
"""
List images or import image from remote disk image repositories (a.k.a. imagestores).
"""
dc_bound = False
def __init__(self, request, name, uuid, data):
super(ImageStoreImageView, self).__init__(request)
self.data = data
self.name = name
self.uuid = uuid
repositories = ImageStore.get_repositories()
try:
self.repo = ImageStore(name, url=repositories[name])
except KeyError: # The name is not in VMS_IMAGE_REPOSITORIES
raise ObjectNotFound(model=ImageStore)
def get_image(self):
uuid = self.uuid
for img in self.repo.images:
if img['uuid'] == uuid:
return img
else:
raise ObjectNotFound(model=Image)
def get(self, many=False):
if many:
assert not self.uuid
if self.full or self.extended:
res = self.repo.images
else:
res = [img['uuid'] for img in self.repo.images]
else:
assert self.uuid
res = self.get_image()
return SuccessTaskResponse(self.request, res, dc_bound=False)
def post(self):
img = self.get_image()
data = self.data
data['manifest_url'] = self.repo.get_image_manifest_url(img['uuid'])
data.pop('file_url', None)
name = data.get('name', None)
if not name:
name = data['name'] = img['name']
return call_api_view(self.request,
'POST',
image_manage,
name,
data=data)
class ImageStoreImageView(APIView):
"""
List images or import image from remote disk image repositories (a.k.a. imagestores).
"""
dc_bound = False
def __init__(self, request, name, uuid, data):
super(ImageStoreImageView, self).__init__(request)
self.data = data
self.name = name
self.uuid = uuid
repositories = ImageStore.get_repositories(
include_image_vm=request.user.is_staff)
try:
self.repo = ImageStore(name, url=repositories[name])
except KeyError: # The name is not in VMS_IMAGE_REPOSITORIES
raise ObjectNotFound(model=ImageStore)
def get_image(self):
uuid = self.uuid
for img in self.repo.images:
if img['uuid'] == uuid:
return img
else:
raise ObjectNotFound(model=Image)
def get(self, many=False):
if many:
assert not self.uuid
if self.full or self.extended:
res = self.repo.images
else:
res = [img['uuid'] for img in self.repo.images]
else:
assert self.uuid
res = self.get_image()
return SuccessTaskResponse(self.request, res, dc_bound=False)
def post(self):
img = self.get_image()
uuid = img['uuid']
data = self.data
data['manifest_url'] = self.repo.get_image_manifest_url(uuid)
data.pop('file_url', None)
name = data.get('name', None)
if not name:
name = data['name'] = img['name']
# Although this is also checked inside the image_manage, doing it here is better because:
# - checking the uniqueness of the UUID is done differently in image_manage and the result is not a 406 error
# - it is faster - in case the name/uuid is not unique we don't have to call another view
if Image.objects.filter(Q(uuid=uuid) | Q(name=name)).exists():
raise ObjectAlreadyExists(model=Image)
return call_api_view(self.request,
'POST',
image_manage,
name,
data=data)