Source code for falcon.http_error

# Copyright 2013 by Rackspace Hosting, Inc.
#
# 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.
"""HTTPError exception class."""

from __future__ import annotations

from typing import MutableMapping, Optional, Type, TYPE_CHECKING, Union
import xml.etree.ElementTree as et

from falcon.constants import MEDIA_JSON
from falcon.util import deprecation
from falcon.util import misc
from falcon.util import uri

if TYPE_CHECKING:
    from falcon._typing import HeaderArg
    from falcon._typing import Link
    from falcon._typing import ResponseStatus
    from falcon.media import BaseHandler


[docs] class HTTPError(Exception): """Represents a generic HTTP error. Raise an instance or subclass of ``HTTPError`` to have Falcon return a formatted error response and an appropriate HTTP status code to the client when something goes wrong. JSON and XML media types are supported by default. To customize the error presentation, implement a custom error serializer and set it on the :class:`~.App` instance via :meth:`~.App.set_error_serializer`. To customize what data is passed to the serializer, subclass ``HTTPError`` and override the ``to_dict()`` method (``to_json()`` is implemented via ``to_dict()``). `status` is the only positional argument allowed, the other arguments are defined as keyword-only. Args: status (Union[str,int]): HTTP status code or line (e.g., ``'400 Bad Request'``). This may be set to a member of :class:`http.HTTPStatus`, an HTTP status line string or byte string (e.g., ``'200 OK'``), or an ``int``. Keyword Args: title (str): Human-friendly error title. If not provided, defaults to the HTTP status line as determined by the ``status`` argument. description (str): Human-friendly description of the error, along with a helpful suggestion or two (default ``None``). headers (dict or list): A ``dict`` of header names and values to set, or a ``list`` of (*name*, *value*) tuples. Both *name* and *value* must be of type ``str`` or ``StringType``, and only character values 0x00 through 0xFF may be used on platforms that use wide characters. Note: The Content-Type header, if present, will be overridden. If you wish to return custom error messages, you can create your own HTTP error class, and install an error handler to convert it into an appropriate HTTP response for the client Note: Falcon can process a list of ``tuple`` slightly faster than a ``dict``. href (str): A URL someone can visit to find out more information (default ``None``). Unicode characters are percent-encoded. href_text (str): If href is given, use this as the friendly title/description for the link (default 'App documentation for this error'). code (int): An internal code that customers can reference in their support request or to help them when searching for knowledge base articles related to this error (default ``None``). """ __slots__ = ( 'status', 'title', 'description', 'headers', 'link', 'code', ) status: ResponseStatus """HTTP status code or line (e.g., ``'200 OK'``). This may be set to a member of :class:`http.HTTPStatus`, an HTTP status line string or byte string (e.g., ``'200 OK'``), or an ``int``. """ title: str """Error title to send to the client. Derived from the ``status`` if not provided. """ description: Optional[str] """Description of the error to send to the client.""" headers: Optional[HeaderArg] """Extra headers to add to the response.""" link: Optional[Link] """An href that the client can provide to the user for getting help.""" code: Optional[int] """An internal application code that a user can reference when requesting support for the error. """ def __init__( self, status: ResponseStatus, *, title: Optional[str] = None, description: Optional[str] = None, headers: Optional[HeaderArg] = None, href: Optional[str] = None, href_text: Optional[str] = None, code: Optional[int] = None, ): self.status = status # TODO(kgriffs): HTTP/2 does away with the "reason phrase". Eventually # we'll probably switch over to making everything code-based to more # easily support HTTP/2. When that happens, should we continue to # include the reason phrase in the title? self.title = title or misc.code_to_http_status(status) self.description = description self.headers = headers self.code = code if href: self.link = { 'text': href_text or 'Documentation related to this error', 'href': uri.encode(href), 'rel': 'help', } else: self.link = None def __repr__(self) -> str: return '<%s: %s>' % (self.__class__.__name__, self.status) __str__ = __repr__ @property def status_code(self) -> int: """HTTP status code normalized from the ``status`` argument passed to the initializer. """ # noqa: D205 return misc.http_status_to_code(self.status)
[docs] def to_dict( self, obj_type: Type[MutableMapping[str, Union[str, int, None, Link]]] = dict ) -> MutableMapping[str, Union[str, int, None, Link]]: """Return a basic dictionary representing the error. This method can be useful when serializing the error to hash-like media types, such as YAML, JSON, and MessagePack. Args: obj_type: A dict-like type that will be used to store the error information (default ``dict``). Returns: dict: A dictionary populated with the error's title, description, etc. """ obj = obj_type() obj['title'] = self.title if self.description is not None: obj['description'] = self.description if self.code is not None: obj['code'] = self.code if self.link is not None: obj['link'] = self.link return obj
[docs] def to_json(self, handler: Optional[BaseHandler] = None) -> bytes: """Return a JSON representation of the error. Args: handler: Handler object that will be used to serialize the representation of this error to JSON. When not provided, a default handler using the builtin JSON library will be used (default ``None``). Returns: bytes: A JSON document for the error. """ obj = self.to_dict() if handler is None: handler = _DEFAULT_JSON_HANDLER # NOTE: the json handler requires the sync serialize interface return handler.serialize(obj, MEDIA_JSON)
def _to_xml(self) -> bytes: """Return an XML-encoded representation of the error.""" error_element = et.Element('error') et.SubElement(error_element, 'title').text = self.title if self.description is not None: et.SubElement(error_element, 'description').text = self.description if self.code is not None: et.SubElement(error_element, 'code').text = str(self.code) if self.link is not None: link_element = et.SubElement(error_element, 'link') for key in ('text', 'href', 'rel'): et.SubElement(link_element, key).text = self.link[key] return b'<?xml version="1.0" encoding="UTF-8"?>' + et.tostring( error_element, encoding='utf-8' )
[docs] @deprecation.deprecated( 'The internal error serialization to XML is deprecated. ' 'Please serialize the output of to_dict() to XML instead.' ) def to_xml(self) -> bytes: """Return an XML-encoded representation of the error. Returns: bytes: An XML document for the error. .. deprecated:: 4.0 Automatic error serialization to XML is deprecated. Please serialize the output of :meth:`to_dict` to XML instead. """ return self._to_xml()
# NOTE: initialized in falcon.media.json, that is always imported since Request/Response # are imported by falcon init. if TYPE_CHECKING: _DEFAULT_JSON_HANDLER: BaseHandler else: _DEFAULT_JSON_HANDLER = None