mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
346 lines
12 KiB
Python
346 lines
12 KiB
Python
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",
|
||
)
|