import asyncio import random from datetime import datetime, timedelta from typing import List from nonebot.plugin import PluginMetadata from nonebot_plugin_alconna import Arparma, Match from nonebot_plugin_apscheduler import scheduler from nonebot_plugin_saa import Image, MessageFactory, Text from nonebot_plugin_session import EventSession from zhenxun.configs.utils import PluginCdBlock, PluginExtraData, RegisterConfig, Task from zhenxun.services.log import logger from zhenxun.utils.image_utils import text2image from .command import ( _group_open_matcher, _knifes_matcher, _multiple_matcher, _my_open_matcher, _open_matcher, _price_matcher, _reload_matcher, _show_case_matcher, _update_image_matcher, _update_matcher, ) from .open_cases_c import ( auto_update, get_my_knifes, group_statistics, open_case, open_multiple_case, total_open_statistics, ) from .utils import ( CASE2ID, KNIFE2ID, CaseManager, build_case_image, download_image, get_skin_case, init_skin_trends, reset_count_daily, update_skin_data, ) __plugin_meta__ = PluginMetadata( name="CSGO开箱", description="csgo模拟开箱[戒赌]", usage=""" 指令： 开箱 ?[武器箱] [1-30]连开箱 ?[武器箱] 我的开箱 我的金色 群开箱统计 查看武器箱?[武器箱] * 不包含[武器箱]时随机开箱 * 示例: 查看武器箱 示例: 查看武器箱英勇 """.strip(), extra=PluginExtraData( author="HibiKier", version="0.1", superuser_help=""" 更新皮肤指令 重置开箱： 重置今日开箱所有次数 指令： 更新武器箱 ?[武器箱/ALL] 更新皮肤 ?[名称/ALL1] 更新皮肤 ?[名称/ALL1] -S: (必定更新罕见皮肤所属箱子) 更新武器箱图片 * 不指定武器箱时则全部更新 * * 过多的爬取会导致账号API被封 * """.strip(), menu_type="抽卡相关", tasks=[Task(module="open_case_reset_remind", name="每日开箱重置提醒")], limits=[PluginCdBlock(result="着什么急啊，慢慢来！")], configs=[ RegisterConfig( key="INITIAL_OPEN_CASE_COUNT", value=20, help="初始每日开箱次数", default_value=20, type=int, ), RegisterConfig( key="EACH_IMPRESSION_ADD_COUNT", value=3, help="每 * 点好感度额外增加开箱次数", default_value=3, type=int, ), RegisterConfig(key="COOKIE", value=None, help="BUFF的cookie"), RegisterConfig( key="DAILY_UPDATE", value=None, help="每日自动更新的武器箱，存在'ALL'时则更新全部武器箱", type=List[str], ), RegisterConfig( key="DEFAULT_OPEN_CASE_RESET_REMIND", module="_task", value=True, help="被动 每日开箱重置提醒 进群默认开关状态", default_value=True, type=bool, ), ], ).dict(), ) # cases_matcher_group = MatcherGroup(priority=5, permission=GROUP, block=True) # k_open_case = cases_matcher_group.on_command("开箱") # reload_count = cases_matcher_group.on_command("重置开箱", permission=SUPERUSER) # total_case_data = cases_matcher_group.on_command( # "我的开箱", aliases={"开箱统计", "开箱查询", "查询开箱"} # ) # group_open_case_statistics = cases_matcher_group.on_command("群开箱统计") # open_multiple = cases_matcher_group.on_regex("(.*)连开箱(.*)?") # update_case = on_command( # "更新武器箱", aliases={"更新皮肤"}, priority=1, permission=SUPERUSER, block=True # ) # update_case_image = on_command( # "更新武器箱图片", priority=1, permission=SUPERUSER, block=True # ) # show_case = on_command("查看武器箱", priority=5, block=True) # my_knifes = on_command("我的金色", priority=1, permission=GROUP, block=True) # show_skin = on_command("查看皮肤", priority=5, block=True) # price_trends = on_command("价格趋势", priority=5, block=True) @_price_matcher.handle() async def _( session: EventSession, arparma: Arparma, name: str, skin: str, abrasion: str, day: Match[int], ): name = name.replace("武器箱", "").strip() _day = 7 if day.available: _day = day.result if _day > 180: await Text("天数必须大于0且小于180").finish() result = await init_skin_trends(name, skin, abrasion, _day) if not result: await Text("未查询到数据...").finish(reply=True) await Image(result.pic2bytes()).send() logger.info( f"查看 [{name}:{skin}({abrasion})] 价格趋势", arparma.header_result, session=session, ) @_reload_matcher.handle() async def _(session: EventSession, arparma: Arparma): await reset_count_daily() logger.info("重置开箱次数", arparma.header_result, session=session) @_open_matcher.handle() async def _(session: EventSession, arparma: Arparma, name: Match[str]): gid = session.id3 or session.id2 if not session.id1: await Text("用户id为空...").finish() if not gid: await Text("群组id为空...").finish() case_name = None if name.available: case_name = name.result.replace("武器箱", "").strip() result = await open_case(session.id1, gid, case_name, session) await result.finish(reply=True) @_my_open_matcher.handle() async def _(session: EventSession, arparma: Arparma): gid = session.id3 or session.id2 if not session.id1: await Text("用户id为空...").finish() if not gid: await Text("群组id为空...").finish() await Text( await total_open_statistics(session.id1, gid), ).send(reply=True) logger.info("查询我的开箱", arparma.header_result, session=session) @_group_open_matcher.handle() async def _(session: EventSession, arparma: Arparma): gid = session.id3 or session.id2 if not gid: await Text("群组id为空...").finish() result = await group_statistics(gid) await Text(result).send(reply=True) logger.info("查询群开箱统计", arparma.header_result, session=session) @_knifes_matcher.handle() async def _(session: EventSession, arparma: Arparma): gid = session.id3 or session.id2 if not session.id1: await Text("用户id为空...").finish() if not gid: await Text("群组id为空...").finish() result = await get_my_knifes(session.id1, gid) await result.send(reply=True) logger.info("查询我的金色", arparma.header_result, session=session) @_multiple_matcher.handle() async def _(session: EventSession, arparma: Arparma, num: int, name: Match[str]): gid = session.id3 or session.id2 if not session.id1: await Text("用户id为空...").finish() if not gid: await Text("群组id为空...").finish() if num > 30: await Text("开箱次数不要超过30啊笨蛋！").finish() if num < 0: await Text("再负开箱就扣你明天开箱数了！").finish() case_name = None if name.available: case_name = name.result.replace("武器箱", "").strip() result = await open_multiple_case(session.id1, gid, case_name, num, session) await result.send(reply=True) logger.info(f"{num}连开箱", arparma.header_result, session=session) @_update_matcher.handle() async def _(session: EventSession, arparma: Arparma, name: Match[str]): case_name = None if name.available: case_name = name.result.strip() if not case_name: case_list = [] skin_list = [] for i, case_name in enumerate(CASE2ID): if case_name in CaseManager.CURRENT_CASES: case_list.append(f"{i+1}.{case_name} [已更新]") else: case_list.append(f"{i+1}.{case_name}") for skin_name in KNIFE2ID: skin_list.append(f"{skin_name}") text = "武器箱:\n" + "\n".join(case_list) + "\n皮肤:\n" + ", ".join(skin_list) img = await text2image(text, padding=20, color="#f9f6f2") await MessageFactory( [Text("未指定武器箱, 当前已包含武器箱/皮肤\n"), Image(img.pic2bytes())] ).finish() if case_name in ["ALL", "ALL1"]: if case_name == "ALL": case_list = list(CASE2ID.keys()) type_ = "武器箱" else: case_list = list(KNIFE2ID.keys()) type_ = "罕见皮肤" await Text(f"即将更新所有{type_}, 请稍等").send() for i, case_name in enumerate(case_list): try: info = await update_skin_data(case_name, arparma.find("s")) if "请先登录" in info: await Text(f"未登录, 已停止更新, 请配置BUFF token...").send() return rand = random.randint(300, 500) result = f"更新全部{type_}完成" if i < len(case_list) - 1: next_case = case_list[i + 1] result = f"将在 {rand} 秒后更新下一{type_}: {next_case}" await Text(f"{info}, {result}").send() logger.info(f"info, {result}", "更新武器箱", session=session) await asyncio.sleep(rand) except Exception as e: logger.error(f"更新{type_}: {case_name}", session=session, e=e) await Text(f"更新{type_}: {case_name} 发生错误: {type(e)}: {e}").send() await Text(f"更新全部{type_}完成").send() else: await Text(f"开始{arparma.header_result}: {case_name}, 请稍等").send() try: await Text(await update_skin_data(case_name, arparma.find("s"))).send( at_sender=True ) except Exception as e: logger.error(f"{arparma.header_result}: {case_name}", session=session, e=e) await Text( f"成功{arparma.header_result}: {case_name} 发生错误: {type(e)}: {e}" ).send() @_show_case_matcher.handle() async def _(session: EventSession, arparma: Arparma, name: Match[str]): case_name = None if name.available: case_name = name.result.strip() result = await build_case_image(case_name) if isinstance(result, str): await Text(result).send() else: await Image(result.pic2bytes()).send() logger.info("查看武器箱", arparma.header_result, session=session) @_update_image_matcher.handle() async def _(session: EventSession, arparma: Arparma, name: Match[str]): case_name = None if name.available: case_name = name.result.strip() await Text("开始更新图片...").send(reply=True) await download_image(case_name) await Text("更新图片完成...").send(at_sender=True) logger.info("更新武器箱图片", arparma.header_result, session=session) # 重置开箱 @scheduler.scheduled_job( "cron", hour=0, minute=1, ) async def _(): await reset_count_daily() @scheduler.scheduled_job( "cron", hour=0, minute=10, ) async def _(): now = datetime.now() hour = random.choice([0, 1, 2, 3]) date = now + timedelta(hours=hour) logger.debug(f"将在 {date} 时自动更新武器箱...", "更新武器箱") scheduler.add_job( auto_update, "date", run_date=date.replace(microsecond=0), id=f"auto_update_csgo_cases", )