コード例 #1
0
ファイル: wsgi.py プロジェクト: iorchard/okidoki
    def serialize(self, request, content_type):
        """Serializes the wrapped object.

        Utility method for serializing the wrapped object.  Returns a
        webob.Response object.
        """

        serializer = self.serializer

        body = None
        if self.obj is not None:
            body = serializer.serialize(self.obj)
        response = webob.Response(body=body)
        if response.headers.get('Content-Length'):
            if six.PY3:
                response.headers['Content-Length'] = (str(
                    response.headers['Content-Length']))
            else:
                # NOTE: we need to encode 'Content-Length' header, since
                # webob.Response auto sets it if "body" attr is presented.
                # github.com/Pylons/webob/blob/1.5.0b0/webob/response.py#L147
                response.headers['Content-Length'] = utils.utf8(
                    response.headers['Content-Length'])
        response.status_int = self.code
        for hdr, value in self._headers.items():
            if six.PY3:
                response.headers[hdr] = str(value)
            else:
                response.headers[hdr] = utils.utf8(value)
        if six.PY3:
            response.headers['Content-Type'] = str(content_type)
        else:
            response.headers['Content-Type'] = utils.utf8(content_type)
        return response
コード例 #2
0
ファイル: wsgi.py プロジェクト: sampathP/openstack-masakari
    def serialize(self, request, content_type):
        """Serializes the wrapped object.

        Utility method for serializing the wrapped object.  Returns a
        webob.Response object.
        """

        serializer = self.serializer

        body = None
        if self.obj is not None:
            body = serializer.serialize(self.obj)
        response = webob.Response(body=body)
        if response.headers.get('Content-Length'):
            # NOTE: we need to encode 'Content-Length' header,
            # since webob.Response auto sets it if "body" attr is presented.
            # https://github.com/Pylons/webob/blob/1.5.0b0/webob/response.py#L147
            response.headers['Content-Length'] = utils.utf8(
                response.headers['Content-Length'])
        response.status_int = self.code
        for hdr, value in self._headers.items():
            response.headers[hdr] = utils.utf8(value)
        response.headers['Content-Type'] = utils.utf8(content_type)
        return response
コード例 #3
0
ファイル: wsgi.py プロジェクト: sampathP/openstack-masakari
    def _process_stack(self, request, action, action_args,
                       content_type, body, accept):
        """Implement the processing stack."""

        # Get the implementing method
        try:
            meth, extensions = self.get_method(request, action,
                                               content_type, body)
        except (AttributeError, TypeError):
            return Fault(webob.exc.HTTPNotFound())
        except KeyError as ex:
            msg = _("There is no such action: %s") % ex.args[0]
            return Fault(webob.exc.HTTPBadRequest(explanation=msg))
        except exception.MalformedRequestBody:
            msg = _("Malformed request body")
            return Fault(webob.exc.HTTPBadRequest(explanation=msg))

        if body:
            msg = _("Action: '%(action)s', calling method: %(meth)s, body: "
                    "%(body)s") % {'action': action,
                                   'body': six.text_type(body, 'utf-8'),
                                   'meth': str(meth)}
            LOG.debug(strutils.mask_password(msg))
        else:
            LOG.debug("Calling method '%(meth)s'",
                      {'meth': str(meth)})

        # Now, deserialize the request body...
        try:
            contents = {}
            if self._should_have_body(request):
                # allow empty body with PUT and POST
                if request.content_length == 0:
                    contents = {'body': None}
                else:
                    contents = self.deserialize(body)
        except exception.MalformedRequestBody:
            msg = _("Malformed request body")
            return Fault(webob.exc.HTTPBadRequest(explanation=msg))

        # Update the action args
        action_args.update(contents)

        project_id = action_args.pop("project_id", None)
        context = request.environ.get('masakari.context')
        if (context and project_id and (project_id != context.project_id)):
            msg = _("Malformed request URL: URL's project_id '%(project_id)s'"
                    " doesn't match Context's project_id"
                    " '%(context_project_id)s'") % {
                'project_id': project_id,
                'context_project_id': context.project_id
            }
            return Fault(webob.exc.HTTPBadRequest(explanation=msg))

        # Run pre-processing extensions
        response, post = self.pre_process_extensions(extensions,
                                                     request, action_args)

        if not response:
            try:
                with ResourceExceptionHandler():
                    action_result = self.dispatch(meth, request, action_args)
            except Fault as ex:
                response = ex

        if not response:
            # No exceptions; convert action_result into a
            # ResponseObject
            resp_obj = None
            if type(action_result) is dict or action_result is None:
                resp_obj = ResponseObject(action_result)
            elif isinstance(action_result, ResponseObject):
                resp_obj = action_result
            else:
                response = action_result

            # Run post-processing extensions
            if resp_obj:
                # Do a preserialize to set up the response object
                if hasattr(meth, 'wsgi_code'):
                    resp_obj._default_code = meth.wsgi_code
                # Process post-processing extensions
                response = self.post_process_extensions(post, resp_obj,
                                                        request, action_args)

            if resp_obj and not response:
                response = resp_obj.serialize(request, accept)

        if hasattr(response, 'headers'):
            for hdr, val in list(response.headers.items()):
                # Headers must be utf-8 strings
                response.headers[hdr] = utils.utf8(val)

            if not request.api_version_request.is_null():
                response.headers[API_VERSION_REQUEST_HEADER] = \
                    'masakari ' + request.api_version_request.get_string()
                response.headers.add('Vary', API_VERSION_REQUEST_HEADER)

        return response
コード例 #4
0
 def test_text_type_with_encoding(self):
     some_value = 'test\u2026config'
     self.assertEqual(some_value, utils.utf8(some_value).decode("utf-8"))
コード例 #5
0
 def test_not_text_type(self):
     return_value = utils.utf8(1)
     self.assertEqual(b"1", return_value)
     self.assertIsInstance(return_value, six.binary_type)
コード例 #6
0
 def test_bytes_value(self):
     some_value = b"fake data"
     return_value = utils.utf8(some_value)
     # check that type of returned value doesn't changed
     self.assertIsInstance(return_value, type(some_value))
     self.assertEqual(some_value, return_value)
コード例 #7
0
 def test_none_value(self):
     self.assertIsInstance(utils.utf8(None), type(None))
コード例 #8
0
ファイル: test_utils.py プロジェクト: openstack/masakari
 def test_text_type_with_encoding(self):
     some_value = 'test\u2026config'
     self.assertEqual(some_value, utils.utf8(some_value).decode("utf-8"))
コード例 #9
0
ファイル: test_utils.py プロジェクト: openstack/masakari
 def test_not_text_type(self):
     return_value = utils.utf8(1)
     self.assertEqual(b"1", return_value)
     self.assertIsInstance(return_value, six.binary_type)
コード例 #10
0
ファイル: test_utils.py プロジェクト: openstack/masakari
 def test_bytes_value(self):
     some_value = b"fake data"
     return_value = utils.utf8(some_value)
     # check that type of returned value doesn't changed
     self.assertIsInstance(return_value, type(some_value))
     self.assertEqual(some_value, return_value)
コード例 #11
0
ファイル: test_utils.py プロジェクト: openstack/masakari
 def test_none_value(self):
     self.assertIsInstance(utils.utf8(None), type(None))
コード例 #12
0
ファイル: wsgi.py プロジェクト: ynwssjx/masakari
    def _process_stack(self, request, action, action_args, content_type, body,
                       accept):
        """Implement the processing stack."""

        # Get the implementing method
        try:
            meth, extensions = self.get_method(request, action, content_type,
                                               body)
        except (AttributeError, TypeError):
            return Fault(webob.exc.HTTPNotFound())
        except KeyError as ex:
            msg = _("There is no such action: %s") % ex.args[0]
            return Fault(webob.exc.HTTPBadRequest(explanation=msg))
        except exception.MalformedRequestBody:
            msg = _("Malformed request body")
            return Fault(webob.exc.HTTPBadRequest(explanation=msg))
        except webob.exc.HTTPMethodNotAllowed as e:
            return Fault(e)

        if body:
            msg = _("Action: '%(action)s', calling method: %(meth)s, body: "
                    "%(body)s") % {
                        'action': action,
                        'body': six.text_type(body, 'utf-8'),
                        'meth': str(meth)
                    }
            LOG.debug(strutils.mask_password(msg))
        else:
            LOG.debug("Calling method '%(meth)s'", {'meth': str(meth)})

        # Now, deserialize the request body...
        try:
            contents = {}
            if self._should_have_body(request):
                # allow empty body with PUT and POST
                if request.content_length == 0:
                    contents = {'body': None}
                else:
                    contents = self.deserialize(body)
        except exception.MalformedRequestBody:
            msg = _("Malformed request body")
            return Fault(webob.exc.HTTPBadRequest(explanation=msg))

        # Update the action args
        action_args.update(contents)

        project_id = action_args.pop("project_id", None)
        context = request.environ.get('masakari.context')
        if (context and project_id and (project_id != context.project_id)):
            msg = _("Malformed request URL: URL's project_id '%(project_id)s'"
                    " doesn't match Context's project_id"
                    " '%(context_project_id)s'") % {
                        'project_id': project_id,
                        'context_project_id': context.project_id
                    }
            return Fault(webob.exc.HTTPBadRequest(explanation=msg))

        # Run pre-processing extensions
        response, post = self.pre_process_extensions(extensions, request,
                                                     action_args)

        if not response:
            try:
                with ResourceExceptionHandler():
                    action_result = self.dispatch(meth, request, action_args)
            except Fault as ex:
                response = ex

        if not response:
            # No exceptions; convert action_result into a
            # ResponseObject
            resp_obj = None
            if type(action_result) is dict or action_result is None:
                resp_obj = ResponseObject(action_result)
            elif isinstance(action_result, ResponseObject):
                resp_obj = action_result
            else:
                response = action_result

            # Run post-processing extensions
            if resp_obj:
                # Do a preserialize to set up the response object
                if hasattr(meth, 'wsgi_code'):
                    resp_obj._default_code = meth.wsgi_code
                # Process post-processing extensions
                response = self.post_process_extensions(
                    post, resp_obj, request, action_args)

            if resp_obj and not response:
                response = resp_obj.serialize(request, accept)

        if hasattr(response, 'headers'):
            for hdr, val in list(response.headers.items()):
                # Headers must be utf-8 strings
                response.headers[hdr] = utils.utf8(val)

            if not request.api_version_request.is_null():
                response.headers[API_VERSION_REQUEST_HEADER] = \
                    'masakari ' + request.api_version_request.get_string()
                response.headers.add('Vary', API_VERSION_REQUEST_HEADER)

        return response