Class Cookies

impit-python/python/impit/cookies.py:29–177

HTTP Cookies, as a mutable mapping.

class Cookies(typing.MutableMapping[str, str]):
    """HTTP Cookies, as a mutable mapping."""

    def __init__(self, cookies: CookieTypes | None = None) -> None:
        if cookies is None or isinstance(cookies, dict):
            self.jar = CookieJar()
            if isinstance(cookies, dict):
                for key, value in cookies.items():
                    self.set(key, value)
        elif isinstance(cookies, list):
            self.jar = CookieJar()
            for key, value in cookies:
                self.set(key, value)
        elif isinstance(cookies, Cookies):
            self.jar = CookieJar()
            for cookie in cookies.jar:
                self.jar.set_cookie(cookie)
        else:
            self.jar = cookies

    def set(self, name: str, value: str, domain: str = '', path: str = '/') -> None:
        """Set a cookie value by name. May optionally include domain and path."""
        kwargs = {
            'version': 0,
            'name': name,
            'value': value,
            'port': None,
            'port_specified': False,
            'domain': domain,
            'domain_specified': bool(domain),
            'domain_initial_dot': domain.startswith('.'),
            'path': path,
            'path_specified': bool(path),
            'secure': False,
            'expires': None,
            'discard': True,
            'comment': None,
            'comment_url': None,
            'rest': {'HttpOnly': None},
            'rfc2109': False,
        }
        cookie = Cookie(**kwargs)  # type: ignore[arg-type]
        self.jar.set_cookie(cookie)

    def get(  # type: ignore[override]
        self,
        name: str,
        default: str | None = None,
        domain: str | None = None,
        path: str | None = None,
    ) -> str | None:
        """Get a cookie by name.

        May optionally include domain and path in order to specify exactly which cookie to retrieve.
        """
        value = None
        for cookie in self.jar:
            if cookie.name == name: