refactor: restructure plugins per MTSS standard, consolidate assets to res/
Per docs/plugin-audit-report.md (plugins normalized to Trigger(handlers) → Service(services) → Model(repository/models) + utils): - Split monolithic __init__.py into handlers/services/utils across dailywife, deer_pipe, dice, galgame_card, helldivers_tools, huoziyinshua, learning_chat, makeaquote, mc_server_status, ncm_saying, picfinder_take, picstatus, random_jm_code, regif, steam_info, video_analysis, group_tools - Move static assets under res/: deer_pipe font/img, makeaquote font, helldivers img/templates, huoziyinshua HuoZiYinShua - Add config.py + register_config_items to ncm_saying, random_jm_code, group_tools; learning_chat unified config bridge - Remove deprecated: voice_trans plugin, bf_bot/test.py, dead code in dailywife/deer_pipe, empty dirs, debug scripts under helldivers temp - Disable brash_general_supercredits_tools (stub comment only) - bot.py: optional stdout/stderr redirect to log file for Web log viewer, force ANSI colorize on non-TTY sinks - Move runtime data (jm_code.json) out of plugin dir into hexi/data - Docs: plugin-audit-report.md; README reflects removed plugins Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
Claude
@@ -0,0 +1,289 @@
|
||||
"""Steam 信息 —— 触发器(Trigger)层。"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import asyncio
|
||||
|
||||
from nonebot import logger, on_command, require
|
||||
from nonebot.adapters import Message, Event
|
||||
from nonebot.adapters.onebot.v11 import Bot, MessageEvent, MessageSegment
|
||||
from nonebot.params import CommandArg, Depends
|
||||
from nonebot.permission import SUPERUSER
|
||||
from nonebot_plugin_alconna import At, Image, Target, Text, UniMessage
|
||||
|
||||
from hexi.plugins.nonebot_plugin_hexi_core.cooldown import cooldown
|
||||
|
||||
# 业务/状态在插件根 __init__ 中(本插件为社区插件,服务层保留在根模块)
|
||||
from .. import (
|
||||
__plugin_meta__,
|
||||
STEAM_ID_OFFSET,
|
||||
avatar_path,
|
||||
bind_data,
|
||||
cache_path,
|
||||
config,
|
||||
db_path,
|
||||
disable_parent_data,
|
||||
draw_friends_status,
|
||||
draw_player_status,
|
||||
fetch_play_time,
|
||||
get_game_cn_names,
|
||||
get_game_icons,
|
||||
get_group_nickname,
|
||||
get_parent_info,
|
||||
get_steam_id,
|
||||
get_steam_users_info,
|
||||
get_target,
|
||||
get_user_data,
|
||||
image_to_bytes,
|
||||
key_select,
|
||||
key_select_2,
|
||||
localize_game_name,
|
||||
parent_data,
|
||||
simplize_steam_player_data,
|
||||
steam_info_data,
|
||||
_fill_missing_icon_hashes,
|
||||
_query_game_names,
|
||||
_upsert_game_icon_hashes,
|
||||
update_steamdb_apps,
|
||||
)
|
||||
|
||||
help = on_command("steamhelp", aliases={"steam帮助"}, priority=10)
|
||||
bind = on_command("steambind", aliases={"绑定steam"}, priority=10)
|
||||
unbind = on_command("steamunbind", aliases={"解绑steam"}, priority=10)
|
||||
info = on_command("steaminfo", aliases={"steam信息"}, priority=10)
|
||||
check = cooldown(30, scope="group", hint="群里刚有人查过,{secs} 秒后再试")(
|
||||
on_command("steamcheck", aliases={"查看steam", "查steam", "谁在玩游戏"}, priority=10)
|
||||
)
|
||||
enable = on_command("steamenable", aliases={"启用steam"}, priority=10)
|
||||
disable = on_command("steamdisable", aliases={"禁用steam"}, priority=10)
|
||||
get_play_time = cooldown(15, hint="查询太频繁了,{secs} 秒后再试")(
|
||||
on_command("steamplaytime", aliases={"游戏时长表"}, priority=10)
|
||||
)
|
||||
update_steamdb = on_command(
|
||||
"steamdbupdate", aliases={"更新steamdb"}, priority=10, permission=SUPERUSER
|
||||
)
|
||||
|
||||
|
||||
@help.handle()
|
||||
async def help_handle():
|
||||
await help.finish(__plugin_meta__.usage)
|
||||
|
||||
|
||||
@bind.handle()
|
||||
async def bind_handle(event: Event, target: Target = Depends(get_target), cmd_arg: Message = CommandArg()):
|
||||
parent_id = target.parent_id or target.id
|
||||
arg = cmd_arg.extract_plain_text()
|
||||
if not arg.isdigit():
|
||||
await bind.finish(
|
||||
"请输入正确的 Steam ID 或 Steam好友代码，格式: steambind [Steam ID 或 Steam好友代码]"
|
||||
)
|
||||
steam_id = get_steam_id(arg)
|
||||
if user_data := bind_data.get(parent_id, event.get_user_id()):
|
||||
user_data["steam_id"] = steam_id
|
||||
bind_data.save()
|
||||
await bind.finish(f"已更新你的 Steam ID 为 {steam_id}")
|
||||
else:
|
||||
bind_data.add(
|
||||
parent_id,
|
||||
{"user_id": event.get_user_id(), "steam_id": steam_id, "nickname": None},
|
||||
)
|
||||
bind_data.save()
|
||||
await bind.finish(f"已绑定你的 Steam ID 为 {steam_id}")
|
||||
|
||||
|
||||
@unbind.handle()
|
||||
async def unbind_handle(event: Event, target: Target = Depends(get_target)):
|
||||
parent_id = target.parent_id or target.id
|
||||
user_id = event.get_user_id()
|
||||
if bind_data.get(parent_id, user_id) is not None:
|
||||
bind_data.remove(parent_id, user_id)
|
||||
bind_data.save()
|
||||
await unbind.finish("已解绑 Steam ID")
|
||||
else:
|
||||
await unbind.finish("未绑定 Steam ID")
|
||||
|
||||
|
||||
@info.handle()
|
||||
async def info_handle(bot: Bot, event: Event, target: Target = Depends(get_target), arg: Message = CommandArg()):
|
||||
parent_id = target.parent_id or target.id
|
||||
uni_arg = await UniMessage.generate(message=arg, event=event, bot=bot)
|
||||
at = uni_arg[At]
|
||||
if len(at) != 0:
|
||||
user_id = at[0].target
|
||||
user_data = bind_data.get(parent_id, user_id)
|
||||
if user_data is None:
|
||||
await info.finish("该用户未绑定 Steam ID")
|
||||
steam_id = user_data["steam_id"]
|
||||
steam_friend_code = str(int(steam_id) - STEAM_ID_OFFSET)
|
||||
elif arg.extract_plain_text().strip() != "":
|
||||
steam_id = int(arg.extract_plain_text().strip())
|
||||
if steam_id < STEAM_ID_OFFSET:
|
||||
steam_friend_code = steam_id
|
||||
steam_id += STEAM_ID_OFFSET
|
||||
else:
|
||||
steam_friend_code = steam_id - STEAM_ID_OFFSET
|
||||
else:
|
||||
user_data = bind_data.get(parent_id, event.get_user_id())
|
||||
if user_data is None:
|
||||
await info.finish(
|
||||
"未绑定 Steam ID, 请使用 “steambind [Steam ID 或 Steam好友代码]” 绑定 Steam ID"
|
||||
)
|
||||
steam_id = user_data["steam_id"]
|
||||
steam_friend_code = str(int(steam_id) - STEAM_ID_OFFSET)
|
||||
|
||||
player_data = await get_user_data(steam_id, cache_path, config.proxy)
|
||||
names = await get_game_cn_names(
|
||||
[g.get("appid") for g in player_data["game_data"] if g.get("appid")],
|
||||
db_path,
|
||||
)
|
||||
for g in player_data["game_data"]:
|
||||
g["game_name"] = localize_game_name(g["game_name"], g.get("appid"), names)
|
||||
draw_data = [
|
||||
{
|
||||
"game_header": game["game_image"],
|
||||
"game_name": game["game_name"],
|
||||
"game_time": f"{game['play_time']} 小时",
|
||||
"last_play_time": game["last_played"],
|
||||
"achievements": game["achievements"],
|
||||
"completed_achievement_number": game.get("completed_achievement_number"),
|
||||
"total_achievement_number": game.get("total_achievement_number"),
|
||||
}
|
||||
for game in player_data["game_data"]
|
||||
]
|
||||
image = draw_player_status(
|
||||
player_data["background"], player_data["avatar"], player_data["player_name"],
|
||||
str(steam_friend_code), player_data["description"],
|
||||
player_data["recent_2_week_play_time"], draw_data,
|
||||
)
|
||||
await info.finish(await UniMessage(Image(raw=image_to_bytes(image))).export(bot))
|
||||
|
||||
|
||||
@check.handle()
|
||||
async def check_handle(bot: Bot, target: Target = Depends(get_target), arg: Message = CommandArg()):
|
||||
if arg.extract_plain_text().strip() != "":
|
||||
return None
|
||||
parent_id = target.parent_id or target.id
|
||||
steam_ids = bind_data.get_all(parent_id)
|
||||
current_key = key_select()
|
||||
logger.info(f"当前调用的key：{current_key}")
|
||||
steam_info = await get_steam_users_info(steam_ids, current_key, config.proxy, wait=False)
|
||||
if steam_info is None:
|
||||
await check.finish("Steam API 请求过于频繁,请稍后再试")
|
||||
players = steam_info["response"]["players"]
|
||||
names = await get_game_cn_names(
|
||||
[int(p["gameid"]) for p in players if p.get("gameid")], db_path,
|
||||
)
|
||||
for p in players:
|
||||
if p.get("gameid"):
|
||||
p["gameextrainfo"] = localize_game_name(p.get("gameextrainfo", ""), int(p["gameid"]), names)
|
||||
game_ids: list[int] = []
|
||||
seen: set[int] = set()
|
||||
for p in players:
|
||||
if p.get("gameid"):
|
||||
gid = int(p["gameid"])
|
||||
if gid not in seen:
|
||||
seen.add(gid)
|
||||
game_ids.append(gid)
|
||||
game_icons = await get_game_icons(game_ids, config.proxy, db_path) if game_ids else {}
|
||||
asyncio.create_task(_fill_missing_icon_hashes(players))
|
||||
parent_avatar, parent_name = await get_parent_info(bot, parent_id)
|
||||
steam_status_data = []
|
||||
for player in players:
|
||||
data = await simplize_steam_player_data(player, config.proxy, avatar_path, names)
|
||||
entry = bind_data.get_by_steam_id(parent_id, player["steamid"])
|
||||
data["nickname"] = await get_group_nickname(bot, parent_id, entry["user_id"]) if entry is not None else None
|
||||
data["game_icon"] = game_icons.get(data["gameid"])
|
||||
steam_status_data.append(data)
|
||||
image = draw_friends_status(parent_avatar, parent_name, steam_status_data)
|
||||
await target.send(UniMessage(Image(raw=image_to_bytes(image))))
|
||||
|
||||
|
||||
@enable.handle()
|
||||
async def enable_handle(target: Target = Depends(get_target)):
|
||||
parent_id = target.parent_id or target.id
|
||||
disable_parent_data.remove(parent_id)
|
||||
disable_parent_data.save()
|
||||
await enable.finish("已启用 Steam 播报")
|
||||
|
||||
|
||||
@disable.handle()
|
||||
async def disable_handle(target: Target = Depends(get_target)):
|
||||
parent_id = target.parent_id or target.id
|
||||
disable_parent_data.add(parent_id)
|
||||
disable_parent_data.save()
|
||||
await disable.finish("已禁用 Steam 播报")
|
||||
|
||||
|
||||
@update_steamdb.handle()
|
||||
async def update_steamdb_handle():
|
||||
await update_steamdb.send("正在拉取 Steam 全量 app 列表...")
|
||||
try:
|
||||
api_key = config.steam_api_key[0]
|
||||
count = await update_steamdb_apps(db_path, api_key)
|
||||
except Exception as e:
|
||||
logger.error(f"SteamDB 数据更新失败: {e}")
|
||||
await update_steamdb.finish(f"SteamDB 数据更新失败: {e}")
|
||||
if count:
|
||||
await update_steamdb.finish(f"Steam 数据更新完成,共 {count} 个游戏")
|
||||
else:
|
||||
await update_steamdb.finish("Steam 数据拉取失败,稍后重试")
|
||||
|
||||
|
||||
@get_play_time.handle()
|
||||
async def get_play_time_handle(bot: Bot, event: MessageEvent, target: Target = Depends(get_target), cmd_arg: Message = CommandArg()):
|
||||
parent_id = target.parent_id or target.id
|
||||
user_data = None
|
||||
logger.info(f"游玩时长表:进入处理, parent={parent_id}")
|
||||
arg_text = cmd_arg.extract_plain_text().strip()
|
||||
if arg_text:
|
||||
if not arg_text.isdigit():
|
||||
await get_play_time.finish(
|
||||
"请输入正确的 Steam ID 或 Steam好友代码，格式: steamplaytime [Steam ID 或 Steam好友代码]"
|
||||
)
|
||||
steam_id = get_steam_id(arg_text)
|
||||
else:
|
||||
user_data = bind_data.get(parent_id, event.get_user_id())
|
||||
if user_data is None:
|
||||
await get_play_time.finish(
|
||||
"未绑定 Steam ID，请使用 steamplaytime [Steam ID 或 Steam好友代码] 查询，或先使用 steambind 绑定"
|
||||
)
|
||||
steam_id = user_data["steam_id"]
|
||||
await get_play_time.send("正在获取数据，请耐心等待")
|
||||
logger.info(f"游玩时长表:开始获取数据, steam_id={steam_id}")
|
||||
games = await fetch_play_time(steam_id)
|
||||
if games is None:
|
||||
await get_play_time.finish("Steam API 请求过于频繁,请稍后再试")
|
||||
if not games:
|
||||
await get_play_time.finish("没有获取到游玩数据（游戏可能未公开）")
|
||||
icon_entries = {int(g["appid"]): g.get("clienticon") for g in games if g.get("appid") and g.get("clienticon")}
|
||||
if icon_entries:
|
||||
_upsert_game_icon_hashes(db_path, icon_entries)
|
||||
appids = [g["appid"] for g in games if g.get("appid")]
|
||||
game_map = _query_game_names(db_path, appids) if appids else {}
|
||||
names = {int(aid): entry["name_ch"] for aid, entry in game_map.items() if entry.get("name_ch")}
|
||||
for g in games:
|
||||
g["name"] = localize_game_name(g["name"], g.get("appid"), names)
|
||||
played = [g for g in games if g["playtime_hours"] > 0]
|
||||
if not played:
|
||||
await get_play_time.finish("没有游玩记录(游戏可能未公开或全部未游玩)")
|
||||
for i, g in enumerate(played, 1):
|
||||
g["rank"] = i
|
||||
try:
|
||||
from ..html_playtime import render_play_time_table
|
||||
steam_name, plugin_nick, avatar = "", "", None
|
||||
try:
|
||||
player_data = await get_user_data(steam_id, cache_path, config.proxy)
|
||||
steam_name = player_data.get("player_name", "")
|
||||
avatar = player_data.get("avatar")
|
||||
except Exception as e:
|
||||
logger.warning(f"获取玩家资料失败,省略头像昵称: {e}")
|
||||
if user_data:
|
||||
plugin_nick = await get_group_nickname(bot, parent_id, user_data["user_id"])
|
||||
img_bytes = await render_play_time_table(
|
||||
played, games, steam_name, plugin_nick, avatar, theme=config.steam_playtime_theme,
|
||||
)
|
||||
except Exception as e:
|
||||
logger.error(f"渲染游玩时长表图片失败: {e}", exc_info=True)
|
||||
await get_play_time.finish("游玩时长表渲染失败,请稍后重试")
|
||||
uni_msg = MessageSegment.reply(event.message_id) + await UniMessage(Image(raw=img_bytes)).export(bot)
|
||||
await get_play_time.finish(uni_msg)
|
||||
Reference in New Issue
Block a user