diff --git a/plugins/draw_card/config.py b/plugins/draw_card/config.py new file mode 100644 index 00000000..71500f8e --- /dev/null +++ b/plugins/draw_card/config.py @@ -0,0 +1,167 @@ +import nonebot +from pathlib import Path +from nonebot.log import logger +from pydantic import BaseModel, Extra +from configs.config import Config as AConfig +from configs.path_config import DATA_PATH + +try: + import ujson as json +except ModuleNotFoundError: + import json + + +# 原神 +class GenshinConfig(BaseModel, extra=Extra.ignore): + GENSHIN_FIVE_P: float = 0.006 + GENSHIN_FOUR_P: float = 0.051 + GENSHIN_THREE_P: float = 0.43 + GENSHIN_G_FIVE_P: float = 0.016 + GENSHIN_G_FOUR_P: float = 0.13 + I72_ADD: float = 0.0585 + + +# 明日方舟 +class PrtsConfig(BaseModel, extra=Extra.ignore): + PRTS_SIX_P: float = 0.02 + PRTS_FIVE_P: float = 0.08 + PRTS_FOUR_P: float = 0.48 + PRTS_THREE_P: float = 0.42 + + +# 赛马娘 +class PrettyConfig(BaseModel, extra=Extra.ignore): + PRETTY_THREE_P: float = 0.03 + PRETTY_TWO_P: float = 0.18 + PRETTY_ONE_P: float = 0.79 + + +# 坎公骑冠剑 +class GuardianConfig(BaseModel, extra=Extra.ignore): + GUARDIAN_THREE_CHAR_P: float = 0.0275 + GUARDIAN_TWO_CHAR_P: float = 0.19 + GUARDIAN_ONE_CHAR_P: float = 0.7825 + GUARDIAN_THREE_CHAR_UP_P: float = 0.01375 + GUARDIAN_THREE_CHAR_OTHER_P: float = 0.01375 + GUARDIAN_EXCLUSIVE_ARMS_P: float = 0.03 + GUARDIAN_FIVE_ARMS_P: float = 0.03 + GUARDIAN_FOUR_ARMS_P: float = 0.09 + GUARDIAN_THREE_ARMS_P: float = 0.27 + GUARDIAN_TWO_ARMS_P: float = 0.58 + GUARDIAN_EXCLUSIVE_ARMS_UP_P: float = 0.01 + GUARDIAN_EXCLUSIVE_ARMS_OTHER_P: float = 0.02 + + +# 公主连结 +class PcrConfig(BaseModel, extra=Extra.ignore): + PCR_THREE_P: float = 0.025 + PCR_TWO_P: float = 0.18 + PCR_ONE_P: float = 0.795 + PCR_G_THREE_P: float = 0.025 + PCR_G_TWO_P: float = 0.975 + + +# 碧蓝航线 +class AzurConfig(BaseModel, extra=Extra.ignore): + AZUR_FIVE_P: float = 0.012 + AZUR_FOUR_P: float = 0.07 + AZUR_THREE_P: float = 0.12 + AZUR_TWO_P: float = 0.51 + AZUR_ONE_P: float = 0.3 + + +# 命运-冠位指定 +class FgoConfig(BaseModel, extra=Extra.ignore): + FGO_SERVANT_FIVE_P: float = 0.01 + FGO_SERVANT_FOUR_P: float = 0.03 + FGO_SERVANT_THREE_P: float = 0.4 + FGO_CARD_FIVE_P: float = 0.04 + FGO_CARD_FOUR_P: float = 0.12 + FGO_CARD_THREE_P: float = 0.4 + + +# 阴阳师 +class OnmyojiConfig(BaseModel, extra=Extra.ignore): + ONMYOJI_SP: float = 0.0025 + ONMYOJI_SSR: float = 0.01 + ONMYOJI_SR: float = 0.2 + ONMYOJI_R: float = 0.7875 + + +class Config(BaseModel, extra=Extra.ignore): + # 开关 + PRTS_FLAG: bool = AConfig.get_config("draw_card", "PRTS_FLAG") + GENSHIN_FLAG: bool = AConfig.get_config("draw_card", "GENSHIN_FLAG") + PRETTY_FLAG: bool = AConfig.get_config("draw_card", "PRETTY_FLAG") + GUARDIAN_FLAG: bool = AConfig.get_config("draw_card", "GUARDIAN_FLAG") + PCR_FLAG: bool = AConfig.get_config("draw_card", "PCR_FLAG") + AZUR_FLAG: bool = AConfig.get_config("draw_card", "AZUR_FLAG") + FGO_FLAG: bool = AConfig.get_config("draw_card", "FGO_FLAG") + ONMYOJI_FLAG: bool = AConfig.get_config("draw_card", "ONMYOJI_FLAG") + + # 其他配置 + PCR_TAI: bool = AConfig.get_config("draw_card", "PCR_TAI") + SEMAPHORE: int = AConfig.get_config("draw_card", "SEMAPHORE") + + # 抽卡概率 + prts: PrtsConfig = PrtsConfig() + genshin: GenshinConfig = GenshinConfig() + pretty: PrettyConfig = PrettyConfig() + guardian: GuardianConfig = GuardianConfig() + pcr: PcrConfig = PcrConfig() + azur: AzurConfig = AzurConfig() + fgo: FgoConfig = FgoConfig() + onmyoji: OnmyojiConfig = OnmyojiConfig() + + +driver = nonebot.get_driver() + +DRAW_PATH = DATA_PATH / "draw_card" +config_path = DRAW_PATH / "draw_card_config" / "draw_card_config.json" + +draw_config: Config = Config() + + +for game_flag, game_name in zip( + [ + "PRTS_FLAG", + "GENSHIN_FLAG", + "PRETTY_FLAG", + "GUARDIAN_FLAG", + "PCR_FLAG", + "AZUR_FLAG", + "FGO_FLAG", + "ONMYOJI_FLAG", + "PCR_TAI", + ], + ["明日方舟", "原神", "赛马娘", "坎公骑冠剑", "公主连结", "碧蓝航线", "命运-冠位指定（FGO）", "阴阳师", "pcr台服卡池"], +): + AConfig.add_plugin_config( + "draw_card", + game_flag, + True, + name="游戏抽卡", + help_=f"{game_name} 抽卡开关", + default_value=True, + ) +AConfig.add_plugin_config( + "draw_card", "SEMAPHORE", 5, help_=f"异步数据下载数量限制", default_value=5 +) + + +@driver.on_startup +def check_config(): + global draw_config + draw_config = Config() + if not config_path.exists(): + config_path.parent.mkdir(parents=True, exist_ok=True) + draw_config = Config() + logger.warning("draw_card：配置文件不存在，已重新生成配置文件.....") + + with config_path.open("w", encoding="utf8") as fp: + json.dump( + draw_config.dict(), + fp, + indent=4, + ensure_ascii=False, + ) diff --git a/plugins/draw_card/util.py b/plugins/draw_card/util.py new file mode 100644 index 00000000..a234cb68 --- /dev/null +++ b/plugins/draw_card/util.py @@ -0,0 +1,55 @@ +import platform +import pypinyin +from PIL.ImageFont import FreeTypeFont +from PIL import Image, ImageDraw, ImageFont +from PIL.Image import Image as IMG +from configs.path_config import FONT_PATH + + +def cn2py(word) -> str: + temp = "" + for i in pypinyin.pinyin(word, style=pypinyin.NORMAL): + temp += "".join(i) + return temp + + +# 移除windows和linux下特殊字符 +def remove_prohibited_str(name: str) -> str: + if platform.system().lower() == "windows": + tmp = "" + for i in name: + if i not in ["\\", "/", ":", "*", "?", '"', "<", ">", "|"]: + tmp += i + name = tmp + else: + name = name.replace("/", "\\") + return name + + +def load_font(font_name: str = "msyh.ttf", fontsize: int = 16) -> FreeTypeFont: + return ImageFont.truetype( + str(FONT_PATH / f"{font_name}"), fontsize, encoding="utf-8" + ) + + +def circled_number(num: int) -> IMG: + font = load_font(fontsize=450) + text = str(num) + text_w = font.getsize(text)[0] + w = 240 + text_w + w = w if w >= 500 else 500 + img = Image.new("RGBA", (w, 500)) + draw = ImageDraw.Draw(img) + draw.ellipse(((0, 0), (500, 500)), fill="red") + draw.ellipse(((w - 500, 0), (w, 500)), fill="red") + draw.rectangle(((250, 0), (w - 250, 500)), fill="red") + draw.text( + (120, -60), + text, + font=font, + fill="white", + stroke_width=10, + stroke_fill="white", + ) + return img +