You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
79 lines
2.3 KiB
79 lines
2.3 KiB
5 years ago
|
from ..utils import cached_property
|
||
|
|
from .auth import WWWAuthenticateMixin
|
||
|
|
from .base_response import BaseResponse
|
||
|
|
from .common_descriptors import CommonResponseDescriptorsMixin
|
||
|
|
from .etag import ETagResponseMixin
|
||
|
|
|
||
|
|
|
||
|
|
class ResponseStream(object):
|
||
|
|
"""A file descriptor like object used by the :class:`ResponseStreamMixin` to
|
||
|
|
represent the body of the stream. It directly pushes into the response
|
||
|
|
iterable of the response object.
|
||
|
|
"""
|
||
|
|
|
||
|
|
mode = "wb+"
|
||
|
|
|
||
|
|
def __init__(self, response):
|
||
|
|
self.response = response
|
||
|
|
self.closed = False
|
||
|
|
|
||
|
|
def write(self, value):
|
||
|
|
if self.closed:
|
||
|
|
raise ValueError("I/O operation on closed file")
|
||
|
|
self.response._ensure_sequence(mutable=True)
|
||
|
|
self.response.response.append(value)
|
||
|
|
self.response.headers.pop("Content-Length", None)
|
||
|
|
return len(value)
|
||
|
|
|
||
|
|
def writelines(self, seq):
|
||
|
|
for item in seq:
|
||
|
|
self.write(item)
|
||
|
|
|
||
|
|
def close(self):
|
||
|
|
self.closed = True
|
||
|
|
|
||
|
|
def flush(self):
|
||
|
|
if self.closed:
|
||
|
|
raise ValueError("I/O operation on closed file")
|
||
|
|
|
||
|
|
def isatty(self):
|
||
|
|
if self.closed:
|
||
|
|
raise ValueError("I/O operation on closed file")
|
||
|
|
return False
|
||
|
|
|
||
|
|
def tell(self):
|
||
|
|
self.response._ensure_sequence()
|
||
|
|
return sum(map(len, self.response.response))
|
||
|
|
|
||
|
|
@property
|
||
|
|
def encoding(self):
|
||
|
|
return self.response.charset
|
||
|
|
|
||
|
|
|
||
|
|
class ResponseStreamMixin(object):
|
||
|
|
"""Mixin for :class:`BaseRequest` subclasses. Classes that inherit from
|
||
|
|
this mixin will automatically get a :attr:`stream` property that provides
|
||
|
|
a write-only interface to the response iterable.
|
||
|
|
"""
|
||
|
|
|
||
|
|
@cached_property
|
||
|
|
def stream(self):
|
||
|
|
"""The response iterable as write-only stream."""
|
||
|
|
return ResponseStream(self)
|
||
|
|
|
||
|
|
|
||
|
|
class Response(
|
||
|
|
BaseResponse,
|
||
|
|
ETagResponseMixin,
|
||
|
|
ResponseStreamMixin,
|
||
|
|
CommonResponseDescriptorsMixin,
|
||
|
|
WWWAuthenticateMixin,
|
||
|
|
):
|
||
|
|
"""Full featured response object implementing the following mixins:
|
||
|
|
|
||
|
|
- :class:`ETagResponseMixin` for etag and cache control handling
|
||
|
|
- :class:`ResponseStreamMixin` to add support for the `stream` property
|
||
|
|
- :class:`CommonResponseDescriptorsMixin` for various HTTP descriptors
|
||
|
|
- :class:`WWWAuthenticateMixin` for HTTP authentication support
|
||
|
|
"""
|