refactor code

plugins/admin_bot_manage/custom_welcome_message.py

@@ -0,0 +1,37 @@
+from nonebot import on_command
+from utils.utils import get_message_text, get_message_imgs
+from nonebot.typing import T_State
+from nonebot.adapters.cqhttp import Bot, GroupMessageEvent
+from .data_source import custom_group_welcome
+from nonebot.adapters.cqhttp.permission import GROUP
+from services.log import logger
+
+
+__plugin_name__ = "自定义进群欢迎消息"
+
+__plugin_usage__ = """
+    自定义进群欢迎消息 [消息] [图片](可省略)
+    示例：自定义进群欢迎消息 欢迎新人！[图片]
+"""
+
+
+custom_welcome = on_command(
+    "自定义进群欢迎消息",
+    aliases={"自定义欢迎消息", "自定义群欢迎消息", "设置群欢迎消息"},
+    permission=GROUP,
+    priority=5,
+    block=True,
+)
+
+
+@custom_welcome.handle()
+async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
+    msg = get_message_text(event.json())
+    imgs = get_message_imgs(event.json())
+    if not msg and not imgs:
+        await custom_welcome.finish(__plugin_usage__)
+    await custom_welcome.send(
+        await custom_group_welcome(msg, imgs, event.user_id, event.group_id),
+        at_sender=True,
+    )
+    logger.info(f"USER {event.user_id} GROUP {event.group_id} 自定义群欢迎消息：{msg}")