impit 0.9.0__cp314-cp314-manylinux_2_28_aarch64.whl

impit/__init__.py

@@ -0,0 +1,96 @@
+from importlib import metadata
+from typing import Literal
+
+from .cookies import Cookies
+from .impit import (
+    AsyncClient,
+    Client,
+    CloseError,
+    ConnectError,
+    ConnectTimeout,
+    CookieConflict,
+    DecodingError,
+    HTTPError,
+    HTTPStatusError,
+    InvalidURL,
+    LocalProtocolError,
+    NetworkError,
+    PoolTimeout,
+    ProtocolError,
+    ProxyError,
+    ReadError,
+    ReadTimeout,
+    RemoteProtocolError,
+    RequestError,
+    RequestNotRead,
+    Response,
+    ResponseNotRead,
+    StreamClosed,
+    StreamConsumed,
+    StreamError,
+    TimeoutException,
+    TooManyRedirects,
+    TransportError,
+    UnsupportedProtocol,
+    WriteError,
+    WriteTimeout,
+    delete,
+    get,
+    head,
+    options,
+    patch,
+    post,
+    put,
+    stream,
+    trace,
+)
+
+__version__ = metadata.version('impit')
+
+__all__ = [
+    'AsyncClient',
+    'Browser',
+    'Client',
+    'CloseError',
+    'ConnectError',
+    'ConnectTimeout',
+    'CookieConflict',
+    'Cookies',
+    'DecodingError',
+    'HTTPError',
+    'HTTPStatusError',
+    'InvalidURL',
+    'LocalProtocolError',
+    'NetworkError',
+    'PoolTimeout',
+    'ProtocolError',
+    'ProxyError',
+    'ReadError',
+    'ReadTimeout',
+    'RemoteProtocolError',
+    'RequestError',
+    'RequestNotRead',
+    'Response',
+    'ResponseNotRead',
+    'StreamClosed',
+    'StreamConsumed',
+    'StreamError',
+    'TimeoutException',
+    'TooManyRedirects',
+    'TransportError',
+    'UnsupportedProtocol',
+    'WriteError',
+    'WriteTimeout',
+    'delete',
+    'get',
+    'head',
+    'options',
+    'patch',
+    'post',
+    'put',
+    'stream',
+    'trace',
+]
+
+
+Browser = Literal['chrome', 'firefox']

impit/cookies.py

@@ -0,0 +1,177 @@
+"""Copyright © 2019, [Encode OSS Ltd](https://www.encode.io/).
+
+All rights reserved.
+
+Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met:
+
+* Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer.
+
+* Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials provided with the distribution.
+
+* Neither the name of the copyright holder nor the names of its contributors may be used to endorse or promote products derived from this software without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+"""
+# ruff: noqa: SIM102, E501
+
+## The Cookies class below is a modification of the `httpx.Cookies` class, licensed under the BSD 3-Clause "New" License.
+
+from __future__ import annotations
+
+import typing
+from http.cookiejar import Cookie, CookieJar
+
+from .impit import CookieConflict
+
+CookieTypes = typing.Union['Cookies', CookieJar, dict[str, str], list[tuple[str, str]]]
+
+
+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:
+                if domain is None or cookie.domain == domain:
+                    if path is None or cookie.path == path:
+                        if value is not None:
+                            message = f'Multiple cookies exist with name={name}'
+                            raise CookieConflict(message)
+                        value = cookie.value
+
+        if value is None:
+            return default
+        return value
+
+    def delete(
+        self,
+        name: str,
+        domain: str | None = None,
+        path: str | None = None,
+    ) -> None:
+        """Delete a cookie by name.
+
+        May optionally include domain and path in order to specify exactly which cookie to delete.
+        """
+        if domain is not None and path is not None:
+            return self.jar.clear(domain, path, name)
+
+        remove = [
+            cookie
+            for cookie in self.jar
+            if cookie.name == name
+            and (domain is None or cookie.domain == domain)
+            and (path is None or cookie.path == path)
+        ]
+
+        for cookie in remove:
+            self.jar.clear(cookie.domain, cookie.path, cookie.name)
+        return None
+
+    def clear(self, domain: str | None = None, path: str | None = None) -> None:
+        """Delete all cookies.
+
+        Optionally include a domain and path in order to only delete a subset of all the cookies.
+        """
+        args = []
+        if domain is not None:
+            args.append(domain)
+        if path is not None:
+            assert domain is not None  # noqa: S101
+            args.append(path)
+        self.jar.clear(*args)
+
+    def update(self, cookies: CookieTypes | None = None) -> None:  # type: ignore[override]
+        """Update the cookie jar with new cookies. Accepts various types."""
+        cookies = Cookies(cookies)
+        for cookie in cookies.jar:
+            self.jar.set_cookie(cookie)
+
+    def __setitem__(self, name: str, value: str) -> None:
+        """Set a cookie by name."""
+        return self.set(name, value)
+
+    def __getitem__(self, name: str) -> str:
+        """Get a cookie by name."""
+        value = self.get(name)
+        if value is None:
+            raise KeyError(name)
+        return value
+
+    def __delitem__(self, name: str) -> None:
+        """Delete a cookie by name."""
+        return self.delete(name)
+
+    def __len__(self) -> int:
+        """Return the number of cookies in the jar."""
+        return len(self.jar)
+
+    def __iter__(self) -> typing.Iterator[str]:
+        """Return an iterator over the names of the cookies in the jar."""
+        return (cookie.name for cookie in self.jar)
+
+    def __bool__(self) -> bool:
+        """Return True if there are any cookies in the jar."""
+        for _ in self.jar:
+            return True
+        return False
+
+    def __repr__(self) -> str:
+        """Return a string representation of the Cookies object."""
+        cookies_repr = ', '.join(
+            [f'<Cookie {cookie.name}={cookie.value} for {cookie.domain} />' for cookie in self.jar]
+        )
+
+        return f'<Cookies[{cookies_repr}]>'