From da888ff020805a38a17e5f83705aeb42ffa992ba Mon Sep 17 00:00:00 2001
From: Kyomotoi <1172294279@qq.com>
Date: Sun, 7 Mar 2021 15:24:17 +0800
Subject: =?UTF-8?q?=E2=99=BB=EF=B8=8F=E2=9A=A1=EF=B8=8F=20=E9=87=8D?=
 =?UTF-8?q?=E6=9E=84=20Service=EF=BC=8C=E4=BC=98=E5=8C=96=E9=83=A8?=
 =?UTF-8?q?=E5=88=86=E4=BB=A3=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ATRI/utils/file.py    | 5 ++++-
 ATRI/utils/request.py | 7 +++++++
 2 files changed, 11 insertions(+), 1 deletion(-)

(limited to 'ATRI/utils')

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:
-- 
cgit v1.2.3