mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 06:12:53 +08:00
123 lines
3.9 KiB
Python
123 lines
3.9 KiB
Python
import os
|
|
import shutil
|
|
from typing import Annotated, Dict
|
|
|
|
import ujson as json
|
|
from nonebot import on_command
|
|
from nonebot.params import Command
|
|
from nonebot.plugin import PluginMetadata
|
|
from nonebot_plugin_alconna import Image
|
|
from nonebot_plugin_alconna import Text as alcText
|
|
from nonebot_plugin_alconna import UniMsg
|
|
from nonebot_plugin_session import EventSession
|
|
|
|
from zhenxun.configs.config import Config
|
|
from zhenxun.configs.path_config import DATA_PATH
|
|
from zhenxun.configs.utils import PluginExtraData, RegisterConfig
|
|
from zhenxun.services.log import logger
|
|
from zhenxun.utils.enum import PluginType
|
|
from zhenxun.utils.http_utils import AsyncHttpx
|
|
from zhenxun.utils.rules import admin_check, ensure_group
|
|
|
|
base_config = Config.get("admin_bot_manage")
|
|
|
|
__plugin_meta__ = PluginMetadata(
|
|
name="自定义群欢迎消息",
|
|
description="自定义群欢迎消息",
|
|
usage="""
|
|
设置欢迎消息 欢迎新人！
|
|
设置欢迎消息 欢迎你 -at
|
|
""".strip(),
|
|
extra=PluginExtraData(
|
|
author="HibiKier",
|
|
version="0.1",
|
|
plugin_type=PluginType.ADMIN,
|
|
admin_level=base_config.get("SET_GROUP_WELCOME_MESSAGE_LEVEL", 2),
|
|
configs=[
|
|
RegisterConfig(
|
|
module="admin_bot_manage",
|
|
key="SET_GROUP_WELCOME_MESSAGE_LEVEL",
|
|
value=2,
|
|
help="设置群欢迎消息所需要的管理员权限等级",
|
|
default_value=2,
|
|
)
|
|
],
|
|
).dict(),
|
|
)
|
|
|
|
_matcher = on_command(
|
|
"设置欢迎消息",
|
|
rule=admin_check("admin_bot_manage", "SET_GROUP_WELCOME_MESSAGE_LEVEL")
|
|
& ensure_group,
|
|
priority=5,
|
|
block=True,
|
|
)
|
|
|
|
|
|
BASE_PATH = DATA_PATH / "welcome_message"
|
|
BASE_PATH.mkdir(parents=True, exist_ok=True)
|
|
|
|
# 旧数据迁移
|
|
old_file = DATA_PATH / "custom_welcome_msg" / "custom_welcome_msg.json"
|
|
if old_file.exists():
|
|
try:
|
|
old_data: Dict[str, str] = json.load(old_file.open(encoding="utf8"))
|
|
for group_id, message in old_data.items():
|
|
file = BASE_PATH / "qq" / f"{group_id}" / "text.json"
|
|
file.parent.mkdir(parents=True, exist_ok=True)
|
|
json.dump(
|
|
{"at": "[at]" in message, "message": message.replace("[at]", "")},
|
|
file.open("w", encoding="utf8"),
|
|
ensure_ascii=False,
|
|
indent=4,
|
|
)
|
|
logger.debug("群欢迎消息数据迁移", group_id=group_id)
|
|
shutil.rmtree(old_file.parent.absolute())
|
|
except Exception as e:
|
|
pass
|
|
|
|
|
|
@_matcher.handle()
|
|
async def _(
|
|
session: EventSession,
|
|
message: UniMsg,
|
|
command: Annotated[tuple[str, ...], Command()],
|
|
):
|
|
path = BASE_PATH / f"{session.platform or session.bot_type}" / f"{session.id2}"
|
|
if session.id3:
|
|
path = (
|
|
BASE_PATH
|
|
/ f"{session.platform or session.bot_type}"
|
|
/ f"{session.id3}"
|
|
/ f"{session.id2}"
|
|
)
|
|
file = path / "text.json"
|
|
idx = 0
|
|
text = ""
|
|
for f in os.listdir(path):
|
|
(path / f).unlink()
|
|
message[0].text = message[0].text.replace(command[0], "").strip()
|
|
for msg in message:
|
|
if isinstance(msg, alcText):
|
|
text += msg.text
|
|
elif isinstance(msg, Image):
|
|
if msg.url:
|
|
text += f"[image:{idx}]"
|
|
await AsyncHttpx.download_file(msg.url, path / f"{idx}.png")
|
|
idx += 1
|
|
else:
|
|
logger.debug("图片 URL 为空...", command[0])
|
|
if not file.exists():
|
|
file.parent.mkdir(exist_ok=True, parents=True)
|
|
is_at = "-at" in message
|
|
text = text.replace("-at", "")
|
|
json.dump(
|
|
{"at": is_at, "message": text},
|
|
file.open("w"),
|
|
ensure_ascii=False,
|
|
indent=4,
|
|
)
|
|
uni_msg = alcText("设置欢迎消息成功: \n") + message
|
|
await uni_msg.send()
|
|
logger.info(f"设置群欢迎消息成功: {text}", command[0], session=session)
|