mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
65 lines
2.3 KiB
Python
65 lines
2.3 KiB
Python
from PIL import ImageFont, ImageDraw, Image
|
||
import textwrap
|
||
from configs.path_config import IMAGE_PATH, TTF_PATH
|
||
from nonebot import on_command
|
||
from nonebot.typing import T_State
|
||
from nonebot.adapters.cqhttp import Bot, MessageEvent
|
||
from util.init_result import image
|
||
from services.log import logger
|
||
from util.utils import UserExistLimiter, get_message_text
|
||
from util.img_utils import pic2b64
|
||
|
||
_ulmt = UserExistLimiter()
|
||
|
||
|
||
luxun = on_command("鲁迅说过", aliases={"鲁迅说"})
|
||
|
||
|
||
@luxun.handle()
|
||
async def handle(bot: Bot, event: MessageEvent, state: T_State):
|
||
if _ulmt.check(event.user_id):
|
||
await luxun.finish('你的鲁迅正在说，等会', at_sender=True)
|
||
args = get_message_text(event.json())
|
||
if args:
|
||
state["content"] = args if args else '烦了，不说了'
|
||
|
||
|
||
@luxun.got("content", prompt="你让鲁迅说点啥?")
|
||
async def handle_event(bot: Bot, event: MessageEvent, state: T_State):
|
||
filename = str(event.user_id) + "_.jpg"
|
||
content = state["content"].strip()
|
||
if content.startswith(',') or content.startswith('，'):
|
||
content = content[1:]
|
||
_ulmt.set_True(event.user_id)
|
||
if len(content) > 20:
|
||
_ulmt.set_False(event.user_id)
|
||
await luxun.finish("太长了, 鲁迅说不完!", at_sender=True)
|
||
else:
|
||
if len(content) >= 12:
|
||
content = content[:12] + '\n' + content[12:]
|
||
img = image(b64=process_pic(content, filename))
|
||
logger.info(f"USER {event.user_id} GROUP "
|
||
f"{event.group_id if event.message_type != 'private' else 'private'} 鲁迅说过 {content}")
|
||
await luxun.send(img)
|
||
_ulmt.set_False(event.user_id)
|
||
|
||
|
||
def process_pic(content, filename) -> str:
|
||
text = content
|
||
para = textwrap.wrap(text, width=15)
|
||
MAX_W, MAX_H = 480, 280
|
||
bk_img = Image.open(IMAGE_PATH + "other/luxun.jpg")
|
||
font_path = TTF_PATH + "/msyh.ttf"
|
||
font = ImageFont.truetype(font_path, 37)
|
||
font2 = ImageFont.truetype(font_path, 30)
|
||
draw = ImageDraw.Draw(bk_img)
|
||
current_h, pad = 300, 10
|
||
for line in para:
|
||
w, h = draw.textsize(line, font=font)
|
||
draw.text(((MAX_W - w) / 2, current_h), line, font=font)
|
||
current_h += h + pad
|
||
draw.text((320, 400), "——鲁迅", font=font2, fill=(255, 255, 255))
|
||
return pic2b64(bk_img)
|
||
|
||
|