mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
299 lines
11 KiB
Python
299 lines
11 KiB
Python
from typing import Optional, Union
|
||
from nonebot.adapters.onebot.v11 import MessageSegment
|
||
from configs.config import Config
|
||
from asyncio.exceptions import TimeoutError
|
||
from services.log import logger
|
||
from configs.path_config import IMAGE_PATH
|
||
from utils.image_utils import BuildImage
|
||
from utils.http_utils import AsyncHttpx
|
||
from utils.utils import get_user_avatar
|
||
from utils.message_builder import image
|
||
from .._utils import get_ds
|
||
from .._models import Genshin
|
||
from io import BytesIO
|
||
from nonebot import Driver
|
||
import asyncio
|
||
import nonebot
|
||
|
||
|
||
driver: Driver = nonebot.get_driver()
|
||
|
||
|
||
memo_path = IMAGE_PATH / "genshin" / "genshin_memo"
|
||
memo_path.mkdir(exist_ok=True, parents=True)
|
||
|
||
|
||
@driver.on_startup
|
||
async def _():
|
||
for name, url in zip(
|
||
[
|
||
"resin.png", "task.png", "resin_discount.png", "chengehu.png",
|
||
"zhibian.png"
|
||
],
|
||
[
|
||
"https://upload-bbs.mihoyo.com/upload/2021/09/29/8819732/54266243c7d15ba31690c8f5d63cc3c6_71491376413333325"
|
||
"20.png?x-oss-process=image//resize,s_600/quality,q_80/auto-orient,0/interlace,1/format,png",
|
||
"https://patchwiki.biligame.com/images/ys/thumb/c/cc/6k6kuj1kte6m1n7hexqfrn92z6h4yhh.png/60px-委托任务logo.png",
|
||
"https://patchwiki.biligame.com/images/ys/d/d9/t1hv6wpucbwucgkhjntmzroh90nmcdv.png",
|
||
"https://s3.bmp.ovh/imgs/2022/07/24/28d9338c7da4bcb2.png",
|
||
"https://genshin.honeyhunterworld.com/img/gadget/i_3016.png",
|
||
],
|
||
):
|
||
file = memo_path / name
|
||
if not file.exists():
|
||
await AsyncHttpx.download_file(url, file)
|
||
logger.info(f"已下载原神便签资源 -> {file}...")
|
||
|
||
|
||
async def get_user_memo(user_id: int, uid: int,
|
||
uname: str) -> Optional[Union[str, MessageSegment]]:
|
||
uid = str(uid)
|
||
if uid[0] in ["1", "2"]:
|
||
server_id = "cn_gf01"
|
||
elif uid[0] == "5":
|
||
server_id = "cn_qd01"
|
||
else:
|
||
return None
|
||
return await parse_data_and_draw(user_id, uid, server_id, uname)
|
||
|
||
|
||
async def get_memo(uid: str, server_id: str) -> "Union[str, dict], int":
|
||
try:
|
||
req = await AsyncHttpx.get(
|
||
url=
|
||
f"https://api-takumi-record.mihoyo.com/game_record/app/genshin/api/dailyNote?server={server_id}&role_id={uid}",
|
||
headers={
|
||
"DS": get_ds(f"role_id={uid}&server={server_id}"),
|
||
"x-rpc-app_version": Config.get_config("genshin",
|
||
"mhyVersion"),
|
||
"User-Agent":
|
||
"Mozilla/5.0 (iPhone; CPU iPhone OS 13_2_3 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) miHoYoBBS/2.11.1",
|
||
"x-rpc-client_type": Config.get_config("genshin",
|
||
"client_type"),
|
||
"Referer": "https://webstatic.mihoyo.com/",
|
||
"Cookie": await Genshin.get_user_cookie(int(uid))
|
||
},
|
||
)
|
||
data = req.json()
|
||
if data["message"] == "OK":
|
||
return data["data"], 200
|
||
return data["message"], 999
|
||
except TimeoutError:
|
||
return "访问超时，请稍后再试", 997
|
||
except Exception as e:
|
||
logger.error(f"便签查询获取失败未知错误 {e}：{e}")
|
||
return "发生了一些错误，请稍后再试", 998
|
||
|
||
|
||
def create_border(image_name: str, content: str, notice_text: str,
|
||
value: str) -> BuildImage:
|
||
border = BuildImage(500,
|
||
75,
|
||
color="#E0D9D1",
|
||
font="HYWenHei-85W.ttf",
|
||
font_size=20)
|
||
text_bk = BuildImage(350,
|
||
75,
|
||
color="#F5F1EB",
|
||
font_size=23,
|
||
font="HYWenHei-85W.ttf")
|
||
_x = 70 if image_name == "resin.png" else 50
|
||
_px = 10 if image_name == "resin.png" else 20
|
||
text_bk.paste(
|
||
BuildImage(_x, _x, background=memo_path / image_name),
|
||
(_px, 0),
|
||
True,
|
||
center_type="by_height",
|
||
)
|
||
text_bk.text((87, 15), content)
|
||
text_bk.paste(
|
||
BuildImage(
|
||
0,
|
||
0,
|
||
plain_text=notice_text,
|
||
font_color=(203, 189, 175),
|
||
font="HYWenHei-85W.ttf",
|
||
font_size=17,
|
||
),
|
||
(87, 45),
|
||
True,
|
||
)
|
||
font_width, _ = border.getsize(value)
|
||
border.text((350 + 76 - int(font_width / 2), 0),
|
||
value,
|
||
center_type="by_height")
|
||
border.paste(text_bk, (2, 0), center_type="by_height")
|
||
return border
|
||
|
||
|
||
async def parse_data_and_draw(user_id: int, uid: str, server_id: str,
|
||
uname: str) -> Union[str, MessageSegment]:
|
||
data, code = await get_memo(uid, server_id)
|
||
if code != 200:
|
||
return data
|
||
user_avatar = BytesIO(await get_user_avatar(user_id))
|
||
for x in data["expeditions"]:
|
||
file_name = x["avatar_side_icon"].split("_")[-1]
|
||
role_avatar = memo_path / "role_avatar" / file_name
|
||
if not role_avatar.exists():
|
||
await AsyncHttpx.download_file(x["avatar_side_icon"], role_avatar)
|
||
return await asyncio.get_event_loop().run_in_executor(
|
||
None, _parse_data_and_draw, data, user_avatar, uid, uname)
|
||
|
||
|
||
def _parse_data_and_draw(data: dict, user_avatar: BytesIO, uid: int,
|
||
uname: str) -> Union[str, MessageSegment]:
|
||
current_resin = data["current_resin"] # 当前树脂
|
||
max_resin = data["max_resin"] # 最大树脂
|
||
resin_recovery_time = data["resin_recovery_time"] # 树脂全部回复时间
|
||
finished_task_num = data["finished_task_num"] # 完成的每日任务
|
||
total_task_num = data["total_task_num"] # 每日任务总数
|
||
remain_resin_discount_num = data["remain_resin_discount_num"] # 值得铭记的强敌总数
|
||
resin_discount_num_limit = data["resin_discount_num_limit"] # 剩余值得铭记的强敌
|
||
current_expedition_num = data["current_expedition_num"] # 当前挖矿人数
|
||
max_expedition_num = data["max_expedition_num"] # 每日挖矿最大人数
|
||
expeditions = data["expeditions"] # 挖矿详情
|
||
current_coin = data["current_home_coin"] # 当前宝钱
|
||
max_coin = data["max_home_coin"] # 最大宝钱
|
||
coin_recovery_time = data["home_coin_recovery_time"] # 宝钱全部回复时间
|
||
transformer_available = data["transformer"]["obtained"] # 参量质变仪可获取
|
||
transformer_state = data["transformer"]["recovery_time"][
|
||
"reached"] # 参量质变仪状态
|
||
transformer_recovery_time = data["transformer"]["recovery_time"][
|
||
"Day"] # 参量质变仪回复时间
|
||
transformer_recovery_hour = data["transformer"]["recovery_time"][
|
||
"Hour"] # 参量质变仪回复时间
|
||
coin_minute, coin_second = divmod(int(coin_recovery_time), 60)
|
||
coin_hour, coin_minute = divmod(coin_minute, 60)
|
||
#print(data)
|
||
minute, second = divmod(int(resin_recovery_time), 60)
|
||
hour, minute = divmod(minute, 60)
|
||
|
||
A = BuildImage(1030,
|
||
570,
|
||
color="#f1e9e1",
|
||
font_size=15,
|
||
font="HYWenHei-85W.ttf")
|
||
A.text((10, 15), "原神便笺 | Create By ZhenXun", (198, 186, 177))
|
||
ava = BuildImage(100, 100, background=user_avatar)
|
||
ava.circle()
|
||
A.paste(ava, (40, 40), True)
|
||
A.paste(
|
||
BuildImage(0,
|
||
0,
|
||
plain_text=uname,
|
||
font_size=20,
|
||
font="HYWenHei-85W.ttf"),
|
||
(160, 62),
|
||
True,
|
||
)
|
||
A.paste(
|
||
BuildImage(
|
||
0,
|
||
0,
|
||
plain_text=f"UID：{uid}",
|
||
font_size=15,
|
||
font="HYWenHei-85W.ttf",
|
||
font_color=(21, 167, 89),
|
||
),
|
||
(160, 92),
|
||
True,
|
||
)
|
||
border = create_border(
|
||
"resin.png",
|
||
"原粹树脂",
|
||
"将在{:0>2d}:{:0>2d}:{:0>2d}秒后全部恢复".format(hour, minute, second),
|
||
f"{current_resin}/{max_resin}",
|
||
)
|
||
|
||
A.paste(border, (10, 155))
|
||
border = create_border(
|
||
"task.png",
|
||
"每日委托",
|
||
"今日委托已全部完成" if finished_task_num == total_task_num else "今日委托完成数量不足",
|
||
f"{finished_task_num}/{total_task_num}",
|
||
)
|
||
A.paste(border, (10, 235))
|
||
border = create_border(
|
||
"resin_discount.png",
|
||
"值得铭记的强敌",
|
||
"本周剩余消耗减半次数",
|
||
f"{remain_resin_discount_num}/{resin_discount_num_limit}",
|
||
)
|
||
A.paste(border, (10, 315))
|
||
border = create_border(
|
||
"chengehu.png",
|
||
"洞天财翁-洞天宝钱",
|
||
"洞天财翁已达到存储上限"
|
||
if current_coin == max_coin else f"{coin_hour}小时{coin_minute}分钟后存满",
|
||
f"{current_coin}/{max_coin}",
|
||
)
|
||
A.paste(border, (10, 395))
|
||
border = create_border(
|
||
"zhibian.png",
|
||
"参量质变仪",
|
||
"不存在" if not transformer_available else
|
||
"已准备完成 " if transformer_state else f"{transformer_recovery_hour}小时后可使用" if not transformer_recovery_time else f"{transformer_recovery_time}天后可使用",
|
||
"不存在" if not transformer_available else
|
||
"可使用" if transformer_state else "冷却中",
|
||
)
|
||
A.paste(border, (10, 475))
|
||
|
||
expeditions_border = BuildImage(470,
|
||
510,
|
||
color="#E0D9D1",
|
||
font="HYWenHei-85W.ttf",
|
||
font_size=20)
|
||
expeditions_text = BuildImage(466,
|
||
506,
|
||
color="#F5F1EB",
|
||
font_size=23,
|
||
font="HYWenHei-85W.ttf")
|
||
expeditions_text.text(
|
||
(5, 5), f"探索派遣限制{current_expedition_num}/{max_expedition_num}",
|
||
(100, 100, 98))
|
||
h = 45
|
||
for x in expeditions:
|
||
_bk = BuildImage(400,
|
||
82,
|
||
color="#ECE3D8",
|
||
font="HYWenHei-85W.ttf",
|
||
font_size=21)
|
||
file_name = x["avatar_side_icon"].split("_")[-1]
|
||
role_avatar = memo_path / "role_avatar" / file_name
|
||
_ava_img = BuildImage(75, 75, background=role_avatar)
|
||
# _ava_img.circle()
|
||
if x["status"] == "Finished":
|
||
msg = "探索完成"
|
||
font_color = (146, 188, 63)
|
||
_circle_color = (146, 188, 63)
|
||
else:
|
||
minute, second = divmod(int(x["remained_time"]), 60)
|
||
hour, minute = divmod(minute, 60)
|
||
font_color = (193, 180, 167)
|
||
msg = "还剩{:0>2d}小时{:0>2d}分钟{:0>2d}秒".format(hour, minute, second)
|
||
_circle_color = "#DE9C58"
|
||
|
||
_circle_bk = BuildImage(60, 60)
|
||
_circle_bk.circle()
|
||
a_circle = BuildImage(55, 55, color=_circle_color)
|
||
a_circle.circle()
|
||
b_circle = BuildImage(47, 47)
|
||
b_circle.circle()
|
||
a_circle.paste(b_circle, (4, 4), True)
|
||
_circle_bk.paste(a_circle, (4, 4), True)
|
||
|
||
_bk.paste(_circle_bk, (25, 0), True, "by_height")
|
||
_bk.paste(_ava_img, (19, -13), True)
|
||
_bk.text((100, 0), msg, font_color, "by_height")
|
||
_bk.circle_corner(20)
|
||
|
||
expeditions_text.paste(_bk, (25, h), True)
|
||
h += 75 + 16
|
||
|
||
expeditions_border.paste(expeditions_text, center_type="center")
|
||
|
||
A.paste(expeditions_border, (550, 45))
|
||
|
||
return image(b64=A.pic2bs4())
|