import asyncio import traceback from dataclasses import dataclass from typing import Optional, Set, Tuple, Any import nonebot from cn2an import cn2an from configs.config import Config from nonebot import on_keyword, on_regex from nonebot.adapters.onebot.v11 import MessageEvent from nonebot.log import logger from nonebot.matcher import Matcher from nonebot.params import RegexGroup from nonebot.permission import SUPERUSER from nonebot.typing import T_Handler from nonebot_plugin_apscheduler import scheduler from .handles.azur_handle import AzurHandle from .handles.ba_handle import BaHandle from .handles.base_handle import BaseHandle from .handles.fgo_handle import FgoHandle from .handles.genshin_handle import GenshinHandle from .handles.guardian_handle import GuardianHandle from .handles.onmyoji_handle import OnmyojiHandle from .handles.pcr_handle import PcrHandle from .handles.pretty_handle import PrettyHandle from .handles.prts_handle import PrtsHandle __zx_plugin_name__ = "游戏抽卡" __plugin_usage__ = """ usage： 模拟赛马娘，原神，明日方舟，坎公骑冠剑，公主连结(国/台)，碧蓝航线，FGO，阴阳师，碧蓝档案进行抽卡 指令： 原神[1-180]抽: 原神常驻池 原神角色[1-180]抽: 原神角色UP池子 原神角色2池[1-180]抽: 原神角色UP池子 原神武器[1-180]抽: 原神武器UP池子 重置原神抽卡: 清空当前卡池的抽卡次数[即从0开始计算UP概率] 方舟[1-300]抽: 方舟卡池，当有当期UP时指向UP池 赛马娘[1-200]抽: 赛马娘卡池，当有当期UP时指向UP池 坎公骑冠剑[1-300]抽: 坎公骑冠剑卡池，当有当期UP时指向UP池 pcr/公主连接[1-300]抽: 公主连接卡池 碧蓝航线/碧蓝[重型/轻型/特型/活动][1-300]抽: 碧蓝航线重型/轻型/特型/活动卡池 fgo[1-300]抽: fgo卡池 阴阳师[1-300]抽: 阴阳师卡池 ba/碧蓝档案[1-200]抽：碧蓝档案卡池 * 以上指令可以通过 XX一井 来指定最大抽取数量 * * 示例：原神一井 * """.strip() __plugin_superuser_usage__ = """ usage： 卡池方面的更新 指令： 更新方舟信息 重载方舟卡池 更新原神信息 重载原神卡池 更新赛马娘信息 重载赛马娘卡池 更新坎公骑冠剑信息 更新碧蓝航线信息 更新fgo信息 更新阴阳师信息 """.strip() __plugin_des__ = "就算是模拟抽卡也不能改变自己是个非酋" __plugin_cmd__ = [ "原神[1-180]抽", "原神角色[1-180]抽", "原神武器[1-180]抽", "重置原神抽卡", "方舟[1-300]抽", "赛马娘[1-200]抽", "坎公骑冠剑[1-300]抽", "pcr/公主连接[1-300]抽", "fgo[1-300]抽", "阴阳师[1-300]抽", "碧蓝档案[1-200]抽", "更新方舟信息 [_superuser]", "重载方舟卡池 [_superuser]", "更新原神信息 [_superuser]", "重载原神卡池 [_superuser]", "更新赛马娘信息 [_superuser]", "重载赛马娘卡池 [_superuser]", "更新坎公骑冠剑信息 [_superuser]", "更新碧蓝航线信息 [_superuser]", "更新fgo信息 [_superuser]", "更新阴阳师信息 [_superuser]", "更新碧蓝档案信息 [_superuser]", ] __plugin_type__ = ("抽卡相关", 1) __plugin_version__ = 0.1 __plugin_author__ = "HibiKier" __plugin_settings__ = { "level": 5, "default_status": True, "limit_superuser": False, "cmd": ["游戏抽卡", "抽卡"], } @dataclass class Game: keywords: Set[str] handle: BaseHandle flag: bool max_count: int = 300 # 一次最大抽卡数 reload_time: Optional[int] = None # 重载UP池时间（小时） has_other_pool: bool = False games = None def create_matchers(): def draw_handler(game: Game) -> T_Handler: async def handler( matcher: Matcher, event: MessageEvent, args: Tuple[Any, ...] = RegexGroup() ): pool_name, pool_type_, num, unit = args if num == "单": num = 1 else: try: num = int(cn2an(num, mode="smart")) except ValueError: await matcher.finish("必！须！是！数！字！") if unit == "井": num *= game.max_count if num < 1: await matcher.finish("虚空抽卡？？？") elif num > game.max_count: await matcher.finish("一井都满不足不了你嘛！快爬开！") pool_name = ( pool_name.replace("池", "") .replace("武器", "arms") .replace("角色", "char") .replace("卡牌", "card") .replace("卡", "card") ) try: if pool_type_ in ["2池", "二池"]: pool_name = pool_name + "1" res = game.handle.draw(num, pool_name=pool_name, user_id=event.user_id) except: logger.warning(traceback.format_exc()) await matcher.finish("出错了...") await matcher.finish(res, at_sender=True) return handler def update_handler(game: Game) -> T_Handler: async def handler(matcher: Matcher): await game.handle.update_info() await matcher.finish("更新完成！") return handler def reload_handler(game: Game) -> T_Handler: async def handler(matcher: Matcher): res = await game.handle.reload_pool() if res: await matcher.finish(res) return handler def reset_handler(game: Game) -> T_Handler: async def handler(matcher: Matcher, event: MessageEvent): if game.handle.reset_count(event.user_id): await matcher.finish("重置成功！") return handler def scheduled_job(game: Game) -> T_Handler: async def handler(): await game.handle.reload_pool() return handler for game in games: pool_pattern = r"([^\s单0-9零一二三四五六七八九百十]{0,3})" num_pattern = r"(单|[0-9零一二三四五六七八九百十]{1,3})" unit_pattern = r"([抽|井|连])" pool_type = "()" if game.has_other_pool: pool_type = r"([2二]池)?" draw_regex = r".*?(?:{})\s*{}\s*{}\s*{}\s*{}".format( "|".join(game.keywords), pool_pattern, pool_type, num_pattern, unit_pattern ) update_keywords = {f"更新{keyword}信息" for keyword in game.keywords} reload_keywords = {f"重载{keyword}卡池" for keyword in game.keywords} reset_keywords = {f"重置{keyword}抽卡" for keyword in game.keywords} if game.flag: on_regex(draw_regex, priority=5, block=True).append_handler( draw_handler(game) ) on_keyword( update_keywords, priority=1, block=True, permission=SUPERUSER ).append_handler(update_handler(game)) on_keyword( reload_keywords, priority=1, block=True, permission=SUPERUSER ).append_handler(reload_handler(game)) on_keyword(reset_keywords, priority=5, block=True).append_handler( reset_handler(game) ) if game.reload_time: scheduler.add_job( scheduled_job(game), trigger="cron", hour=game.reload_time, minute=1 ) # create_matchers() # 更新资源 @scheduler.scheduled_job( "cron", hour=4, minute=1, ) async def _(): tasks = [] for game in games: if game.flag: tasks.append(asyncio.ensure_future(game.handle.update_info())) await asyncio.gather(*tasks) driver = nonebot.get_driver() @driver.on_startup async def _(): global games if not games: from .config import draw_config games = ( Game( {"azur", "碧蓝航线"}, AzurHandle(), Config.get_config("draw_card", "AZUR_FLAG", True), ), Game( {"fgo", "命运冠位指定"}, FgoHandle(), Config.get_config("draw_card", "FGO_FLAG", True), ), Game( {"genshin", "原神"}, GenshinHandle(), Config.get_config("draw_card", "GENSHIN_FLAG", True), max_count=180, reload_time=18, has_other_pool=True, ), Game( {"guardian", "坎公骑冠剑"}, GuardianHandle(), Config.get_config("draw_card", "GUARDIAN_FLAG", True), reload_time=4, ), Game( {"onmyoji", "阴阳师"}, OnmyojiHandle(), Config.get_config("draw_card", "ONMYOJI_FLAG", True), ), Game( {"pcr", "公主连结", "公主连接", "公主链接", "公主焊接"}, PcrHandle(), Config.get_config("draw_card", "PCR_FLAG", True), ), Game( {"pretty", "马娘", "赛马娘"}, PrettyHandle(), Config.get_config("draw_card", "PRETTY_FLAG", True), max_count=200, reload_time=4, ), Game( {"prts", "方舟", "明日方舟"}, PrtsHandle(), Config.get_config("draw_card", "PRTS_FLAG", True), reload_time=4, ), Game( {"ba", "碧蓝档案"}, BaHandle(), Config.get_config("draw_card", "BA_FLAG", True), ), ) create_matchers() tasks = [] for game in games: if game.flag: game.handle.init_data() if not game.handle.data_exists(): tasks.append(asyncio.ensure_future(game.handle.update_info())) await asyncio.gather(*tasks)