新增原神壁纸

Paimon_Plugins/__init__.py

@@ -5,15 +5,18 @@ from nonebot.params import CommandArg
 from typing import Union
 from nonebot.adapters.onebot.v11 import GroupMessageEvent, MessageEvent, MessageSegment
 from ..utils.util import FreqLimiter
+from ..utils.config import config
 import random
 
 duilian = on_command('对联', aliases={'对对联'}, priority=15, block=True)
 cat_pic = on_command('猫图', aliases={'来点猫片', '看看猫猫', '来个猫猫'}, priority=15, block=True)
-tcy_pic = on_command('二次元图', aliases={'来点二次元', '来点二刺螈'}, priority=15, block=True)
+ecy_pic = on_command('二次元图', aliases={'来点二次元', '来点二刺螈'}, priority=15, block=True)
+ys_pic = on_command('原神壁纸', aliases={'来点原神图', '来点原神壁纸'}, priority=15, block=True)
 
-duilian_limit = FreqLimiter(5)
-cat_lmt = FreqLimiter(10)
-tcy_lmt = FreqLimiter(5)
+duilian_limit = FreqLimiter(config.paimon_duilian_cd)
+cat_lmt = FreqLimiter(config.paimon_cat_cd)
+ecy_lmt = FreqLimiter(config.paimon_ecy_cd)
+ys_lmt = FreqLimiter(config.paimon_ysp_cd)
 
 
 @duilian.handle()
@@ -23,14 +26,23 @@ async def duilian_handler(event: Union[GroupMessageEvent, MessageEvent], msg=Com
     if not duilian_limit.check(event.group_id or event.user_id):
         await duilian.finish(f'对联冷却ing(剩余{duilian_limit.left_time(event.group_id or event.user_id)}秒)')
     else:
-        duilian_limit.start_cd(event.group_id or event.user_id, 10)
-        text = quote(str(msg))
+        msg = str(msg).split(' ')
+        word = msg[0].strip()
+        try:
+            num = int(msg[1]) if len(msg) > 1 else 1
+        except:
+            num = 1
+        num = num if num < 10 else 10
+        duilian_limit.start_cd(event.group_id or event.user_id, config.paimon_duilian_cd)
+        text = quote(str(word))
         url = f'https://ai-backend.binwang.me/v0.2/couplet/{text}'
         async with ClientSession() as session:
             res = await session.get(url)
             res = await res.json()
-            result = res['output'][0]
-            await duilian.finish(f'上联：{msg}\n下联：{result}')
+            result = ''
+            for n in range(0, num):
+                result += res['output'][n] + '\n'
+            await duilian.finish(f'上联：{word}\n下联：{result}')
 
 
 @cat_pic.handle()
@@ -38,7 +50,7 @@ async def cat_pic_handler(event: Union[GroupMessageEvent, MessageEvent]):
     if not cat_lmt.check(event.group_id or event.user_id):
         await cat_pic.finish(f'猫片冷却ing(剩余{cat_lmt.left_time(event.group_id or event.user_id)}秒)')
     else:
-        cat_lmt.start_cd(event.group_id or event.user_id, 10)
+        cat_lmt.start_cd(event.group_id or event.user_id, config.paimon_cat_cd)
         url = 'http://edgecats.net/'
         async with ClientSession() as session:
             res = await session.get(url)
@@ -46,18 +58,36 @@ async def cat_pic_handler(event: Union[GroupMessageEvent, MessageEvent]):
             await cat_pic.finish(MessageSegment.image(res))
 
 
-@tcy_pic.handle()
-async def tcy_pic_handler(event: Union[GroupMessageEvent, MessageEvent]):
+@ecy_pic.handle()
+async def ecy_pic_handler(event: Union[GroupMessageEvent, MessageEvent]):
     urls = [
         'https://www.dmoe.cc/random.php',
         'https://acg.toubiec.cn/random.php',
         'https://api.ixiaowai.cn/api/api.php'
     ]
-    if not tcy_lmt.check(event.group_id or event.user_id):
-        await tcy_pic.finish(f'二次元图片冷却ing(剩余{tcy_lmt.left_time(event.group_id or event.user_id)}秒)')
+    if not ecy_lmt.check(event.group_id or event.user_id):
+        await ecy_pic.finish(f'二次元图片冷却ing(剩余{ecy_lmt.left_time(event.group_id or event.user_id)}秒)')
     else:
-        tcy_lmt.start_cd(event.group_id or event.user_id, 5)
+        ecy_lmt.start_cd(event.group_id or event.user_id, config.paimon_ecy_cd)
         async with ClientSession() as session:
             res = await session.get(random.choice(urls))
             res = await res.read()
             await cat_pic.finish(MessageSegment.image(res))
+
+
+@ys_pic.handle()
+async def ys_pic_handler(event: Union[GroupMessageEvent, MessageEvent]):
+    urls = [
+        'https://api.r10086.com/img-api.php?type=%E5%8E%9F%E7%A5%9E%E6%A8%AA%E5%B1%8F%E7%B3%BB%E5%88%971',
+        'https://api.r10086.com/img-api.php?type=%E5%8E%9F%E7%A5%9E%E7%AB%96%E5%B1%8F%E7%B3%BB%E5%88%971',
+        'https://api.dujin.org/pic/yuanshen/',
+        'https://api.dreamofice.cn/random-v0/img.php?game=ys'
+    ]
+    if not ys_lmt.check(event.group_id or event.user_id):
+        await ys_pic.finish(f'原神壁纸冷却ing(剩余{ys_lmt.left_time(event.group_id or event.user_id)}秒)')
+    else:
+        ys_lmt.start_cd(event.group_id or event.user_id, config.paimon_ysp_cd)
+        async with ClientSession() as session:
+            res = await session.get(random.choice(urls))
+            res = await res.read()
+            await ys_pic.finish(MessageSegment.image(res))

README.md

@@ -10,6 +10,7 @@ nonebot2版本的小派蒙
   * 对联
   * 来点猫片
   * 来点二次元
+  * 来点原神壁纸
 
 具体插件命令与hoshino版一致<br>
 nonebot2刚上手，测试时间较短，如有bug，请及时反馈。

utils/config.py

@@ -11,6 +11,10 @@ class PluginConfig(BaseModel):
     paimon_remind_limit: int = 3
     paimon_sign_hour: int = 0
     paimon_sign_minute: int = 0
+    paimon_duilian_cd: int = 6
+    paimon_cat_cd: int = 12
+    paimon_ecy_cd: int = 6
+    paimon_ysp_cd: int = 10
 
 
 driver = get_driver()