mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
118 lines
3.9 KiB
Python
Executable File
118 lines
3.9 KiB
Python
Executable File
import asyncio
|
||
import os
|
||
import time
|
||
from datetime import datetime, timedelta
|
||
from typing import List
|
||
|
||
import nonebot
|
||
from nonebot import Driver, on_command
|
||
from nonebot.adapters.onebot.v11 import GroupMessageEvent, Message, MessageEvent
|
||
from nonebot.permission import SUPERUSER
|
||
|
||
from configs.path_config import IMAGE_PATH
|
||
from services.log import logger
|
||
from utils.browser import get_browser
|
||
from utils.image_utils import BuildImage
|
||
from utils.message_builder import image
|
||
|
||
__zx_plugin_name__ = "今日素材"
|
||
__plugin_usage__ = """
|
||
usage：
|
||
看看原神今天要刷什么
|
||
指令：
|
||
今日素材/今天素材
|
||
""".strip()
|
||
__plugin_superuser_usage__ = """
|
||
usage：
|
||
更新原神今日素材
|
||
指令：
|
||
更新原神今日素材
|
||
""".strip()
|
||
__plugin_des__ = "看看原神今天要刷什么"
|
||
__plugin_cmd__ = ["今日素材/今天素材", "更新原神今日素材 [_superuser]"]
|
||
__plugin_type__ = ("原神相关",)
|
||
__plugin_version__ = 0.1
|
||
__plugin_author__ = "HibiKier"
|
||
__plugin_settings__ = {
|
||
"level": 5,
|
||
"default_status": True,
|
||
"limit_superuser": False,
|
||
"cmd": ["今日素材", "今天素材"],
|
||
}
|
||
|
||
driver: Driver = nonebot.get_driver()
|
||
|
||
material = on_command("今日素材", aliases={"今日材料", "今天素材", "今天材料"}, priority=5, block=True)
|
||
|
||
super_cmd = on_command("更新原神今日素材", permission=SUPERUSER, priority=1, block=True)
|
||
|
||
|
||
@material.handle()
|
||
async def _(event: MessageEvent):
|
||
if time.strftime("%w") == "0":
|
||
await material.send("今天是周日，所有材料副本都开放了。")
|
||
return
|
||
file_name = str((datetime.now() - timedelta(hours=4)).date())
|
||
if not (IMAGE_PATH / "genshin" / "material" / f"{file_name}.png").exists():
|
||
await update_image()
|
||
await material.send(
|
||
Message(
|
||
image(f"{file_name}.png", "genshin/material")
|
||
+ "\n※ 每日素材数据来源于米游社"
|
||
)
|
||
)
|
||
logger.info(
|
||
f"(USER {event.user_id}, GROUP {event.group_id if isinstance(event, GroupMessageEvent) else 'private'})"
|
||
f" 发送查看今日素材"
|
||
)
|
||
|
||
|
||
@super_cmd.handle()
|
||
async def _():
|
||
if await update_image():
|
||
await super_cmd.send("更新成功...")
|
||
logger.info(f"更新每日天赋素材成功...")
|
||
else:
|
||
await super_cmd.send(f"更新失败...")
|
||
|
||
|
||
async def update_image():
|
||
page = None
|
||
try:
|
||
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 环境....")
|
||
return False
|
||
# url = "https://genshin.pub/daily"
|
||
url = "https://bbs.mihoyo.com/ys/obc/channel/map/193"
|
||
page = await browser.new_page(viewport={'width': 860, 'height': 3000})
|
||
await page.goto(url)
|
||
await page.wait_for_timeout(3000)
|
||
file_name = str((datetime.now() - timedelta(hours=4)).date())
|
||
# background_img.save(f"{IMAGE_PATH}/genshin/material/{file_name}.png")
|
||
await page.locator('//*[@id="__layout"]/div/div[2]/div[2]/div/div[1]/div[2]/div/div').screenshot(path=f"{IMAGE_PATH}/genshin/material/{file_name}.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_img: List[str]) -> int:
|
||
height = 0
|
||
for weapons in weapons_img:
|
||
height += BuildImage(0, 0, background=weapons).size[1]
|
||
last_weapon = BuildImage(0, 0, background=weapons_img[-1])
|
||
w, h = last_weapon.size
|
||
last_weapon.crop((0, 0, w, h - 10))
|
||
last_weapon.save(weapons_img[-1])
|
||
|
||
return height
|