mirror of
https://github.com/Xtao-Labs/iShotaBot.git
synced 2024-11-16 04:35:55 +00:00
49 lines
1.7 KiB
Python
49 lines
1.7 KiB
Python
import re
|
|
from io import BytesIO
|
|
|
|
from pyrogram import Client, filters, ContinuePropagation
|
|
from pyrogram.types import Message
|
|
|
|
from PIL import Image
|
|
|
|
from defs.mihoyo_bbs import get_mihoyo_screenshot
|
|
from defs.button import gen_button, Button
|
|
|
|
|
|
@Client.on_message(
|
|
filters.incoming
|
|
& filters.text
|
|
& filters.regex(r"(https://)?(m\.)?www.miyoushe.com/.+/article/\d+")
|
|
)
|
|
async def bili_dynamic(_: Client, message: Message):
|
|
# sourcery skip: use-named-expression
|
|
try:
|
|
p = re.compile(r"(https://)?(m\.)?www.miyoushe.com/.+/article/\d+")
|
|
article = p.search(message.text)
|
|
if article:
|
|
article_url = article.group()
|
|
if not article_url.startswith(("https://", "http://")):
|
|
article_url = f"https://{article_url}"
|
|
image = await get_mihoyo_screenshot(article_url)
|
|
if image:
|
|
# 将bytes结果转化为字节流
|
|
photo = BytesIO(image)
|
|
photo.name = "screenshot.png"
|
|
pillow_photo = Image.open(BytesIO(image))
|
|
width, height = pillow_photo.size
|
|
if abs(height - width) > 1300:
|
|
await message.reply_document(
|
|
document=photo,
|
|
quote=True,
|
|
reply_markup=gen_button([Button(0, "Link", article_url)]),
|
|
)
|
|
else:
|
|
await message.reply_photo(
|
|
photo,
|
|
quote=True,
|
|
reply_markup=gen_button([Button(0, "Link", article_url)]),
|
|
)
|
|
except Exception as e:
|
|
print(f"截取米哈游帖子时发生错误：{e}")
|
|
raise ContinuePropagation
|