summaryrefslogtreecommitdiff
path: root/ATRI/utils
diff options
context:
space:
mode:
Diffstat (limited to 'ATRI/utils')
-rw-r--r--ATRI/utils/file.py5
-rw-r--r--ATRI/utils/request.py7
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: