mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 06:12:53 +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",
|
|||
|
|
)
|