diff options
Diffstat (limited to 'ATRI/utils/request.py')
-rw-r--r-- | ATRI/utils/request.py | 26 |
1 files changed, 14 insertions, 12 deletions
diff --git a/ATRI/utils/request.py b/ATRI/utils/request.py index 6a7e3ba..46a78f0 100644 --- a/ATRI/utils/request.py +++ b/ATRI/utils/request.py @@ -1,46 +1,48 @@ -from aiohttp import ClientSession, ClientResponse +from typing import Optional +from aiohttp import ClientSession, ClientResponse, FormData class Response: + def __init__(self, response: ClientResponse) -> None: self.raw_response = response - + @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() @@ -49,5 +51,5 @@ async def get(url, params: dict = None, **kwargs) -> Response: return Response(await ClientSession().get(url=url, params=params, **kwargs)) -async def post(url, data: dict = None, **kwargs) -> Response: +async def post(url, data: Optional[FormData] = None, **kwargs) -> Response: return Response(await ClientSession().post(url=url, data=data, **kwargs)) |