[feature] Add batch send functionality for anime images; introduce ne… (#9)

* [feature] Add batch send functionality for anime images; introduce new cat image plugin

---------

Co-authored-by: jerrita <[email protected]>

Author: xunxun2001

plugins/example/anime_picture.py

@@ -1,11 +1,14 @@
 import json
+import re
+from typing import Optional
 
 import aiohttp
+import chinese2digits as c2d
 
 from anon import Storage
 from anon.event import MessageEvent
 from anon.logger import logger
-from anon.message import Image, ImageCategory
+from anon.message import Image
 from anon.plugin import PluginManager
 
 pm = PluginManager()
@@ -15,12 +18,13 @@
         "r18": 0,
         "ai": 0,
         "level": "0-2",
-        "taste": "1,2"
+        "taste": "1,2",
     }
 
 
-async def fetch_and_save_image(tag: list):
+async def fetch_and_save_image(tag: list, num: int = 1) -> Optional[list]:
     params = storage['param'].copy()
+    params["num"] = num if 1 < num <= 5 else 1
     if tag:
         params["tag"] = tag
     logger.info(f"tag:{tag},params:{params}")
@@ -34,9 +38,12 @@ async def fetch_and_save_image(tag: list):
         if not datas["data"]:
             return None
 
-        img_url = datas["data"][0]["urls"]["original"]
-        img_url = img_url.replace('i.pixiv.cat', 'i.pixiv.re')
-        return img_url
+        img_url_list = []
+        for i_url in range(num):
+            img_url = datas["data"][i_url]["urls"]["original"]
+            img_url = img_url.replace('i.pixiv.cat', 'i.pixiv.re')
+            img_url_list.append(img_url)
+        return img_url_list
 
 
 @pm.register_event([MessageEvent])
@@ -59,4 +66,23 @@ async def handle_message(event: MessageEvent):
         if img_url is None:
             await event.reply("图库查找无结果", quote=True)
         else:
-            await event.reply(Image(img_url), quote=False)
+            await event.reply(Image(img_url[0]), quote=False)
+    elif parts:
+        match = re.match(r'来(\d+|零|两|一|二|三|四|五|六|七|八|九)(张|份)(色图|涩图)', parts[0])
+        if match:
+            quantity = match.group(1)
+            try:
+                num = int(quantity)
+            except:
+                num = int(c2d.chineseToDigits(quantity))
+            tag = parts[1:] if len(parts) > 1 else None
+            logger.info(f"数量: {num}, 标签: {tag}")
+            img_url = await fetch_and_save_image(tag, num)
+            if img_url is None:
+                await event.reply("图库查找无结果", quote=True)
+            else:
+                index = 1
+                for i_url in img_url:
+                    logger.info(f"第{index}份色图发送中")
+                    index += 1
+                    await event.reply(Image(i_url), quote=False)

plugins/example/cat.py

@@ -0,0 +1,50 @@
+import re
+
+import aiohttp
+
+from anon.event import MessageEvent
+from anon.logger import logger
+from anon.message import Image
+from anon.plugin import PluginManager
+
+pm = PluginManager()
+
+
+async def fetch_and_save_image(limit: bool = False) -> list:
+    params = {}
+    num = 10 if limit else 1
+    params["limit"] = num
+
+    async with aiohttp.ClientSession() as session:
+        async with session.get("https://api.thecatapi.com/v1/images/search", params=params) as response:
+            logger.info(f"params:{params}")
+            datas = await response.json()
+        logger.info(datas)
+
+        img_url_list = []
+        for i_url in range(num):
+            img_url = datas[i_url]["url"]
+            img_url_list.append(img_url)
+        return img_url_list
+
+
+@pm.register_event([MessageEvent])
+async def cat(event: MessageEvent):
+    message = event.msg.text()
+    if re.fullmatch(r'喵+', message):
+        # 计算“喵”的数量
+        count = len(message) // len('喵')
+        if count == 1:
+            img_url = await fetch_and_save_image(False)
+            logger.info('喵')
+        elif count > 1:
+            img_url = await fetch_and_save_image(True)
+            await event.reply("不管多少个喵，只要大于1个都会返回10张图的喵~", quote=True)
+            logger.info(f'收到了{count}个喵！')
+        else:
+            return
+        index = 1
+        for i_url in img_url:
+            logger.info(f"第{index}份猫图发送中")
+            index += 1
+            await event.reply(Image(i_url), quote=False)

requirements.txt

@@ -4,6 +4,8 @@ aiocron~=1.8
 
 # Example plugins dependencies
 pytz~=2024.1
+aiohttp~=3.9.3
+chinese2digits~=6.0.2
 
 # Extras
-aiohttp~=3.9.3
+zhipuai~=2.0.1