Chris Coutinho
35 lines
815 B
Python
35 lines
815 B
Python
"""Bearer token authentication for httpx."""
|
|
|
|
from httpx import Auth, Request
|
|
|
|
|
|
class BearerAuth(Auth):
|
|
"""
|
|
Bearer token authentication flow for httpx.
|
|
|
|
This auth class adds the Authorization: Bearer <token> header
|
|
to all outgoing requests.
|
|
"""
|
|
|
|
def __init__(self, token: str):
|
|
"""
|
|
Initialize bearer authentication.
|
|
|
|
Args:
|
|
token: The bearer token to use for authentication
|
|
"""
|
|
self.token = token
|
|
|
|
def auth_flow(self, request: Request):
|
|
"""
|
|
Add Authorization header to the request.
|
|
|
|
Args:
|
|
request: The outgoing HTTP request
|
|
|
|
Yields:
|
|
The modified request with Authorization header
|
|
"""
|
|
request.headers["Authorization"] = f"Bearer {self.token}"
|
|
yield request
|