mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
423 lines
16 KiB
Python
423 lines
16 KiB
Python
|
import random
|
|||
|
|
import re
|
|||
|
|
from datetime import datetime
|
|||
|
|
from urllib.parse import unquote
|
|||
|
|
|
|||
|
|
import dateparser
|
|||
|
|
import ujson as json
|
|||
|
|
from bs4 import BeautifulSoup
|
|||
|
|
from lxml import etree
|
|||
|
|
from nonebot_plugin_saa import Image, MessageFactory, Text
|
|||
|
|
from PIL import ImageDraw
|
|||
|
|
from pydantic import ValidationError
|
|||
|
|
|
|||
|
|
from zhenxun.services.log import logger
|
|||
|
|
from zhenxun.utils.image_utils import BuildImage
|
|||
|
|
|
|||
|
|
from ..config import draw_config
|
|||
|
|
from ..util import cn2py, load_font, remove_prohibited_str
|
|||
|
|
from .base_handle import BaseData, BaseHandle, UpChar, UpEvent
|
|||
|
|
|
|||
|
|
|
|||
|
|
class PrettyData(BaseData):
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
|
|||
|
|
class PrettyChar(PrettyData):
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
|
|||
|
|
class PrettyCard(PrettyData):
|
|||
|
|
@property
|
|||
|
|
def star_str(self) -> str:
|
|||
|
|
return ["R", "SR", "SSR"][self.star - 1]
|
|||
|
|
|
|||
|
|
|
|||
|
|
class PrettyHandle(BaseHandle[PrettyData]):
|
|||
|
|
def __init__(self):
|
|||
|
|
super().__init__("pretty", "赛马娘")
|
|||
|
|
self.data_files.append("pretty_card.json")
|
|||
|
|
self.max_star = 3
|
|||
|
|
self.game_card_color = "#eff2f5"
|
|||
|
|
self.config = draw_config.pretty
|
|||
|
|
|
|||
|
|
self.ALL_CHAR: list[PrettyChar] = []
|
|||
|
|
self.ALL_CARD: list[PrettyCard] = []
|
|||
|
|
self.UP_CHAR: UpEvent | None = None
|
|||
|
|
self.UP_CARD: UpEvent | None = None
|
|||
|
|
|
|||
|
|
def get_card(self, pool_name: str, mode: int = 1) -> PrettyData:
|
|||
|
|
if mode == 1:
|
|||
|
|
star = self.get_star(
|
|||
|
|
[3, 2, 1],
|
|||
|
|
[
|
|||
|
|
self.config.PRETTY_THREE_P,
|
|||
|
|
self.config.PRETTY_TWO_P,
|
|||
|
|
self.config.PRETTY_ONE_P,
|
|||
|
|
],
|
|||
|
|
)
|
|||
|
|
else:
|
|||
|
|
star = self.get_star(
|
|||
|
|
[3, 2], [self.config.PRETTY_THREE_P, self.config.PRETTY_TWO_P]
|
|||
|
|
)
|
|||
|
|
up_pool = None
|
|||
|
|
if pool_name == "char":
|
|||
|
|
up_pool = self.UP_CHAR
|
|||
|
|
all_list = self.ALL_CHAR
|
|||
|
|
else:
|
|||
|
|
up_pool = self.UP_CARD
|
|||
|
|
all_list = self.ALL_CARD
|
|||
|
|
|
|||
|
|
all_char = [x for x in all_list if x.star == star and not x.limited]
|
|||
|
|
acquire_char = None
|
|||
|
|
# 有UP池子
|
|||
|
|
if up_pool and star in [x.star for x in up_pool.up_char]:
|
|||
|
|
up_list = [x.name for x in up_pool.up_char if x.star == star]
|
|||
|
|
# 抽到UP
|
|||
|
|
if random.random() < 1 / len(all_char) * (0.7 / 0.1385):
|
|||
|
|
up_name = random.choice(up_list)
|
|||
|
|
try:
|
|||
|
|
acquire_char = [x for x in all_list if x.name == up_name][0]
|
|||
|
|
except IndexError:
|
|||
|
|
pass
|
|||
|
|
if not acquire_char:
|
|||
|
|
acquire_char = random.choice(all_char)
|
|||
|
|
return acquire_char
|
|||
|
|
|
|||
|
|
def get_cards(self, count: int, pool_name: str) -> list[tuple[PrettyData, int]]:
|
|||
|
|
card_list = []
|
|||
|
|
card_count = 0 # 保底计算
|
|||
|
|
for i in range(count):
|
|||
|
|
card_count += 1
|
|||
|
|
# 十连保底
|
|||
|
|
if card_count == 10:
|
|||
|
|
card = self.get_card(pool_name, 2)
|
|||
|
|
card_count = 0
|
|||
|
|
else:
|
|||
|
|
card = self.get_card(pool_name, 1)
|
|||
|
|
if card.star > self.max_star - 2:
|
|||
|
|
card_count = 0
|
|||
|
|
card_list.append((card, i + 1))
|
|||
|
|
return card_list
|
|||
|
|
|
|||
|
|
def format_pool_info(self, pool_name: str) -> str:
|
|||
|
|
info = ""
|
|||
|
|
up_event = self.UP_CHAR if pool_name == "char" else self.UP_CARD
|
|||
|
|
if up_event:
|
|||
|
|
star3_list = [x.name for x in up_event.up_char if x.star == 3]
|
|||
|
|
star2_list = [x.name for x in up_event.up_char if x.star == 2]
|
|||
|
|
star1_list = [x.name for x in up_event.up_char if x.star == 1]
|
|||
|
|
if star3_list:
|
|||
|
|
if pool_name == "char":
|
|||
|
|
info += f'三星UP：{" ".join(star3_list)}\n'
|
|||
|
|
else:
|
|||
|
|
info += f'SSR UP：{" ".join(star3_list)}\n'
|
|||
|
|
if star2_list:
|
|||
|
|
if pool_name == "char":
|
|||
|
|
info += f'二星UP：{" ".join(star2_list)}\n'
|
|||
|
|
else:
|
|||
|
|
info += f'SR UP：{" ".join(star2_list)}\n'
|
|||
|
|
if star1_list:
|
|||
|
|
if pool_name == "char":
|
|||
|
|
info += f'一星UP：{" ".join(star1_list)}\n'
|
|||
|
|
else:
|
|||
|
|
info += f'R UP：{" ".join(star1_list)}\n'
|
|||
|
|
info = f"当前up池：{up_event.title}\n{info}"
|
|||
|
|
return info.strip()
|
|||
|
|
|
|||
|
|
async def draw(self, count: int, pool_name: str, **kwargs) -> MessageFactory:
|
|||
|
|
pool_name = "char" if not pool_name else pool_name
|
|||
|
|
index2card = self.get_cards(count, pool_name)
|
|||
|
|
cards = [card[0] for card in index2card]
|
|||
|
|
up_event = self.UP_CHAR if pool_name == "char" else self.UP_CARD
|
|||
|
|
up_list = [x.name for x in up_event.up_char] if up_event else []
|
|||
|
|
result = self.format_result(index2card, up_list=up_list)
|
|||
|
|
pool_info = self.format_pool_info(pool_name)
|
|||
|
|
img = await self.generate_img(cards)
|
|||
|
|
return MessageFactory([Text(pool_info), Image(img.pic2bytes()), Text(result)])
|
|||
|
|
|
|||
|
|
async def generate_card_img(self, card: PrettyData) -> BuildImage:
|
|||
|
|
if isinstance(card, PrettyChar):
|
|||
|
|
star_h = 30
|
|||
|
|
img_w = 200
|
|||
|
|
img_h = 219
|
|||
|
|
font_h = 50
|
|||
|
|
bg = BuildImage(img_w, img_h + font_h, color="#EFF2F5")
|
|||
|
|
star_path = str(self.img_path / "star.png")
|
|||
|
|
star = BuildImage(star_h, star_h, background=star_path)
|
|||
|
|
img_path = str(self.img_path / f"{cn2py(card.name)}.png")
|
|||
|
|
img = BuildImage(img_w, img_h, background=img_path)
|
|||
|
|
star_w = star_h * card.star
|
|||
|
|
for i in range(card.star):
|
|||
|
|
await bg.paste(star, (int((img_w - star_w) / 2) + star_h * i, 0))
|
|||
|
|
await bg.paste(img, (0, 0))
|
|||
|
|
# 加名字
|
|||
|
|
text = card.name[:5] + "..." if len(card.name) > 6 else card.name
|
|||
|
|
font = load_font(fontsize=30)
|
|||
|
|
text_w, _ = font.getsize(text)
|
|||
|
|
draw = ImageDraw.Draw(bg.markImg)
|
|||
|
|
draw.text(
|
|||
|
|
((img_w - text_w) / 2, img_h),
|
|||
|
|
text,
|
|||
|
|
font=font,
|
|||
|
|
fill="gray",
|
|||
|
|
)
|
|||
|
|
return bg
|
|||
|
|
else:
|
|||
|
|
sep_w = 10
|
|||
|
|
img_w = 200
|
|||
|
|
img_h = 267
|
|||
|
|
font_h = 75
|
|||
|
|
bg = BuildImage(img_w + sep_w * 2, img_h + font_h, color="#EFF2F5")
|
|||
|
|
label_path = str(self.img_path / f"{card.star}_label.png")
|
|||
|
|
label = BuildImage(40, 40, background=label_path)
|
|||
|
|
img_path = str(self.img_path / f"{cn2py(card.name)}.png")
|
|||
|
|
img = BuildImage(img_w, img_h, background=img_path)
|
|||
|
|
await bg.paste(img, (sep_w, 0))
|
|||
|
|
await bg.paste(label, (30, 3))
|
|||
|
|
# 加名字
|
|||
|
|
text = ""
|
|||
|
|
texts = []
|
|||
|
|
font = load_font(fontsize=25)
|
|||
|
|
for t in card.name:
|
|||
|
|
if BuildImage.get_text_size((text + t), font)[0] > 190:
|
|||
|
|
texts.append(text)
|
|||
|
|
text = ""
|
|||
|
|
if len(texts) >= 2:
|
|||
|
|
texts[-1] += "..."
|
|||
|
|
break
|
|||
|
|
else:
|
|||
|
|
text += t
|
|||
|
|
if text:
|
|||
|
|
texts.append(text)
|
|||
|
|
text = "\n".join(texts)
|
|||
|
|
text_w, _ = font.getsize_multiline(text)
|
|||
|
|
draw = ImageDraw.Draw(bg.markImg)
|
|||
|
|
draw.text(
|
|||
|
|
((img_w - text_w) / 2, img_h),
|
|||
|
|
text,
|
|||
|
|
font=font,
|
|||
|
|
align="center",
|
|||
|
|
fill="gray",
|
|||
|
|
)
|
|||
|
|
return bg
|
|||
|
|
|
|||
|
|
def _init_data(self):
|
|||
|
|
self.ALL_CHAR = [
|
|||
|
|
PrettyChar(
|
|||
|
|
name=value["名称"],
|
|||
|
|
star=int(value["初始星级"]),
|
|||
|
|
limited=False,
|
|||
|
|
)
|
|||
|
|
for value in self.load_data().values()
|
|||
|
|
]
|
|||
|
|
self.ALL_CARD = [
|
|||
|
|
PrettyCard(
|
|||
|
|
name=value["中文名"],
|
|||
|
|
star=["R", "SR", "SSR"].index(value["稀有度"]) + 1,
|
|||
|
|
limited=True if "卡池" not in value["获取方式"] else False,
|
|||
|
|
)
|
|||
|
|
for value in self.load_data("pretty_card.json").values()
|
|||
|
|
]
|
|||
|
|
self.load_up_char()
|
|||
|
|
|
|||
|
|
def load_up_char(self):
|
|||
|
|
try:
|
|||
|
|
data = self.load_data(f"draw_card_up/{self.game_name}_up_char.json")
|
|||
|
|
self.UP_CHAR = UpEvent.parse_obj(data.get("char", {}))
|
|||
|
|
self.UP_CARD = UpEvent.parse_obj(data.get("card", {}))
|
|||
|
|
except ValidationError:
|
|||
|
|
logger.warning(f"{self.game_name}_up_char 解析出错")
|
|||
|
|
|
|||
|
|
def dump_up_char(self):
|
|||
|
|
if self.UP_CHAR and self.UP_CARD:
|
|||
|
|
data = {
|
|||
|
|
"char": json.loads(self.UP_CHAR.json()),
|
|||
|
|
"card": json.loads(self.UP_CARD.json()),
|
|||
|
|
}
|
|||
|
|
self.dump_data(data, f"draw_card_up/{self.game_name}_up_char.json")
|
|||
|
|
|
|||
|
|
async def _update_info(self):
|
|||
|
|
# pretty.json
|
|||
|
|
pretty_info = {}
|
|||
|
|
url = "https://wiki.biligame.com/umamusume/赛马娘图鉴"
|
|||
|
|
result = await self.get_url(url)
|
|||
|
|
if not result:
|
|||
|
|
logger.warning(f"更新 {self.game_name_cn} 出错")
|
|||
|
|
else:
|
|||
|
|
dom = etree.HTML(result, etree.HTMLParser())
|
|||
|
|
char_list = dom.xpath("//table[@id='CardSelectTr']/tbody/tr")
|
|||
|
|
for char in char_list:
|
|||
|
|
try:
|
|||
|
|
name = char.xpath("./td[1]/a/@title")[0]
|
|||
|
|
avatar = char.xpath("./td[1]/a/img/@srcset")[0]
|
|||
|
|
star = len(char.xpath("./td[3]/img"))
|
|||
|
|
except IndexError:
|
|||
|
|
continue
|
|||
|
|
member_dict = {
|
|||
|
|
"头像": unquote(str(avatar).split(" ")[-2]),
|
|||
|
|
"名称": remove_prohibited_str(name),
|
|||
|
|
"初始星级": star,
|
|||
|
|
}
|
|||
|
|
pretty_info[member_dict["名称"]] = member_dict
|
|||
|
|
self.dump_data(pretty_info)
|
|||
|
|
logger.info(f"{self.game_name_cn} 更新成功")
|
|||
|
|
# pretty_card.json
|
|||
|
|
pretty_card_info = {}
|
|||
|
|
url = "https://wiki.biligame.com/umamusume/支援卡图鉴"
|
|||
|
|
result = await self.get_url(url)
|
|||
|
|
if not result:
|
|||
|
|
logger.warning(f"更新 {self.game_name_cn} 卡牌出错")
|
|||
|
|
else:
|
|||
|
|
dom = etree.HTML(result, etree.HTMLParser())
|
|||
|
|
char_list = dom.xpath("//table[@id='CardSelectTr']/tbody/tr")
|
|||
|
|
for char in char_list:
|
|||
|
|
try:
|
|||
|
|
name = char.xpath("./td[1]/div/a/@title")[0]
|
|||
|
|
name_cn = char.xpath("./td[3]/a/text()")[0]
|
|||
|
|
avatar = char.xpath("./td[1]/div/a/img/@srcset")[0]
|
|||
|
|
star = str(char.xpath("./td[5]/text()")[0]).strip()
|
|||
|
|
sources = str(char.xpath("./td[7]/text()")[0]).strip()
|
|||
|
|
except IndexError:
|
|||
|
|
continue
|
|||
|
|
member_dict = {
|
|||
|
|
"头像": unquote(str(avatar).split(" ")[-2]),
|
|||
|
|
"名称": remove_prohibited_str(name),
|
|||
|
|
"中文名": remove_prohibited_str(name_cn),
|
|||
|
|
"稀有度": star,
|
|||
|
|
"获取方式": [sources] if sources else [],
|
|||
|
|
}
|
|||
|
|
pretty_card_info[member_dict["中文名"]] = member_dict
|
|||
|
|
self.dump_data(pretty_card_info, "pretty_card.json")
|
|||
|
|
logger.info(f"{self.game_name_cn} 卡牌更新成功")
|
|||
|
|
# 下载头像
|
|||
|
|
for value in pretty_info.values():
|
|||
|
|
await self.download_img(value["头像"], value["名称"])
|
|||
|
|
for value in pretty_card_info.values():
|
|||
|
|
await self.download_img(value["头像"], value["中文名"])
|
|||
|
|
# 下载星星
|
|||
|
|
PRETTY_URL = "https://patchwiki.biligame.com/images/umamusume"
|
|||
|
|
await self.download_img(
|
|||
|
|
PRETTY_URL + "/1/13/e1hwjz4vmhtvk8wlyb7c0x3ld1s2ata.png", "star"
|
|||
|
|
)
|
|||
|
|
# 下载稀有度标志
|
|||
|
|
idx = 1
|
|||
|
|
for url in [
|
|||
|
|
"/f/f7/afqs7h4snmvovsrlifq5ib8vlpu2wvk.png",
|
|||
|
|
"/3/3b/d1jmpwrsk4irkes1gdvoos4ic6rmuht.png",
|
|||
|
|
"/0/06/q23szwkbtd7pfkqrk3wcjlxxt9z595o.png",
|
|||
|
|
]:
|
|||
|
|
await self.download_img(PRETTY_URL + url, f"{idx}_label")
|
|||
|
|
idx += 1
|
|||
|
|
await self.update_up_char()
|
|||
|
|
|
|||
|
|
async def update_up_char(self):
|
|||
|
|
announcement_url = "https://wiki.biligame.com/umamusume/公告"
|
|||
|
|
result = await self.get_url(announcement_url)
|
|||
|
|
if not result:
|
|||
|
|
logger.warning(f"{self.game_name_cn}获取公告出错")
|
|||
|
|
return
|
|||
|
|
dom = etree.HTML(result, etree.HTMLParser())
|
|||
|
|
announcements = dom.xpath("//div[@id='mw-content-text']/div/div/span/a")
|
|||
|
|
title = ""
|
|||
|
|
url = ""
|
|||
|
|
for announcement in announcements:
|
|||
|
|
try:
|
|||
|
|
title = announcement.xpath("./@title")[0]
|
|||
|
|
url = "https://wiki.biligame.com/" + announcement.xpath("./@href")[0]
|
|||
|
|
if re.match(r".*?\d{8}$", title) or re.match(
|
|||
|
|
r"^\d{1,2}月\d{1,2}日.*?", title
|
|||
|
|
):
|
|||
|
|
break
|
|||
|
|
except IndexError:
|
|||
|
|
continue
|
|||
|
|
if not title:
|
|||
|
|
logger.warning(f"{self.game_name_cn}未找到新UP公告")
|
|||
|
|
return
|
|||
|
|
result = await self.get_url(url)
|
|||
|
|
if not result:
|
|||
|
|
logger.warning(f"{self.game_name_cn}获取UP公告出错")
|
|||
|
|
return
|
|||
|
|
try:
|
|||
|
|
start_time = None
|
|||
|
|
end_time = None
|
|||
|
|
char_img = ""
|
|||
|
|
card_img = ""
|
|||
|
|
up_chars = []
|
|||
|
|
up_cards = []
|
|||
|
|
soup = BeautifulSoup(result, "lxml")
|
|||
|
|
heads = soup.find_all("span", {"class": "mw-headline"})
|
|||
|
|
for head in heads:
|
|||
|
|
if "时间" in head.text:
|
|||
|
|
time = head.find_next("p").text.split("\n")[0]
|
|||
|
|
if "～" in time:
|
|||
|
|
start, end = time.split("～")
|
|||
|
|
start_time = dateparser.parse(start)
|
|||
|
|
end_time = dateparser.parse(end)
|
|||
|
|
elif "赛马娘" in head.text:
|
|||
|
|
char_img = head.find_next("a", {"class": "image"}).find("img")[
|
|||
|
|
"src"
|
|||
|
|
]
|
|||
|
|
lines = str(head.find_next("p").text).split("\n")
|
|||
|
|
chars = [
|
|||
|
|
line
|
|||
|
|
for line in lines
|
|||
|
|
if "★" in line and "（" in line and "）" in line
|
|||
|
|
]
|
|||
|
|
for char in chars:
|
|||
|
|
star = char.count("★")
|
|||
|
|
name = re.split(r"[（）]", char)[-2].strip()
|
|||
|
|
up_chars.append(
|
|||
|
|
UpChar(name=name, star=star, limited=False, zoom=70)
|
|||
|
|
)
|
|||
|
|
elif "支援卡" in head.text:
|
|||
|
|
card_img = head.find_next("a", {"class": "image"}).find("img")[
|
|||
|
|
"src"
|
|||
|
|
]
|
|||
|
|
lines = str(head.find_next("p").text).split("\n")
|
|||
|
|
cards = [
|
|||
|
|
line
|
|||
|
|
for line in lines
|
|||
|
|
if "R" in line and "（" in line and "）" in line
|
|||
|
|
]
|
|||
|
|
for card in cards:
|
|||
|
|
star = 3 if "SSR" in card else 2 if "SR" in card else 1
|
|||
|
|
name = re.split(r"[（）]", card)[-2].strip()
|
|||
|
|
up_cards.append(
|
|||
|
|
UpChar(name=name, star=star, limited=False, zoom=70)
|
|||
|
|
)
|
|||
|
|
if start_time and end_time:
|
|||
|
|
if start_time <= datetime.now() <= end_time:
|
|||
|
|
self.UP_CHAR = UpEvent(
|
|||
|
|
title=title,
|
|||
|
|
pool_img=char_img,
|
|||
|
|
start_time=start_time,
|
|||
|
|
end_time=end_time,
|
|||
|
|
up_char=up_chars,
|
|||
|
|
)
|
|||
|
|
self.UP_CARD = UpEvent(
|
|||
|
|
title=title,
|
|||
|
|
pool_img=card_img,
|
|||
|
|
start_time=start_time,
|
|||
|
|
end_time=end_time,
|
|||
|
|
up_char=up_cards,
|
|||
|
|
)
|
|||
|
|
self.dump_up_char()
|
|||
|
|
logger.info(
|
|||
|
|
f"成功获取{self.game_name_cn}当前up信息...当前up池: {title}"
|
|||
|
|
)
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.warning(f"{self.game_name_cn}UP更新出错", e=e)
|
|||
|
|
|
|||
|
|
async def _reload_pool(self) -> MessageFactory | None:
|
|||
|
|
await self.update_up_char()
|
|||
|
|
self.load_up_char()
|
|||
|
|
if self.UP_CHAR and self.UP_CARD:
|
|||
|
|
return MessageFactory(
|
|||
|
|
[
|
|||
|
|
Text(f"重载成功！\n当前UP池子：{self.UP_CHAR.title}"),
|
|||
|
|
Image(self.UP_CHAR.pool_img),
|
|||
|
|
Image(self.UP_CARD.pool_img),
|
|||
|
|
]
|
|||
|
|
)
|