| 449 | |
| 450 | |
| 451 | class BaseHTTPResponse(io.IOBase): |
| 452 | CONTENT_DECODERS = ["gzip", "x-gzip", "deflate"] |
| 453 | if brotli is not None: |
| 454 | CONTENT_DECODERS += ["br"] |
| 455 | if HAS_ZSTD: |
| 456 | CONTENT_DECODERS += ["zstd"] |
| 457 | REDIRECT_STATUSES = [301, 302, 303, 307, 308] |
| 458 | |
| 459 | DECODER_ERROR_CLASSES: tuple[type[Exception], ...] = (IOError, zlib.error) |
| 460 | if brotli is not None: |
| 461 | DECODER_ERROR_CLASSES += (brotli.error,) |
| 462 | |
| 463 | if HAS_ZSTD: |
| 464 | DECODER_ERROR_CLASSES += (zstd.ZstdError,) |
| 465 | |
| 466 | def __init__( |
| 467 | self, |
| 468 | *, |
| 469 | headers: typing.Mapping[str, str] | typing.Mapping[bytes, bytes] | None = None, |
| 470 | status: int, |
| 471 | version: int, |
| 472 | version_string: str, |
| 473 | reason: str | None, |
| 474 | decode_content: bool, |
| 475 | request_url: str | None, |
| 476 | retries: Retry | None = None, |
| 477 | ) -> None: |
| 478 | if isinstance(headers, HTTPHeaderDict): |
| 479 | self.headers = headers |
| 480 | else: |
| 481 | self.headers = HTTPHeaderDict(headers) # type: ignore[arg-type] |
| 482 | self.status = status |
| 483 | self.version = version |
| 484 | self.version_string = version_string |
| 485 | self.reason = reason |
| 486 | self.decode_content = decode_content |
| 487 | self._has_decoded_content = False |
| 488 | self._request_url: str | None = request_url |
| 489 | self.retries = retries |
| 490 | |
| 491 | self.chunked = False |
| 492 | tr_enc = self.headers.get("transfer-encoding", "").lower() |
| 493 | # Don't incur the penalty of creating a list and then discarding it |
| 494 | encodings = (enc.strip() for enc in tr_enc.split(",")) |
| 495 | if "chunked" in encodings: |
| 496 | self.chunked = True |
| 497 | |
| 498 | self._decoder: ContentDecoder | None = None |
| 499 | self.length_remaining: int | None |
| 500 | |
| 501 | def get_redirect_location(self) -> str | None | typing.Literal[False]: |
| 502 | """ |
| 503 | Should we redirect and where to? |
| 504 | |
| 505 | :returns: Truthy redirect location string if we got a redirect status |
| 506 | code and valid location. ``None`` if redirect status and no |
| 507 | location. ``False`` if not a redirect status code. |
| 508 | """ |