From 6a7617600161f1b4da0a5bc7365d33ccd6ad65bf Mon Sep 17 00:00:00 2001 From: HibiKier <45528451+HibiKier@users.noreply.github.com> Date: Tue, 27 Jul 2021 17:28:22 +0800 Subject: [PATCH] Update __init__.py --- plugins/genshin/material_remind/__init__.py | 62 ++++++++++++++++++--- 1 file changed, 54 insertions(+), 8 deletions(-) diff --git a/plugins/genshin/material_remind/__init__.py b/plugins/genshin/material_remind/__init__.py index 2b10a5f93..5bc6ab065 100644 --- a/plugins/genshin/material_remind/__init__.py +++ b/plugins/genshin/material_remind/__init__.py @@ -1,6 +1,7 @@ from nonebot import on_command, Driver from nonebot.typing import T_State from nonebot.adapters.cqhttp import Bot, MessageEvent, Message +from utils.img_utils import CreateImg from utils.init_result import image from utils.browser import get_browser from configs.path_config import IMAGE_PATH @@ -8,7 +9,9 @@ from services.log import logger from utils.utils import scheduler from nonebot.permission import SUPERUSER +from typing import List import os +import asyncio import time @@ -24,7 +27,7 @@ async def _(bot: Bot, event: MessageEvent, state: T_State): if time.strftime("%w") == "0": await material.send("今天是周日，所有材料副本都开放了。") return - await material.send(Message(image('daily_material.png', 'genshin') + '\n※ 黄历数据来源于 genshin.pub')) + await material.send(Message(image('daily_material.png', 'genshin/material') + '\n※ 黄历数据来源于 genshin.pub')) logger.info( f"(USER {event.user_id}, GROUP {event.group_id if event.message_type != 'private' else 'private'})" f" 发送查看今日素材") @@ -41,9 +44,12 @@ async def _(bot: Bot, event: MessageEvent, state: T_State): @driver.on_startup async def update_image(): + page = None try: - if os.path.exists(f'{IMAGE_PATH}/genshin/daily_material.png'): - os.remove(f'{IMAGE_PATH}/genshin/daily_material.png') + if not os.path.exists(f'{IMAGE_PATH}/genshin/material'): + os.mkdir(f'{IMAGE_PATH}/genshin/material') + for file in os.listdir(f'{IMAGE_PATH}/genshin/material'): + os.remove(f'{IMAGE_PATH}/genshin/material/{file}') browser = await get_browser() if not browser: logger.warning('获取 browser 失败，请部署至 linux 环境....') @@ -52,24 +58,64 @@ async def update_image(): page = await browser.new_page() await page.goto(url, wait_until='networkidle', timeout=10000) await page.set_viewport_size({"width": 2560, "height": 1080}) - await page.evaluate(""" document.getElementsByClassName('GSTitleBar_gs_titlebar__2IJqy')[0].remove(); e = document.getElementsByClassName('GSContainer_gs_container__2FbUz')[0]; e.setAttribute("style", "height:880px"); """) - await page.click("button") - card = await page.query_selector(".GSContainer_content_box__1sIXz") - card = await card.bounding_box() - await page.screenshot(path=f'{IMAGE_PATH}/genshin/daily_material.png', clip=card, timeout=100000) + div = await page.query_selector(".GSContainer_content_box__1sIXz") + for i, card in enumerate(await page.query_selector_all(".GSTraitCotainer_trait_section__1f3bc")): + index = 0 + type_ = 'char' if not i else 'weapons' + for x in await card.query_selector_all("xpath=child::*"): + await x.screenshot(path=f'{IMAGE_PATH}/genshin/material/{type_}_{index}.png', timeout=100000) + # 下滑两次 + for _ in range(3): + await div.press("PageDown") + index += 1 + # 结束后上滑至顶 + for _ in range(index * 3): + await div.press("PageUp") + file_list = os.listdir(f'{IMAGE_PATH}/genshin/material') + char_imgs = [f'{IMAGE_PATH}/genshin/material/{x}' for x in file_list if x.startswith('char')] + weapons_imgs = [f'{IMAGE_PATH}/genshin/material/{x}' for x in file_list if x.startswith('weapons')] + char_imgs.sort() + weapons_imgs.sort() + height = await asyncio.get_event_loop().run_in_executor(None, get_background_height, weapons_imgs) + background_img = CreateImg(1200, height + 100, color='#f6f2ee') + current_width = 50 + for imgs in [char_imgs, weapons_imgs]: + current_height = 20 + for img in imgs: + x = CreateImg(0, 0, background=img) + background_img.paste(x, (current_width, current_height)) + current_height += x.size[1] + current_width += 600 + background_img.save(f'{IMAGE_PATH}/genshin/material/daily_material.png') await page.close() return True except Exception as e: logger.error(f'原神每日素材更新出错... {type(e)}: {e}') + if page: + await page.close() return False +# 获取背景高度以及修改最后一张图片的黑边 +def get_background_height(weapons_imgs: List[str]) -> int: + height = 0 + for weapons in weapons_imgs: + height += CreateImg(0, 0, background=weapons).size[1] + last_weapon = CreateImg(0, 0, background=weapons_imgs[-1]) + w, h = last_weapon.size + print(w, h) + last_weapon.crop((0, 0, w, h - 10)) + last_weapon.save(weapons_imgs[-1]) + + return height + + @scheduler.scheduled_job( 'cron', hour=4,