Diuse/zhenxun_bot
1
0
Fork 0
mirror of https://github.com/zhenxun-org/zhenxun_bot.git synced 2025-12-15 14:22:55 +08:00
Code Issues Actions 7 Packages Projects Releases Wiki Activity

Create data_source.py

Browse Source
This commit is contained in:
HibiKier 2021-08-02 19:19:22 +08:00 committed by GitHub
parent 4ca6925327
commit b62399e203
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
plugins/send_setu/data_source.py
View File

@ -101,7 +101,7 @@ headers = {
} }
async def search_online_setu(url_: str) -> "MessageSegment, int": async def search_online_setu(url_: str, id_: int = None, path_: str = None) -> "MessageSegment, int":
if "i.pixiv.cat" in url_: if "i.pixiv.cat" in url_:
url_ = url_.replace("i.pixiv.cat", "i.pximg.net") url_ = url_.replace("i.pixiv.cat", "i.pximg.net")
async with aiohttp.ClientSession(headers=headers) as session: async with aiohttp.ClientSession(headers=headers) as session:
@ -110,17 +110,20 @@ async def search_online_setu(url_: str) -> "MessageSegment, int":
try: try:
async with session.get(url_, proxy=get_local_proxy(), timeout=3) as res: async with session.get(url_, proxy=get_local_proxy(), timeout=3) as res:
if res.status == 200: if res.status == 200:
index = str(random.randint(1, 100000)) index = str(random.randint(1, 100000)) if id_ is None else id_
async with aiofiles.open( path_ = 'temp' if path_ is None else path_
IMAGE_PATH + "temp/" + index + "_temp_setu.jpg", "wb" if id_:
) as f: file = f'{index}.jpg'
else:
file = f'{index}_temp_setu.jpg'
async with aiofiles.open(f'{IMAGE_PATH}/{path_}/{file}', "wb") as f:
try: try:
await f.write(await res.read()) await f.write(await res.read())
except TimeoutError: except TimeoutError:
# return '\n这图没下载过来~（网太差？）', -1, False # return '\n这图没下载过来~（网太差？）', -1, False
continue continue
logger.info(f"下载 lolicon图片 {url_} 成功， id：{index}") logger.info(f"下载 lolicon图片 {url_} 成功， id：{index}")
return image(f"{index}_temp_setu.jpg", "temp"), index return image(file, path_), index
else: else:
logger.warning(f"访问 lolicon图片 {url_} 失败 status：{res.status}") logger.warning(f"访问 lolicon图片 {url_} 失败 status：{res.status}")
# return '\n这图好难下载啊！QAQ', -1, False # return '\n这图好难下载啊！QAQ', -1, False
@ -137,7 +140,7 @@ async def check_local_exists_or_download(setu_image: Setu) -> "MessageSegment, i
path_ = path path_ = path
if os.path.exists(f"{IMAGE_PATH}/{path_}/{setu_image.local_id}.jpg"): if os.path.exists(f"{IMAGE_PATH}/{path_}/{setu_image.local_id}.jpg"):
return image(f"{setu_image.local_id}.jpg", path_), 200 return image(f"{setu_image.local_id}.jpg", path_), 200
return await search_online_setu(setu_image.img_url) return await search_online_setu(setu_image.img_url, setu_image.local_id, path_)
# 添加涩图数据到数据库 # 添加涩图数据到数据库
@ -225,4 +228,4 @@ async def find_img_index(img_url, user_id):
f"author：{setu_img.author}\n" f"author：{setu_img.author}\n"
f"PID：{setu_img.pid}" f"PID：{setu_img.pid}"
) )
return "该图不在色图库中或色图库未更新！." return "该图不在色图库中或色图库未更新！"