diff options
Diffstat (limited to 'AyaBot/aio/requests.py')
-rw-r--r-- | AyaBot/aio/requests.py | 71 |
1 files changed, 71 insertions, 0 deletions
diff --git a/AyaBot/aio/requests.py b/AyaBot/aio/requests.py new file mode 100644 index 0000000..4f7b9d3 --- /dev/null +++ b/AyaBot/aio/requests.py @@ -0,0 +1,71 @@ +from typing import Optional, Any + +import requests +from requests import * + +from . import run_sync_func + + +class AsyncResponse: + def __init__(self, response: requests.Response): + self.raw_response = response + + @property + def ok(self) -> bool: + return self.raw_response.ok + + def __repr__(self): + return '<AsyncResponse [%s]>' % self.raw_response.status_code + + def __bool__(self): + return self.ok + + @property + async def content(self) -> Optional[bytes]: + return await run_sync_func(lambda: self.raw_response.content) + + @property + async def text(self) -> str: + return await run_sync_func(lambda: self.raw_response.text) + + async def json(self, **kwargs) -> Any: + return await run_sync_func(self.raw_response.json, **kwargs) + + +async def request(method, url, **kwargs) -> AsyncResponse: + return AsyncResponse(await run_sync_func(requests.request, + method=method, url=url, **kwargs)) + + +async def get(url, params=None, **kwargs) -> AsyncResponse: + return AsyncResponse( + await run_sync_func(requests.get, url=url, params=params, **kwargs)) + + +async def options(url, **kwargs) -> AsyncResponse: + return AsyncResponse( + await run_sync_func(requests.options, url=url, **kwargs)) + + +async def head(url, **kwargs) -> AsyncResponse: + return AsyncResponse(await run_sync_func(requests.head, url=url, **kwargs)) + + +async def post(url, data=None, json=None, **kwargs) -> AsyncResponse: + return AsyncResponse(await run_sync_func(requests.post, url=url, + data=data, json=json, **kwargs)) + + +async def put(url, data=None, **kwargs) -> AsyncResponse: + return AsyncResponse( + await run_sync_func(requests.put, url=url, data=data, **kwargs)) + + +async def patch(url, data=None, **kwargs) -> AsyncResponse: + return AsyncResponse( + await run_sync_func(requests.patch, url=url, data=data, **kwargs)) + + +async def delete(url, **kwargs) -> AsyncResponse: + return AsyncResponse( + await run_sync_func(requests.delete, url=url, **kwargs)) |