diff options
Diffstat (limited to 'ATRI/utils')
-rw-r--r-- | ATRI/utils/file.py | 5 | ||||
-rw-r--r-- | ATRI/utils/request.py | 7 |
2 files changed, 11 insertions, 1 deletions
diff --git a/ATRI/utils/file.py b/ATRI/utils/file.py index 4699f5d..7afbead 100644 --- a/ATRI/utils/file.py +++ b/ATRI/utils/file.py @@ -1,7 +1,10 @@ import aiofiles +import urllib from pathlib import Path -from ATRI.exceptions import WriteError +from ATRI.exceptions import RequestTimeOut, WriteError + +from .request import get_content async def write_file(path: Path, text, encoding='utf-8') -> None: diff --git a/ATRI/utils/request.py b/ATRI/utils/request.py index 31285e8..145b470 100644 --- a/ATRI/utils/request.py +++ b/ATRI/utils/request.py @@ -18,6 +18,13 @@ async def get_bytes(url: str, headers: Optional[dict] = None) -> bytes: return result +async def get_content(url: str, headers: Optional[dict] = None): + async with ClientSession() as session: + async with session.get(url, headers=headers) as r: + result = await r.content.read() + return result + + async def post_bytes(url: str, params: Optional[dict] = None, json: Optional[dict] = None) -> bytes: |