feat✨: 禁言检测

zhenxun/utils/image_utils.py

@@ -10,7 +10,9 @@ from imagehash import ImageHash
 from nonebot.utils import is_coroutine_callable
 from PIL import Image
 
-from zhenxun.configs.path_config import IMAGE_PATH
+from zhenxun.configs.path_config import IMAGE_PATH, TEMP_PATH
+from zhenxun.services.log import logger
+from zhenxun.utils.http_utils import AsyncHttpx
 
 from ._build_image import BuildImage, ColorAlias
 from ._build_mat import BuildMat, MatType
@@ -381,3 +383,24 @@ def get_img_hash(image_file: str | Path) -> str:
     with open(image_file, "rb") as fp:
         hash_value = imagehash.average_hash(Image.open(fp))
     return str(hash_value)
+
+
+async def get_download_image_hash(url: str, mark: str) -> str:
+    """下载图片获取哈希值
+
+    参数:
+        url: 图片url
+        mark: 随机标志符
+
+    返回:
+        str: 哈希值
+    """
+    try:
+        if await AsyncHttpx.download_file(
+            url, TEMP_PATH / f"compare_download_{mark}_img.jpg"
+        ):
+            img_hash = get_img_hash(TEMP_PATH / f"compare_download_{mark}_img.jpg")
+            return str(img_hash)
+    except Exception as e:
+        logger.warning(f"下载读取图片Hash出错", e=e)
+    return ""

zhenxun/utils/platform.py

@@ -50,6 +50,23 @@ class UserData(BaseModel):
 
 class PlatformUtils:
 
+    @classmethod
+    async def ban_user(cls, bot: Bot, user_id: str, group_id: str, duration: int):
+        """禁言
+
+        参数:
+            bot: Bot
+            user_id: 用户id
+            group_id: 群组id
+            duration: 禁言时长(分钟)
+        """
+        if isinstance(bot, v11Bot):
+            await bot.set_group_ban(
+                group_id=int(group_id),
+                user_id=int(user_id),
+                duration=duration * 60,
+            )
+
     @classmethod
     async def send_superuser(
         cls,