mirror of
https://github.com/xuthus83/LittlePaimon.git
synced 2024-10-21 16:27:15 +08:00
37 lines
1.4 KiB
Python
37 lines
1.4 KiB
Python
|
from aiohttp import ClientSession
|
||
|
|
from urllib.parse import quote
|
||
|
|
from nonebot import on_command
|
||
|
|
from nonebot.params import CommandArg
|
||
|
|
from nonebot.adapters.onebot.v11 import MessageEvent
|
||
|
|
from ..utils.util import FreqLimiter, get_id
|
||
|
|
from ..utils.config import config
|
||
|
|
|
||
|
|
couplets = on_command('对联', aliases={'对对联'}, priority=13, block=True)
|
||
|
|
|
||
|
|
couplets_limit = FreqLimiter(config.paimon_couplets_cd)
|
||
|
|
|
||
|
|
|
||
|
|
@couplets.handle()
|
||
|
|
async def couplets_handler(event: MessageEvent, msg=CommandArg()):
|
||
|
|
if not msg:
|
||
|
|
await couplets.finish('请输入对联内容')
|
||
|
|
if not couplets_limit.check(get_id(event)):
|
||
|
|
await couplets.finish(f'对联冷却ing(剩余{couplets_limit.left_time(get_id(event))}秒)')
|
||
|
|
else:
|
||
|
|
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
|
||
|
|
couplets_limit.start_cd(get_id(event), config.paimon_couplets_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 = ''
|
||
|
|
for n in range(0, num):
|
||
|
|
result += res['output'][n] + '\n'
|
||
|
|
await couplets.finish(f'上联：{word}\n下联：{result}')
|