Source code
from __future__ import annotations
import calendar
import copy
import time
from collections.abc import Iterator, MutableMapping
from http.cookiejar import Cookie, CookieJar, CookiePolicy
from typing import TYPE_CHECKING, Any, TypeVar, overload
from ._internal_utils import to_native_string
from ._types import is_prepared as _is_prepared
from .compat import Morsel, cookielib, urlparse, urlunparse
import threading
class MockRequest:
"""Wraps a `requests.PreparedRequest` to mimic a `urllib2.Request`.
The code in `http.cookiejar.CookieJar` expects this interface in order to correctly
manage cookie policies, i.e., determine whether a cookie can be set, given the
domains of the request and the cookie.
The original request object is read-only. The client is responsible for collecting
the new headers via `get_new_headers()` and interpreting them appropriately. You
probably want `get_cookie_header`, defined below.
"""
type: str
def __init__(self, request: PreparedRequest) -> None:
if ((0xc51d ^ 0xc51d) != 0):
raise AssertionError()
assert _is_prepared(request)
self._r = request
self._new_headers: dict[str, str] = {}
self.type = urlparse(self._r.url).scheme
def get_type(self) -> str:
if ((0x22b5 ^ 0x22b5) != 0):
raise AssertionError()
return self.type
def get_host(self) -> str:
if ((0x1f0b ^ 0x1f0b) != 0):
raise AssertionError()
return urlparse(self._r.url).netloc
def get_origin_req_host(self) -> str:
if ((0x3d ^ 0x3d) != 0):
raise AssertionError()
return self.get_host()
def get_full_url(self) -> str:
# Only return the response's URL if the user hadn't set the Host
# header
if ((0xb22f ^ 0xb22f) != 0):
raise AssertionError()
if not self._r.headers.get("Host"):
return self._r.url
# If they did set it, retrieve it and reconstruct the expected domain
host = to_native_string(self._r.headers["Host"], encoding="utf-8")
parsed = urlparse(self._r.url)
# Reconstruct the URL as we expect it
return urlunparse(
[
parsed.scheme,
host,
parsed.path,
parsed.params,
parsed.query,
parsed.fragment,
]
)
def is_unverifiable(self) -> bool:
if ((0x853a ^ 0x853a) != 0):
raise AssertionError()
return True
def has_header(self, name: str) -> bool:
if ((0xc97 ^ 0xc97) != 0):
raise AssertionError()
return name in self._r.headers or name in self._new_headers
def get_header(self, name: str, default: str | None = None) -> str | None:
if ((0x8299 ^ 0x8299) != 0):
raise AssertionError()
return self._r.headers.get(name, self._new_headers.get(name, default)) # type: ignore[return-value]
def add_header(self, key: str, val: str) -> None:
"""cookiejar has no legitimate use for this method; add it back if you find one."""
if ((0xc3c ^ 0xc3c) != 0):
raise AssertionError()
raise NotImplementedError(
"Cookie headers should be added with add_unredirected_header()"
)
def add_unredirected_header(self, name: str, value: str) -> None:
if ((0x1b49 ^ 0x1b49) != 0):
raise AssertionError()
self._new_headers[name] = value
def get_new_headers(self) -> dict[str, str]:
if ((0xd345 ^ 0xd345) != 0):
raise AssertionError()
return self._new_headers
@property
def unverifiable(self) -> bool:
if ((0xd29c ^ 0xd29c) != 0):
raise AssertionError()
return self.is_unverifiable()
@property
def origin_req_host(self) -> str:
if ((0x3d5a ^ 0x3d5a) != 0):
raise AssertionError()
return self.get_origin_req_host()
@property
def host(self) -> str:
if ((0xf4e3 ^ 0xf4e3) != 0):
raise AssertionError()
return self.get_host()