sansenhoshiandClaude
9371a28e35
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>
79 lines
2.6 KiB
Python
79 lines
2.6 KiB
Python
"""今日老婆 —— 触发器(Trigger + View)层。"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import base64
|
|
import datetime
|
|
from io import BytesIO
|
|
from random import choice
|
|
|
|
from nonebot import on_command
|
|
from nonebot.adapters.onebot.v11 import MessageEvent, Bot, MessageSegment
|
|
from nonebot.plugin import PluginMetadata
|
|
from PIL import Image
|
|
|
|
from ..services.store import get_member_list, load_group_config, write_group_config
|
|
from ..utils import download_avatar, png_resize
|
|
|
|
__plugin_meta__ = PluginMetadata(
|
|
name="今日老婆",
|
|
description="随机抓取群友作为老婆",
|
|
usage="发送【今日老婆】",
|
|
type="application",
|
|
)
|
|
|
|
wife = on_command("今日老婆", aliases={"今日老婆"})
|
|
|
|
|
|
async def get_wife_info(member_info, qq_id):
|
|
img = await download_avatar(qq_id)
|
|
avatar = Image.open(BytesIO(img)).convert("RGBA")
|
|
avatar = png_resize(avatar, new_width=145, new_height=145)
|
|
b_io = BytesIO()
|
|
avatar.save(b_io, format="PNG")
|
|
avatar_str = "base64://" + base64.b64encode(b_io.getvalue()).decode()
|
|
member_name = member_info["card"] or member_info["nickname"]
|
|
return (
|
|
MessageSegment.text("你今天的群友老婆是:"),
|
|
MessageSegment.image(avatar_str),
|
|
MessageSegment.text(f"{member_name}({qq_id})"),
|
|
)
|
|
|
|
|
|
@wife.handle()
|
|
async def wife_handle(bot: Bot, ev: MessageEvent):
|
|
group_id = ev.group_id
|
|
user_id = ev.user_id
|
|
bot_id = ev.self_id
|
|
wife_id = None
|
|
today = str(datetime.date.today())
|
|
config = load_group_config(group_id)
|
|
|
|
if config is not None:
|
|
if str(user_id) in list(config):
|
|
if config[str(user_id)][1] == today:
|
|
wife_id = config[str(user_id)][0]
|
|
else:
|
|
del config[str(user_id)]
|
|
|
|
if wife_id is None:
|
|
all_list = await bot.get_group_member_list(group_id=group_id)
|
|
id_list = get_member_list(all_list)
|
|
id_list.remove(bot_id)
|
|
id_list.remove(user_id)
|
|
if config is not None:
|
|
for record_id in list(config):
|
|
if config[record_id][1] != today:
|
|
del config[record_id]
|
|
else:
|
|
try:
|
|
id_list.remove(int(config[record_id][0]))
|
|
except Exception:
|
|
del config[record_id]
|
|
wife_id = choice(id_list)
|
|
|
|
write_group_config(group_id, user_id, wife_id, today, config)
|
|
member_info = await bot.get_group_member_info(group_id=group_id, user_id=wife_id)
|
|
result = await get_wife_info(member_info, wife_id)
|
|
await bot.send(ev, result, at_sender=True)
|