from django.core.handlers.wsgi import STATUS_CODE_TEXT
from django.template.response import SimpleTemplateResponse


class Response(SimpleTemplateResponse):
    """
    An HttpResponse that allows it's data to be rendered into
    arbitrary media types.
    """

    def __init__(self, data=None, status=None, headers=None,
                 renderer=None, accepted_media_type=None):
        """
        Alters the init arguments slightly.
        For example, drop 'template_name', and instead use 'data'.

        Setting 'renderer' and 'media_type' will typically be defered,
        For example being set automatically by the `APIView`.
        """
        super(Response, self).__init__(None, status=status)
        self.data = data
        self.headers = headers and headers[:] or []
        self.renderer = renderer

        # Accepted media type is the portion of the request Accept header
        # that the renderer satisfied.  It could be '*/*', or somthing like
        # application/json; indent=4
        #
        # This is NOT the value that will be returned in the 'Content-Type'
        # header, but we do need to know the value in case there are
        # any specific parameters which affect the rendering process.
        self.accepted_media_type = accepted_media_type

    @property
    def rendered_content(self):
        assert self.renderer, "No renderer set on Response"

        self['Content-Type'] = self.renderer.media_type
        if self.data is None:
            return self.renderer.render()
        render_media_type = self.accepted_media_type or self.renderer.media_type
        return self.renderer.render(self.data, render_media_type)

    @property
    def status_text(self):
        """
        Returns reason text corresponding to our HTTP response status code.
        Provided for convenience.
        """
        return STATUS_CODE_TEXT.get(self.status_code, '')