♻️⚡️ 重构 Service，优化部分代码

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: