diff options
Diffstat (limited to 'ATRI/utils/request.py')
-rw-r--r-- | ATRI/utils/request.py | 57 |
1 files changed, 7 insertions, 50 deletions
diff --git a/ATRI/utils/request.py b/ATRI/utils/request.py index 6ce3903..c208bbf 100644 --- a/ATRI/utils/request.py +++ b/ATRI/utils/request.py @@ -1,54 +1,11 @@ -from typing import Optional -from aiohttp import ClientSession, ClientResponse, FormData +import httpx -class Response: - def __init__(self, response: ClientResponse) -> None: - self.raw_response = response +async def get(url: str, **kwargs): + async with httpx.AsyncClient(proxies=proxy) as client: + return await client.get(url, **kwargs) - @property - def status(self) -> int: - return self.raw_response.status - @property - def url(self): - return self.raw_response.url - - @property - def real_url(self): - return self.raw_response.real_url - - @property - def host(self): - return self.raw_response.host - - @property - def headers(self): - return self.raw_response.headers - - @property - def raw_headers(self): - return self.raw_response.raw_headers - - @property - def request_info(self): - return self.raw_response.request_info - - @property - async def read(self): - return await self.raw_response.read() - - @property - async def text(self): - return await self.raw_response.text() - - async def json(self): - return await self.raw_response.json() - - -async def get(url, params: dict = None, **kwargs) -> Response: - return Response(await ClientSession().get(url=url, params=params, **kwargs)) - - -async def post(url, data: Optional[FormData] = None, **kwargs) -> Response: - return Response(await ClientSession().post(url=url, data=data, **kwargs)) +async def post(url: str, **kwargs): + async with httpx.AsyncClient(proxies=proxy) as client: + return await client.post(url, **kwargs) |