mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 06:12:53 +08:00
89 lines
3.0 KiB
Python
89 lines
3.0 KiB
Python
import aiohttp
|
||
from io import BytesIO
|
||
from random import choice
|
||
from nonebot import on_regex
|
||
from nonebot.typing import T_State
|
||
from nonebot.adapters.cqhttp import Bot, GroupMessageEvent
|
||
from utils.utils import get_message_text, get_message_at
|
||
from utils.message_builder import image
|
||
import re
|
||
from utils.image_utils import CreateImg
|
||
from asyncio.exceptions import TimeoutError
|
||
|
||
__zx_plugin_name__ = "我有一个朋友"
|
||
__plugin_usage__ = """
|
||
usage：
|
||
我有一个朋友他...，不知道是不是你
|
||
指令：
|
||
我有一个朋友想问问 [文本] ?[at]: 当at时你的朋友就是艾特对象
|
||
""".strip()
|
||
__plugin_des__ = "我有一个朋友想问问..."
|
||
__plugin_cmd__ = ["我有一个朋友想问问[文本] ?[at]"]
|
||
__plugin_version__ = 0.1
|
||
__plugin_author__ = "HibiKier"
|
||
__plugin_settings__ = {
|
||
"level": 5,
|
||
"default_status": True,
|
||
"limit_superuser": False,
|
||
"cmd": ["我有一个朋友想问问", "我有一个朋友"],
|
||
}
|
||
|
||
one_friend = on_regex(
|
||
"^我.*?朋友.*?(想问问|说|让我问问|想问|让我问|想知道|让我帮他问问|让我帮他问|让我帮忙问|让我帮忙问问|问).*",
|
||
priority=4,
|
||
block=True,
|
||
)
|
||
|
||
|
||
async def get_pic(qq):
|
||
url = f"http://q1.qlogo.cn/g?b=qq&nk={qq}&s=100"
|
||
async with aiohttp.ClientSession() as session:
|
||
for _ in range(3):
|
||
try:
|
||
async with session.get(url, timeout=5) as response:
|
||
return await response.read()
|
||
except TimeoutError:
|
||
pass
|
||
return None
|
||
|
||
|
||
@one_friend.handle()
|
||
async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
|
||
msg = get_message_text(event.json())
|
||
qq = get_message_at(event.json())
|
||
if not qq:
|
||
qq = choice(
|
||
[
|
||
x["user_id"]
|
||
for x in await bot.get_group_member_list(
|
||
self_id=event.self_id, group_id=event.group_id
|
||
)
|
||
]
|
||
)
|
||
user_name = "朋友"
|
||
else:
|
||
qq = qq[0]
|
||
at_user = await bot.get_group_member_info(group_id=event.group_id, user_id=qq)
|
||
user_name = at_user["card"] if at_user["card"] else at_user["nickname"]
|
||
msg = re.search(
|
||
r"^我.*?朋友.*?(想问问|说|让我问问|想问|让我问|想知道|让我帮他问问|让我帮他问|让我帮忙问|让我帮忙问问|问)(.*)", msg
|
||
)
|
||
msg = msg.group(2)
|
||
if not msg:
|
||
msg = "都不知道问什么"
|
||
msg = msg.replace("他", "我").replace("她", "我").replace("它", "我")
|
||
x = await get_pic(qq)
|
||
if x:
|
||
ava = CreateImg(100, 100, background=BytesIO(await get_pic(qq)))
|
||
else:
|
||
ava = CreateImg(100, 100, color=(0, 0, 0))
|
||
ava.circle()
|
||
text = CreateImg(300, 30, font_size=30)
|
||
text.text((0, 0), user_name)
|
||
A = CreateImg(700, 150, font_size=25, color="white")
|
||
A.paste(ava, (30, 25), True)
|
||
A.paste(text, (150, 38))
|
||
A.text((150, 85), msg, (125, 125, 125))
|
||
|
||
await one_friend.send(image(b64=A.pic2bs4()))
|