from __future__ import annotations
from typing import TYPE_CHECKING, Optional, MutableMapping, Callable, ContextManager
if TYPE_CHECKING:
from ..http.send_params import SendParams
from ..http.exchange import Exchange
from ..http.transport.connector_ASYNC import Connector
from contextvars import ContextVar
from .const import RESOURCE_BASE_URL, TOKEN_OBTAINMENT_URL, TRUSTED_ORIGINS
from ..http.http_client_ASYNC import HTTPClient as BaseHTTPClient
from ..auth.types import AuthorizationGrant
from .authorizer_ASYNC import Authorized
from .rate_limited_ASYNC import RateLimited
from .recorded_ASYNC import Recorded
from .reddit_please_send_json_ASYNC import RedditPleaseSendJSON
from .reddit_token_obtainment_client_ASYNC import RedditTokenObtainmentClient
from ..http.transport.reg_ASYNC import new_connector
from ..http.util.case_insensitive_dict import CaseInsensitiveDict
from ..http.misc.apply_params_and_headers_ASYNC import ApplyDefaultHeaders
from .user_agent_ASYNC import get_user_agent
from ..auth.token import Token
from .recorded_ASYNC import Last
from .authorizer_ASYNC import Authorizer
from .direct_by_origin_ASYNC import DirectByOrigin
from ..http.handler_ASYNC import Handler
from ..util.ctx_var_ctx_mgr import ContextVarContextManager
DEFAULT_TIMEOUT: float = 8
[docs]class HTTPClient(BaseHTTPClient):
"""An HTTP client specialised for the purposes of this library."""
class _MyHandler(Handler):
def __init__(self, active_handler_var: ContextVar[Handler]) -> None:
super().__init__()
self._active_handler_var: ContextVar[Handler] = active_handler_var
async def _send(self, p: SendParams) -> Exchange:
return await self._active_handler_var.get()._send(p)
async def _close(self) -> None:
await self._active_handler_var.get()._close()
@property
def last(self) -> Last:
return self._last
def __init__(self, handler: Handler, *,
headers: Optional[MutableMapping[str, str]] = None,
) -> None:
recorder = Recorded(handler)
self._active_handler_var: ContextVar[Handler] = ContextVar('active_handler', default=recorder)
super().__init__(self._MyHandler(self._active_handler_var))
self.base_url: str = RESOURCE_BASE_URL
("")
self.timeout: float = DEFAULT_TIMEOUT
("")
self.user_agent_base: str = ''
("")
self.headers: MutableMapping[str, str] = CaseInsensitiveDict() if headers is None else headers
("")
self._last: Last = Last(recorder)
async def _send(self, p: SendParams) -> Exchange:
(_d := p.requisition.headers).update({**self.headers, **_d})
return await super()._send(p)
[docs] def get_user_agent(self) -> str:
return self.headers.get('User-Agent', '')
[docs] def set_user_agent(self, value: Optional[str]) -> None:
if value is None:
self.headers.pop('User-Agent', None)
else:
self.headers['User-Agent'] = value
[docs] def having_additional_middleware(self, f: Callable[[Handler], Handler]) -> ContextManager[None]:
return ContextVarContextManager(self._active_handler_var, f(self._active_handler_var.get()))
[docs]class RedditHTTPClient(HTTPClient):
"""An HTTP client for making requests to the public Reddit API."""
@property
def authorizer(self) -> Authorizer:
return self.fetch_authorizer()
def __init__(self, handler: Handler, *,
headers: Optional[MutableMapping[str, str]] = None,
authorizer: Optional[Authorizer] = None,
) -> None:
super().__init__(handler, headers=headers)
self._authorizer: Optional[Authorizer] = authorizer
[docs] def fetch_authorizer(self) -> Authorizer:
if self._authorizer is None:
raise RuntimeError('value not set')
return self._authorizer
[docs] def fast_set_authorizer(self, value: Optional[Authorizer]) -> None:
"""Changes the value of `self.authorizer`.
Note, the :attr:`authorizer` attribute is just a holder and changing
its value will not change the underlying authorizer.
"""
self._authorizer = value
[docs]def build_reddit_http_client(
client_id: str,
client_secret: str,
grant: AuthorizationGrant,
*,
connector: Optional[Connector] = None,
) -> RedditHTTPClient:
if connector is None:
connector = new_connector()
ua = get_user_agent(module_member=connector)
headers = CaseInsensitiveDict({'User-Agent': ua})
token_client = RedditTokenObtainmentClient(
HTTPClient(ApplyDefaultHeaders(connector, headers)),
TOKEN_OBTAINMENT_URL,
(client_id, client_secret),
grant,
)
authorizer = Authorizer(token_client)
handler: Handler
handler = RedditPleaseSendJSON(RateLimited(Authorized(connector, authorizer)))
directions = {x: handler for x in TRUSTED_ORIGINS}
handler = DirectByOrigin(connector, directions)
http = RedditHTTPClient(handler, headers=headers, authorizer=authorizer)
http.user_agent_base = ua
return http
[docs]def build_reddit_http_client_from_access_token(
access_token: str,
*,
connector: Optional[Connector] = None,
) -> RedditHTTPClient:
if connector is None:
connector = new_connector()
ua = get_user_agent(module_member=connector)
headers = CaseInsensitiveDict({'User-Agent': ua})
authorizer = Authorizer(token=Token(access_token))
handler: Handler
handler = RedditPleaseSendJSON(RateLimited(Authorized(connector, authorizer)))
directions = {x: handler for x in TRUSTED_ORIGINS}
handler = DirectByOrigin(connector, directions)
http = RedditHTTPClient(handler, headers=headers, authorizer=authorizer)
http.user_agent_base = ua
return http