punkreq 0.1.0__py3-none-any.whl

punkreq/__init__.py

@@ -0,0 +1,84 @@
+from ._auth import Auth, BasicAuth, BearerAuth
+from ._client import USE_CLIENT_DEFAULT, ResponseHandle, UseClientDefault
+from ._config import Limits, Proxy, Timeout
+from ._content import AsyncByteStream, ByteStream
+from ._cookies import Cookies
+from ._exceptions import (
+    CloseError,
+    ConnectError,
+    ConnectTimeout,
+    CookieConflict,
+    DecodingError,
+    HTTPError,
+    HTTPStatusError,
+    InvalidURL,
+    LocalProtocolError,
+    NetworkError,
+    PoolTimeout,
+    ProtocolError,
+    ProxyError,
+    ReadError,
+    ReadTimeout,
+    RemoteProtocolError,
+    RequestError,
+    RequestNotRead,
+    StreamClosed,
+    StreamConsumed,
+    StreamError,
+    TimeoutException,
+    TooManyRedirects,
+    TransportError,
+    UnsupportedProtocol,
+    WriteError,
+)
+from ._headers import Headers
+from ._models import Request, Response
+from ._urls import URL, QueryParams
+from ._version import __version__ as __version__
+
+
+__all__ = [
+    "URL",
+    "USE_CLIENT_DEFAULT",
+    "AsyncByteStream",
+    "Auth",
+    "BasicAuth",
+    "BearerAuth",
+    "ByteStream",
+    "CloseError",
+    "ConnectError",
+    "ConnectTimeout",
+    "CookieConflict",
+    "Cookies",
+    "DecodingError",
+    "HTTPError",
+    "HTTPStatusError",
+    "Headers",
+    "InvalidURL",
+    "Limits",
+    "LocalProtocolError",
+    "NetworkError",
+    "PoolTimeout",
+    "ProtocolError",
+    "Proxy",
+    "ProxyError",
+    "QueryParams",
+    "ReadError",
+    "ReadTimeout",
+    "RemoteProtocolError",
+    "Request",
+    "RequestError",
+    "RequestNotRead",
+    "Response",
+    "ResponseHandle",
+    "StreamClosed",
+    "StreamConsumed",
+    "StreamError",
+    "Timeout",
+    "TimeoutException",
+    "TooManyRedirects",
+    "TransportError",
+    "UnsupportedProtocol",
+    "UseClientDefault",
+    "WriteError",
+]

punkreq/_auth.py

@@ -0,0 +1,43 @@
+from __future__ import annotations
+
+import base64
+import typing
+
+from ._models import Request
+
+
+__all__ = ["Auth", "BasicAuth", "BearerAuth"]
+
+AuthTypes = typing.Union["Auth", typing.Tuple[str, str], None]
+
+
+class Auth:
+    """Base class: override `apply` to mutate the outgoing request."""
+
+    def apply(self, request: Request) -> None:
+        raise NotImplementedError()
+
+
+class BasicAuth(Auth):
+    def __init__(self, username: str, password: str = "") -> None:
+        token = base64.b64encode(f"{username}:{password}".encode()).decode("ascii")
+        self._header = f"Basic {token}"
+
+    def apply(self, request: Request) -> None:
+        request.headers["authorization"] = self._header
+
+
+class BearerAuth(Auth):
+    def __init__(self, token: str) -> None:
+        self._header = f"Bearer {token}"
+
+    def apply(self, request: Request) -> None:
+        request.headers["authorization"] = self._header
+
+
+def coerce_auth(auth: AuthTypes) -> Auth | None:
+    if auth is None or isinstance(auth, Auth):
+        return auth
+    if isinstance(auth, tuple) and len(auth) == 2:
+        return BasicAuth(auth[0], auth[1])
+    raise TypeError(f"Invalid 'auth' argument: {auth!r}")