update code

plugins/group_handle/__init__.py

@@ -1,8 +1,6 @@
 from nonebot import on_notice, on_request
 from configs.path_config import IMAGE_PATH, DATA_PATH
 from utils.message_builder import image
-import os
-import random
 from models.group_member_info import GroupInfoUser
 from datetime import datetime
 from services.log import logger
@@ -13,9 +11,12 @@ from nonebot.adapters.cqhttp import (
     GroupDecreaseNoticeEvent,
 )
 from nonebot.adapters.cqhttp.exception import ActionFailed
-from pathlib import Path
+from configs.config import plugins2info_dict
+from utils.static_data import group_manager
 from models.group_info import GroupInfo
-
+from pathlib import Path
+import random
+import os
 try:
     import ujson as json
 except ModuleNotFoundError:
@@ -37,46 +38,52 @@ add_group = on_request(priority=1, block=False)
 
 @group_increase_handle.handle()
 async def _(bot: Bot, event: GroupIncreaseNoticeEvent, state: dict):
-    join_time = datetime.now()
-    user_info = await bot.get_group_member_info(
-        group_id=event.group_id, user_id=event.user_id
-    )
-    if await GroupInfoUser.add_member_info(
-        user_info["user_id"],
-        user_info["group_id"],
-        user_info["nickname"],
-        join_time,
-    ):
-        logger.info(f"用户{user_info['user_id']} 所属{user_info['group_id']} 更新成功")
+    if event.user_id == int(bot.self_id):
+        if event.group_id not in group_manager['group_manager'].keys():
+            for plugin in plugins2info_dict.keys():
+                if not plugins2info_dict[plugin]['default_status']:
+                    group_manager.block_plugin(plugin, str(event.group_id))
     else:
-        logger.info(f"用户{user_info['user_id']} 所属{user_info['group_id']} 更新失败")
-
-    # 群欢迎消息
-    if await GroupRemind.get_status(event.group_id, "hy"):
-        msg = ""
-        img = ""
-        at_flag = False
-        custom_welcome_msg_json = (
-            Path() / "data" / "custom_welcome_msg" / "custom_welcome_msg.json"
+        join_time = datetime.now()
+        user_info = await bot.get_group_member_info(
+            group_id=event.group_id, user_id=event.user_id
         )
-        if custom_welcome_msg_json.exists():
-            data = json.load(open(custom_welcome_msg_json, "r"))
-            if data.get(str(event.group_id)):
-                msg = data[str(event.group_id)]
-                if msg.find("[at]") != -1:
-                    msg = msg.replace("[at]", "")
-                    at_flag = True
-        if os.path.exists(DATA_PATH + f"custom_welcome_msg/{event.group_id}.jpg"):
-            img = image(abspath=DATA_PATH + f"custom_welcome_msg/{event.group_id}.jpg")
-        if msg or img:
-            await group_increase_handle.send(
-                "\n" + msg.strip() + img, at_sender=at_flag
-            )
+        if await GroupInfoUser.add_member_info(
+            user_info["user_id"],
+            user_info["group_id"],
+            user_info["nickname"],
+            join_time,
+        ):
+            logger.info(f"用户{user_info['user_id']} 所属{user_info['group_id']} 更新成功")
         else:
-            await group_increase_handle.send(
-                "新人快跑啊！！本群现状↓（快使用自定义！）"
-                + image(random.choice(os.listdir(IMAGE_PATH + "qxz/")), "qxz")
+            logger.info(f"用户{user_info['user_id']} 所属{user_info['group_id']} 更新失败")
+
+        # 群欢迎消息
+        if await GroupRemind.get_status(event.group_id, "hy"):
+            msg = ""
+            img = ""
+            at_flag = False
+            custom_welcome_msg_json = (
+                Path() / "data" / "custom_welcome_msg" / "custom_welcome_msg.json"
             )
+            if custom_welcome_msg_json.exists():
+                data = json.load(open(custom_welcome_msg_json, "r"))
+                if data.get(str(event.group_id)):
+                    msg = data[str(event.group_id)]
+                    if msg.find("[at]") != -1:
+                        msg = msg.replace("[at]", "")
+                        at_flag = True
+            if os.path.exists(DATA_PATH + f"custom_welcome_msg/{event.group_id}.jpg"):
+                img = image(abspath=DATA_PATH + f"custom_welcome_msg/{event.group_id}.jpg")
+            if msg or img:
+                await group_increase_handle.send(
+                    "\n" + msg.strip() + img, at_sender=at_flag
+                )
+            else:
+                await group_increase_handle.send(
+                    "新人快跑啊！！本群现状↓（快使用自定义！）"
+                    + image(random.choice(os.listdir(IMAGE_PATH + "qxz/")), "qxz")
+                )
 
 
 @group_decrease_handle.handle()