¡@
OpenStack Study: image_data.py**** CubicPower OpenStack Study ****
# Copyright 2012 OpenStack Foundation. # All Rights Reserved. # # Licensed under the Apache License, Version 2.0 (the "License"); you may # not use this file except in compliance with the License. You may obtain # a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the # License for the specific language governing permissions and limitations # under the License. import webob.exc import glance.api.policy from glance.common import exception from glance.common import utils from glance.common import wsgi import glance.db import glance.gateway import glance.notifier import glance.openstack.common.log as logging import glance.store LOG = logging.getLogger(__name__)
**** CubicPower OpenStack Study **** class ImageDataController(object):
**** CubicPower OpenStack Study **** def __init__(self, db_api=None, store_api=None, policy_enforcer=None, notifier=None, gateway=None): if gateway is None: db_api = db_api or glance.db.get_api() store_api = store_api or glance.store policy = policy_enforcer or glance.api.policy.Enforcer() notifier = notifier or glance.notifier.Notifier() gateway = glance.gateway.Gateway(db_api, store_api, notifier, policy) self.gateway = gateway
**** CubicPower OpenStack Study **** def _restore(self, image_repo, image): """ Restore the image to queued status. :param image_repo: The instance of ImageRepo :param image: The image will be restored """ try: if image_repo and image: image.status = 'queued' image_repo.save(image) except Exception as e: msg = _("Unable to restore image %(image_id)s: %(e)s") % \ {'image_id': image.image_id, 'e': unicode(e)} LOG.exception(msg) @utils.mutating
**** CubicPower OpenStack Study **** def upload(self, req, image_id, data, size): image_repo = self.gateway.get_repo(req.context) try: image = image_repo.get(image_id) image.status = 'saving' try: image_repo.save(image) image.set_data(data, size) image_repo.save(image) except exception.NotFound as e: msg = (_("Image %(id)s could not be found after upload." "The image may have been deleted during the upload: " "%(error)s Cleaning up the chunks uploaded") % {'id': image_id, 'error': e}) LOG.warn(msg) # NOTE(sridevi): Cleaning up the uploaded chunks. try: image.delete() except exception.NotFound: # NOTE(sridevi): Ignore this exception pass raise webob.exc.HTTPGone(explanation=msg, request=req, content_type='text/plain') except ValueError as e: LOG.debug("Cannot save data for image %s: %s", image_id, e) self._restore(image_repo, image) raise webob.exc.HTTPBadRequest(explanation=unicode(e)) except exception.InvalidImageStatusTransition as e: msg = unicode(e) LOG.debug(msg) raise webob.exc.HTTPConflict(explanation=e.msg, request=req) except exception.Forbidden as e: msg = (_("Not allowed to upload image data for image %s") % image_id) LOG.debug(msg) raise webob.exc.HTTPForbidden(explanation=msg, request=req) except exception.NotFound as e: raise webob.exc.HTTPNotFound(explanation=e.msg) except exception.StorageFull as e: msg = _("Image storage media is full: %s") % e LOG.error(msg) self._restore(image_repo, image) raise webob.exc.HTTPRequestEntityTooLarge(explanation=msg, request=req) except exception.StorageQuotaFull as e: msg = _("Image exceeds the storage quota: %s") % e LOG.error(msg) self._restore(image_repo, image) raise webob.exc.HTTPRequestEntityTooLarge(explanation=msg, request=req) except exception.ImageSizeLimitExceeded as e: msg = _("The incoming image is too large: %s") % e LOG.error(msg) self._restore(image_repo, image) raise webob.exc.HTTPRequestEntityTooLarge(explanation=msg, request=req) except exception.StorageWriteDenied as e: msg = _("Insufficient permissions on image storage media: %s") % e LOG.error(msg) self._restore(image_repo, image) raise webob.exc.HTTPServiceUnavailable(explanation=msg, request=req) except webob.exc.HTTPError as e: LOG.error(_("Failed to upload image data due to HTTP error")) self._restore(image_repo, image) raise except Exception as e: LOG.exception(_("Failed to upload image data due to " "internal error")) self._restore(image_repo, image) raise
**** CubicPower OpenStack Study **** def download(self, req, image_id): image_repo = self.gateway.get_repo(req.context) try: image = image_repo.get(image_id) if not image.locations: raise exception.ImageDataNotFound() except exception.ImageDataNotFound as e: raise webob.exc.HTTPNoContent(explanation=e.msg) except exception.NotFound as e: raise webob.exc.HTTPNotFound(explanation=e.msg) except exception.Forbidden as e: raise webob.exc.HTTPForbidden(explanation=e.msg) return image
**** CubicPower OpenStack Study **** class RequestDeserializer(wsgi.JSONRequestDeserializer):
**** CubicPower OpenStack Study **** def upload(self, request): try: request.get_content_type(('application/octet-stream',)) except exception.InvalidContentType as e: raise webob.exc.HTTPUnsupportedMediaType(explanation=e.msg) image_size = request.content_length or None return {'size': image_size, 'data': request.body_file}
**** CubicPower OpenStack Study **** class ResponseSerializer(wsgi.JSONResponseSerializer):
**** CubicPower OpenStack Study **** def download(self, response, image): response.headers['Content-Type'] = 'application/octet-stream' try: # NOTE(markwash): filesystem store (and maybe others?) cause a # problem with the caching middleware if they are not wrapped in # an iterator very strange response.app_iter = iter(image.get_data()) except exception.Forbidden as e: raise webob.exc.HTTPForbidden(explanation=e.msg) #NOTE(saschpe): "response.app_iter = ..." currently resets Content-MD5 # (https://github.com/Pylons/webob/issues/86), so it should be set # afterwards for the time being. if image.checksum: response.headers['Content-MD5'] = image.checksum #NOTE(markwash): "response.app_iter = ..." also erroneously resets the # content-length response.headers['Content-Length'] = str(image.size)
**** CubicPower OpenStack Study **** def upload(self, response, result): response.status_int = 204 def create_resource(): """Image data resource factory method""" deserializer = RequestDeserializer() serializer = ResponseSerializer() controller = ImageDataController() return wsgi.Resource(controller, deserializer, serializer)
**** CubicPower OpenStack Study **** def create_resource(): """Image data resource factory method""" deserializer = RequestDeserializer() serializer = ResponseSerializer() controller = ImageDataController() return wsgi.Resource(controller, deserializer, serializer)
|