jsdio/voice_assistant
Template
1
0
Fork 0
mirror of https://github.com/ijaric/voice_assistant.git synced 2025-05-24 14:33:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: [#38] class inherits all methods of httpx.Asynclient

Browse Source
This commit is contained in:
ksieuk 2023-10-07 01:56:46 +03:00
parent fa601a3eca
commit c44c8d2503
1 changed files with 4 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/assistant/lib/clients/http_client.py
View File

@ -5,7 +5,7 @@ import httpx
import lib.app.split_settings as app_split_settings import lib.app.split_settings as app_split_settings
class AsyncHttpClient: class AsyncHttpClient(httpx.AsyncClient):
def __init__( def __init__(
self, self,
proxy_settings: app_split_settings.ProxySettings, proxy_settings: app_split_settings.ProxySettings,
@ -17,7 +17,7 @@ class AsyncHttpClient:
self.proxies = self.__get_proxies_from_settings() self.proxies = self.__get_proxies_from_settings()
self.client_params = client_params self.client_params = client_params
self.client = self._get_client() super().__init__(base_url=self.base_url, proxies=self.proxies, **client_params)
def __get_proxies_from_settings(self) -> dict[str, str] | None: def __get_proxies_from_settings(self) -> dict[str, str] | None:
if not self.proxy_settings.enable: if not self.proxy_settings.enable:
@ -25,14 +25,5 @@ class AsyncHttpClient:
proxies = {"all://": self.proxy_settings.dsn} proxies = {"all://": self.proxy_settings.dsn}
return proxies return proxies
def _get_client(self): async def close(self) -> None:
return httpx.AsyncClient( await self.aclose()
base_url=self.base_url,
proxies=self.proxies, # type: ignore
**self.client_params,
)
async def close(self):
if not self.client:
return
await self.client.aclose()