mirror of
https://github.com/Xtao-Labs/iShotaBot.git
synced 2024-11-23 16:27:53 +00:00
89 lines
2.9 KiB
Python
89 lines
2.9 KiB
Python
import re
|
|
from io import BytesIO
|
|
|
|
from bilibili_api import ResponseCodeException
|
|
from pyrogram import Client, filters, ContinuePropagation
|
|
from pyrogram.types import Message
|
|
|
|
from defs.bilibili import (
|
|
b23_extract,
|
|
video_info_get,
|
|
binfo_image_create,
|
|
get_dynamic_screenshot_pc,
|
|
check_and_refresh_credential,
|
|
)
|
|
from defs.button import gen_button, Button
|
|
from defs.glover import bili_auth_user, bili_auth_chat
|
|
from init import bot
|
|
from scheduler import scheduler, add_delete_message_job
|
|
|
|
|
|
@bot.on_message(
|
|
filters.incoming
|
|
& filters.text
|
|
& filters.regex(r"av(\d{1,12})|BV(\w{10})|b23.tv")
|
|
& ~(
|
|
filters.command(["download", "bilibili_fav"])
|
|
& (filters.user(bili_auth_user) | filters.chat(bili_auth_chat))
|
|
)
|
|
)
|
|
async def bili_resolve(_: Client, message: Message):
|
|
"""
|
|
解析 bilibili 链接
|
|
"""
|
|
if "b23.tv" in message.text:
|
|
message.text = await b23_extract(message.text)
|
|
p = re.compile(r"av(\d{1,12})|BV(\w{10})|b23.tv")
|
|
video_number = p.search(message.text)
|
|
if video_number:
|
|
video_number = video_number[0]
|
|
try:
|
|
video_info = await video_info_get(video_number) if video_number else None
|
|
except ResponseCodeException as e:
|
|
text = f"视频解析失败：[{e.code}] {e.msg}"
|
|
reply = await message.reply(text, quote=True)
|
|
add_delete_message_job(reply)
|
|
raise ContinuePropagation
|
|
if video_info:
|
|
image = await binfo_image_create(video_info)
|
|
buttons = [Button(0, "Link", "https://b23.tv/" + video_info["bvid"])]
|
|
if (message.from_user and message.from_user.id in bili_auth_user) or (
|
|
message.chat and message.chat.id in bili_auth_chat
|
|
):
|
|
buttons.append(Button(1, "Download", "download_" + video_info["bvid"]))
|
|
await message.reply_photo(
|
|
image,
|
|
quote=True,
|
|
reply_markup=gen_button(buttons),
|
|
)
|
|
raise ContinuePropagation
|
|
|
|
|
|
@bot.on_message(
|
|
filters.incoming & filters.text & filters.regex(r"t.bilibili.com/([0-9]*)")
|
|
)
|
|
async def bili_dynamic(_: Client, message: Message):
|
|
# sourcery skip: use-named-expression
|
|
p = re.compile(r"t.bilibili.com/([0-9]*)")
|
|
dynamic_number = p.search(message.text)
|
|
if dynamic_number:
|
|
dynamic_id = dynamic_number[1]
|
|
image = await get_dynamic_screenshot_pc(dynamic_id)
|
|
if image:
|
|
# 将bytes结果转化为字节流
|
|
photo = BytesIO(image)
|
|
photo.name = "screenshot.png"
|
|
await message.reply_photo(
|
|
photo,
|
|
quote=True,
|
|
reply_markup=gen_button(
|
|
[Button(0, "Link", f"https://t.bilibili.com/{dynamic_id}")]
|
|
),
|
|
)
|
|
raise ContinuePropagation
|
|
|
|
|
|
@scheduler.scheduled_job("interval", hours=1, id="bili_cookie_refresh")
|
|
async def bili_cookie_refresh():
|
|
await check_and_refresh_credential()
|