From f5ceb8927f2e7f2a9e29d62c8e4cef876f917249 Mon Sep 17 00:00:00 2001 From: Kyomotoi <1172294279@qq.com> Date: Sat, 6 Feb 2021 00:32:26 +0800 Subject: =?UTF-8?q?=F0=9F=8F=97=20=F0=9F=92=A9=20=E6=9B=B4=E6=94=B9?= =?UTF-8?q?=E9=A1=B9=E7=9B=AE=E7=BB=93=E6=9E=84=EF=BC=8C=E4=BF=AE=E5=A4=8D?= =?UTF-8?q?=E5=95=A5b=20BUG?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ATRI/request.py | 68 --------------------------------------------------------- 1 file changed, 68 deletions(-) delete mode 100644 ATRI/request.py (limited to 'ATRI/request.py') diff --git a/ATRI/request.py b/ATRI/request.py deleted file mode 100644 index 78bad6e..0000000 --- a/ATRI/request.py +++ /dev/null @@ -1,68 +0,0 @@ -import os -import string -import aiofiles -from random import sample -from pathlib import Path -from typing import Optional -from aiohttp.client import ClientSession - -from .exceptions import InvalidWriteText - - -TEMP_PATH = Path('.') / 'ATRI' / 'data' / 'temp' -os.makedirs(TEMP_PATH, exist_ok=True) - - -class Request: - @staticmethod - async def get_text(url: str, headers: Optional[dict] = None) -> str: - ''' - 异步方式请求接口 - :return: text - ''' - async with ClientSession() as session: - async with session.get(url, headers=headers) as r: - result = await r.text() - return result - - @staticmethod - async def get_bytes(url: str, headers: Optional[dict] = None) -> bytes: - ''' - 异步方式请求接口 - :return: bytes - ''' - async with ClientSession() as session: - async with session.get(url, headers=headers) as r: - result = await r.read() - return result - - @classmethod - async def get_image(cls, url: str, params: Optional[dict] = None) -> str: - ''' - 异步方式下载图片 - :return: img file - ''' - file_path = TEMP_PATH / 'img' - os.makedirs(file_path, exist_ok=True) - file_name = ''.join(sample(string.ascii_letters + string.digits, 16)) - path = os.path.abspath(file_path) - path = path + f'\\{file_name}.png' - async with ClientSession() as session: - async with session.get(url) as res: - try: - async with aiofiles.open(path, 'wb') as f: - await f.write(await res.read()) - except InvalidWriteText: - raise InvalidWriteText('Writing file failed!') - return path - - @staticmethod - async def post_bytes(url: str, params: Optional[dict] = None) -> bytes: - ''' - 异步方式 Post 接口 - :return: bytes - ''' - async with ClientSession() as session: - async with session.post(url, params=params) as r: - result = await r.read() - return result \ No newline at end of file -- cgit v1.2.3