feat✨: CSGO开箱

zhenxun/plugins/open_cases/__init__.py

@@ -0,0 +1,345 @@
+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",
+    )