mirror of
https://github.com/python-kasa/python-kasa.git
synced 2025-10-12 02:18:02 +00:00
Encapsulate http client dependency (#642)
* Encapsulate http client dependency * Store cookie dict as variable * Update post-review
This commit is contained in:
89
kasa/httpclient.py
Normal file
89
kasa/httpclient.py
Normal file
@@ -0,0 +1,89 @@
|
||||
"""Module for HttpClientSession class."""
|
||||
import logging
|
||||
from typing import Any, Dict, Optional, Tuple, Type, Union
|
||||
|
||||
import httpx
|
||||
|
||||
from .deviceconfig import DeviceConfig
|
||||
from .exceptions import ConnectionException, SmartDeviceException, TimeoutException
|
||||
|
||||
logging.getLogger("httpx").propagate = False
|
||||
|
||||
InnerHttpType = Type[httpx.AsyncClient]
|
||||
|
||||
|
||||
class HttpClient:
|
||||
"""HttpClient Class."""
|
||||
|
||||
def __init__(self, config: DeviceConfig) -> None:
|
||||
self._config = config
|
||||
self._client: httpx.AsyncClient = None
|
||||
|
||||
@property
|
||||
def client(self) -> httpx.AsyncClient:
|
||||
"""Return the underlying http client."""
|
||||
if self._config.http_client and issubclass(
|
||||
self._config.http_client.__class__, httpx.AsyncClient
|
||||
):
|
||||
return self._config.http_client
|
||||
|
||||
if not self._client:
|
||||
self._client = httpx.AsyncClient()
|
||||
return self._client
|
||||
|
||||
async def post(
|
||||
self,
|
||||
url: str,
|
||||
*,
|
||||
params: Optional[Dict[str, Any]] = None,
|
||||
data: Optional[bytes] = None,
|
||||
json: Optional[Dict] = None,
|
||||
headers: Optional[Dict[str, str]] = None,
|
||||
cookies_dict: Optional[Dict[str, str]] = None,
|
||||
) -> Tuple[int, Optional[Union[Dict, bytes]]]:
|
||||
"""Send an http post request to the device."""
|
||||
response_data = None
|
||||
cookies = None
|
||||
if cookies_dict:
|
||||
cookies = httpx.Cookies()
|
||||
for name, value in cookies_dict.items():
|
||||
cookies.set(name, value)
|
||||
self.client.cookies.clear()
|
||||
try:
|
||||
resp = await self.client.post(
|
||||
url,
|
||||
params=params,
|
||||
data=data,
|
||||
json=json,
|
||||
timeout=self._config.timeout,
|
||||
cookies=cookies,
|
||||
headers=headers,
|
||||
)
|
||||
except httpx.ConnectError as ex:
|
||||
raise ConnectionException(
|
||||
f"Unable to connect to the device: {self._config.host}: {ex}"
|
||||
) from ex
|
||||
except httpx.TimeoutException as ex:
|
||||
raise TimeoutException(
|
||||
"Unable to query the device, " + f"timed out: {self._config.host}: {ex}"
|
||||
) from ex
|
||||
except Exception as ex:
|
||||
raise SmartDeviceException(
|
||||
f"Unable to query the device: {self._config.host}: {ex}"
|
||||
) from ex
|
||||
|
||||
if resp.status_code == 200:
|
||||
response_data = resp.json() if json else resp.content
|
||||
|
||||
return resp.status_code, response_data
|
||||
|
||||
def get_cookie(self, cookie_name: str) -> str:
|
||||
"""Return the cookie with cookie_name."""
|
||||
return self._client.cookies.get(cookie_name)
|
||||
|
||||
async def close(self) -> None:
|
||||
"""Close the protocol."""
|
||||
client = self._client
|
||||
self._client = None
|
||||
if client:
|
||||
await client.aclose()
|
Reference in New Issue
Block a user