import random from urllib.parse import unquote import dateparser import ujson as json 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 from .base_handle import UpChar as _UpChar from .base_handle import UpEvent as _UpEvent class AzurChar(BaseData): type_: str # 舰娘类型 @property def star_str(self) -> str: return ["白", "蓝", "紫", "金"][self.star - 1] class UpChar(_UpChar): type_: str # 舰娘类型 class UpEvent(_UpEvent): up_char: list[UpChar] # up对象 class AzurHandle(BaseHandle[AzurChar]): def __init__(self): super().__init__("azur", "碧蓝航线") self.max_star = 4 self.config = draw_config.azur self.ALL_CHAR: list[AzurChar] = [] self.UP_EVENT: UpEvent | None = None def get_card(self, pool_name: str, **kwargs) -> AzurChar: if pool_name == "轻型": type_ = ["驱逐", "轻巡", "维修"] elif pool_name == "重型": type_ = ["重巡", "战列", "战巡", "重炮"] else: type_ = ["维修", "潜艇", "重巡", "轻航", "航母"] up_pool_flag = pool_name == "活动" # Up up_ship = ( [x for x in self.UP_EVENT.up_char if x.zoom > 0] if self.UP_EVENT else [] ) # print(up_ship) acquire_char = None if up_ship and up_pool_flag: up_zoom: list[tuple[float, float]] = [(0, up_ship[0].zoom / 100)] # 初始化概率 cur_ = up_ship[0].zoom / 100 for i in range(len(up_ship)): try: up_zoom.append((cur_, cur_ + up_ship[i + 1].zoom / 100)) cur_ += up_ship[i + 1].zoom / 100 except IndexError: pass rand = random.random() # 抽取up for i, zoom in enumerate(up_zoom): if zoom[0] <= rand <= zoom[1]: try: acquire_char = [ x for x in self.ALL_CHAR if x.name == up_ship[i].name ][0] except IndexError: pass # 没有up或者未抽取到up if not acquire_char: star = self.get_star( # [4, 3, 2, 1], [4, 3, 2, 2], [ self.config.AZUR_FOUR_P, self.config.AZUR_THREE_P, self.config.AZUR_TWO_P, self.config.AZUR_ONE_P, ], ) acquire_char = random.choice( [ x for x in self.ALL_CHAR if x.star == star and x.type_ in type_ and not x.limited ] ) return acquire_char async def draw(self, count: int, **kwargs) -> MessageFactory: index2card = self.get_cards(count, **kwargs) cards = [card[0] for card in index2card] up_list = [x.name for x in self.UP_EVENT.up_char] if self.UP_EVENT else [] result = self.format_result(index2card, **{**kwargs, "up_list": up_list}) gen_img = await self.generate_img(cards) return MessageFactory([Image(gen_img.pic2bytes()), Text(result)]) async def generate_card_img(self, card: AzurChar) -> BuildImage: sep_w = 5 sep_t = 5 sep_b = 20 w = 100 h = 100 bg = BuildImage(w + sep_w * 2, h + sep_t + sep_b) frame_path = str(self.img_path / f"{card.star}_star.png") frame = BuildImage(w, h, background=frame_path) img_path = str(self.img_path / f"{cn2py(card.name)}.png") img = BuildImage(w, h, background=img_path) # 加圆角 await frame.circle_corner(6) await img.circle_corner(6) await bg.paste(img, (sep_w, sep_t)) await bg.paste(frame, (sep_w, sep_t)) # 加名字 text = card.name[:6] + "..." if len(card.name) > 7 else card.name font = load_font(fontsize=14) text_w, text_h = BuildImage.get_text_size(text, font) draw = ImageDraw.Draw(bg.markImg) draw.text( (sep_w + (w - text_w) / 2, h + sep_t + (sep_b - text_h) / 2), text, font=font, fill=["#808080", "#3b8bff", "#8000ff", "#c90", "#ee494c"][card.star - 1], ) return bg def _init_data(self): self.ALL_CHAR = [ AzurChar( name=value["名称"], star=int(value["星级"]), limited="可以建造" not in value["获取途径"], type_=value["类型"], ) for value in self.load_data().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_EVENT = UpEvent.parse_obj(data.get("char", {})) except ValidationError: logger.warning(f"{self.game_name}_up_char 解析出错") def dump_up_char(self): if self.UP_EVENT: data = {"char": json.loads(self.UP_EVENT.json())} self.dump_data(data, f"draw_card_up/{self.game_name}_up_char.json") self.dump_data(data, f"draw_card_up/{self.game_name}_up_char.json") async def _update_info(self): info = {} # 更新图鉴 url = "https://wiki.biligame.com/blhx/舰娘图鉴" result = await self.get_url(url) if not result: logger.warning(f"更新 {self.game_name_cn} 出错") return dom = etree.HTML(result, etree.HTMLParser()) contents = dom.xpath( "//div[@class='mw-body-content mw-content-ltr']/div[@class='mw-parser-output']" ) for index, content in enumerate(contents): char_list = content.xpath("./div[@id='CardSelectTr']/div") for char in char_list: try: name = char.xpath("./span/a/text()")[0] frame = char.xpath("./div/div/a/img/@alt")[0] avatar = char.xpath("./div/img/@srcset")[0] except IndexError: continue member_dict = { "名称": remove_prohibited_str(name), "头像": unquote(str(avatar).split(" ")[-2]), "星级": self.parse_star(frame), "类型": char.xpath("./@data-param1")[0].split(",")[-1], } info[member_dict["名称"]] = member_dict # 更新额外信息 for key in info.keys(): # TODO: 各种舰娘·改获取错误 url = f"https://wiki.biligame.com/blhx/{key}" result = await self.get_url(url) if not result: info[key]["获取途径"] = [] logger.warning(f"{self.game_name_cn} 获取额外信息错误 {key}") continue try: dom = etree.HTML(result, etree.HTMLParser()) time = dom.xpath( "//table[@class='wikitable sv-general']/tbody[1]/tr[4]/td[2]//text()" )[0] sources = [] if "无法建造" in time: sources.append("无法建造") elif "活动已关闭" in time: sources.append("活动限定") else: sources.append("可以建造") info[key]["获取途径"] = sources except IndexError: info[key]["获取途径"] = [] logger.warning(f"{self.game_name_cn} 获取额外信息错误 {key}") self.dump_data(info) logger.info(f"{self.game_name_cn} 更新成功") # 下载头像 for value in info.values(): await self.download_img(value["头像"], value["名称"]) # 下载头像框 idx = 1 BLHX_URL = "https://patchwiki.biligame.com/images/blhx" for url in [ "/1/15/pxho13xsnkyb546tftvh49etzdh74cf.png", "/a/a9/k8t7nx6c8pan5vyr8z21txp45jxeo66.png", "/a/a5/5whkzvt200zwhhx0h0iz9qo1kldnidj.png", "/a/a2/ptog1j220x5q02hytpwc8al7f229qk9.png", "/6/6d/qqv5oy3xs40d3055cco6bsm0j4k4gzk.png", ]: await self.download_img(BLHX_URL + url, f"{idx}_star") idx += 1 await self.update_up_char() @staticmethod def parse_star(star: str) -> int: if star in ["舰娘头像外框普通.png", "舰娘头像外框白色.png"]: return 1 elif star in ["舰娘头像外框稀有.png", "舰娘头像外框蓝色.png"]: return 2 elif star in ["舰娘头像外框精锐.png", "舰娘头像外框紫色.png"]: return 3 elif star in ["舰娘头像外框超稀有.png", "舰娘头像外框金色.png"]: return 4 elif star in ["舰娘头像外框海上传奇.png", "舰娘头像外框彩色.png"]: return 5 elif star in [ "舰娘头像外框最高方案.png", "舰娘头像外框决战方案.png", "舰娘头像外框超稀有META.png", "舰娘头像外框精锐META.png", ]: return 6 else: return 6 async def update_up_char(self): url = "https://wiki.biligame.com/blhx/游戏活动表" result = await self.get_url(url) if not result: logger.warning(f"{self.game_name_cn}获取活动表出错") return try: dom = etree.HTML(result, etree.HTMLParser()) dd = dom.xpath("//div[@class='timeline2']/dl/dd/a")[0] url = "https://wiki.biligame.com" + dd.xpath("./@href")[0] title = dd.xpath("string(.)") result = await self.get_url(url) if not result: logger.warning(f"{self.game_name_cn}获取活动页面出错") return dom = etree.HTML(result, etree.HTMLParser()) timer = dom.xpath("//span[@class='eventTimer']")[0] start_time = dateparser.parse(timer.xpath("./@data-start")[0]) end_time = dateparser.parse(timer.xpath("./@data-end")[0]) ships = dom.xpath("//table[@class='shipinfo']") up_chars = [] for ship in ships: name = ship.xpath("./tbody/tr/td[2]/p/a/@title")[0] type_ = ship.xpath("./tbody/tr/td[2]/p/small/text()")[0] # 舰船类型 try: p = float(str(ship.xpath(".//sup/text()")[0]).strip("%")) except (IndexError, ValueError): p = 0 star = self.parse_star( ship.xpath("./tbody/tr/td[1]/div/div/div/a/img/@alt")[0] ) up_chars.append( UpChar(name=name, star=star, limited=False, zoom=p, type_=type_) ) self.UP_EVENT = UpEvent( title=title, pool_img="", start_time=start_time, end_time=end_time, up_char=up_chars, ) self.dump_up_char() 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_EVENT: return MessageFactory( [Text(f"重载成功！\n当前活动：{self.UP_EVENT.title}")] )