mirror of
https://github.com/PaiGramTeam/PaiGram.git
synced 2024-12-28 03:58:35 +00:00
594 lines
28 KiB
Python
594 lines
28 KiB
Python
import math
|
||
import os
|
||
import re
|
||
from asyncio import create_subprocess_shell, subprocess
|
||
from functools import partial
|
||
from typing import List, Optional, Tuple, TYPE_CHECKING, Union
|
||
|
||
import aiofiles
|
||
from arkowrapper import ArkoWrapper
|
||
from bs4 import BeautifulSoup
|
||
from httpx import Timeout
|
||
from telegram import (
|
||
InlineKeyboardButton,
|
||
InlineKeyboardMarkup,
|
||
InputMediaPhoto,
|
||
ReplyKeyboardMarkup,
|
||
ReplyKeyboardRemove,
|
||
InputMediaDocument,
|
||
InputMediaVideo,
|
||
)
|
||
from telegram.constants import MessageLimit, ParseMode
|
||
from telegram.error import BadRequest
|
||
from telegram.ext import ConversationHandler, filters
|
||
from telegram.helpers import escape_markdown
|
||
|
||
from core.config import config
|
||
from core.plugin import Plugin, conversation, handler
|
||
from gram_core.basemodel import Settings, SettingsConfigDict
|
||
from gram_core.dependence.redisdb import RedisDB
|
||
from modules.apihelper.client.components.hoyolab import Hoyolab
|
||
from modules.apihelper.client.components.hyperion import Hyperion, HyperionBase
|
||
from modules.apihelper.error import APIHelperException
|
||
from modules.apihelper.models.genshin.hyperion import PostTypeEnum
|
||
from utils.helpers import sha1
|
||
from utils.log import logger
|
||
|
||
if TYPE_CHECKING:
|
||
from bs4 import Tag
|
||
from telegram import Update, Message
|
||
from telegram.ext import ContextTypes
|
||
from modules.apihelper.models.genshin.hyperion import ArtworkImage
|
||
|
||
|
||
class PostHandlerData:
|
||
def __init__(self):
|
||
self.post_text: str = ""
|
||
self.post_images: Optional[List["ArtworkImage"]] = None
|
||
self.delete_photo: Optional[List[int]] = []
|
||
self.channel_id: int = -1
|
||
self.tags: Optional[List[str]] = []
|
||
|
||
|
||
class PostConfig(Settings):
|
||
"""文章推送配置"""
|
||
|
||
chat_id: Optional[int] = 0
|
||
|
||
model_config = SettingsConfigDict(env_prefix="post_")
|
||
|
||
|
||
CHECK_POST, SEND_POST, CHECK_COMMAND, GTE_DELETE_PHOTO = range(10900, 10904)
|
||
GET_POST_CHANNEL, GET_TAGS, GET_TEXT = range(10904, 10907)
|
||
post_config = PostConfig()
|
||
|
||
|
||
class Post(Plugin.Conversation):
|
||
"""文章推送"""
|
||
|
||
MENU_KEYBOARD = ReplyKeyboardMarkup([["推送频道", "添加TAG"], ["编辑文字", "删除图片"], ["退出"]], True, True)
|
||
|
||
def __init__(self, redis: RedisDB):
|
||
self.gids = 2
|
||
self.short_name = "ys"
|
||
self.ffmpeg_enable = False
|
||
self.cache_dir = os.path.join(os.getcwd(), "cache")
|
||
self.cache = redis.client
|
||
self.cache_key = "plugin:post:pushed"
|
||
|
||
def get_cache_key(self, bbs_type: "PostTypeEnum") -> str:
|
||
return f"{self.cache_key}:{bbs_type.value}"
|
||
|
||
async def is_posted(self, bbs_type: "PostTypeEnum", post_id: int) -> bool:
|
||
key = self.get_cache_key(bbs_type)
|
||
return await self.cache.sismember(key, post_id)
|
||
|
||
async def set_posted(self, bbs_type: "PostTypeEnum", post_id: int) -> bool:
|
||
key = self.get_cache_key(bbs_type)
|
||
return await self.cache.sadd(key, post_id)
|
||
|
||
async def is_posted_empty(self, bbs_type: "PostTypeEnum") -> bool:
|
||
key = self.get_cache_key(bbs_type)
|
||
return await self.cache.scard(key) == 0
|
||
|
||
@staticmethod
|
||
def get_bbs_client(bbs_type: "PostTypeEnum") -> "HyperionBase":
|
||
class_type = Hyperion if bbs_type == PostTypeEnum.CN else Hoyolab
|
||
return class_type(
|
||
timeout=Timeout(
|
||
connect=config.connect_timeout,
|
||
read=config.read_timeout,
|
||
write=config.write_timeout,
|
||
pool=config.pool_timeout,
|
||
)
|
||
)
|
||
|
||
async def initialize(self):
|
||
if config.channels and len(config.channels) > 0:
|
||
logger.success("文章定时推送处理已经开启")
|
||
cn_task = partial(self.task, post_type=PostTypeEnum.CN)
|
||
os_task = partial(self.task, post_type=PostTypeEnum.OS)
|
||
self.application.job_queue.run_repeating(cn_task, 30, name="post_cn_task")
|
||
self.application.job_queue.run_repeating(os_task, 30, name="post_os_task")
|
||
logger.success("文章定时推送处理已经开启")
|
||
output, _ = await self.execute("ffmpeg -version")
|
||
if "ffmpeg version" in output:
|
||
self.ffmpeg_enable = True
|
||
logger.info("检测到 ffmpeg 可用 已经启动编码转换")
|
||
logger.debug("ffmpeg version info\n%s", output)
|
||
else:
|
||
logger.warning("ffmpeg 不可用 已经禁用编码转换")
|
||
|
||
async def task(self, context: "ContextTypes.DEFAULT_TYPE", post_type: "PostTypeEnum"):
|
||
bbs = self.get_bbs_client(post_type)
|
||
|
||
# 请求推荐POST列表并处理
|
||
try:
|
||
official_recommended_posts = await bbs.get_official_recommended_posts(self.gids)
|
||
except APIHelperException as exc:
|
||
logger.error("获取首页推荐信息失败 %s", str(exc))
|
||
return
|
||
|
||
temp_post_id_list = [post.post_id for post in official_recommended_posts]
|
||
|
||
# 判断是否为空
|
||
if await self.is_posted_empty(post_type):
|
||
for temp_list in temp_post_id_list:
|
||
await self.set_posted(post_type, temp_list)
|
||
return
|
||
|
||
# 筛选出新推送的文章
|
||
new_post_id_list = [post_id for post_id in temp_post_id_list if not await self.is_posted(post_type, post_id)]
|
||
if not new_post_id_list:
|
||
return
|
||
|
||
chat_id = post_config.chat_id or config.owner
|
||
|
||
for post_id in new_post_id_list:
|
||
try:
|
||
post_info = await bbs.get_post_info(self.gids, post_id)
|
||
except APIHelperException as exc:
|
||
logger.error("获取文章信息失败 %s", str(exc))
|
||
text = f"获取 post_id[{post_id}] 文章信息失败 {str(exc)}"
|
||
try:
|
||
await context.bot.send_message(chat_id, text)
|
||
except BadRequest as _exc:
|
||
logger.error("发送消息失败 %s", _exc.message)
|
||
return
|
||
type_name = post_info.type_enum.value
|
||
buttons = [
|
||
[
|
||
InlineKeyboardButton("确认", callback_data=f"post_admin|confirm|{type_name}|{post_info.post_id}"),
|
||
InlineKeyboardButton("取消", callback_data=f"post_admin|cancel|{type_name}|{post_info.post_id}"),
|
||
]
|
||
]
|
||
url = post_info.get_fix_url(self.short_name)
|
||
tag = f"#{self.short_name} #{post_type.value} #{self.short_name}_{post_type.value}"
|
||
text = f"发现官网推荐文章 <a href='{url}'>{post_info.subject}</a>\n是否开始处理 {tag}"
|
||
try:
|
||
await context.bot.send_message(
|
||
chat_id,
|
||
text,
|
||
parse_mode=ParseMode.HTML,
|
||
reply_markup=InlineKeyboardMarkup(buttons),
|
||
)
|
||
await self.set_posted(post_type, post_id)
|
||
except BadRequest as exc:
|
||
logger.error("发送消息失败 %s", exc.message)
|
||
await bbs.close()
|
||
|
||
@staticmethod
|
||
def parse_post_text(soup: BeautifulSoup, post_subject: str) -> Tuple[str, bool]:
|
||
def parse_tag(_tag: "Tag") -> str:
|
||
if _tag.name == "a":
|
||
href = _tag.get("href")
|
||
if href and href.startswith("/"):
|
||
href = f"https://www.miyoushe.com{href}"
|
||
if href and href.startswith("http"):
|
||
return f"[{escape_markdown(_tag.get_text(), version=2)}]({href})"
|
||
return escape_markdown(_tag.get_text(), version=2)
|
||
|
||
post_text = f"*{escape_markdown(post_subject, version=2)}*\n\n"
|
||
start = True
|
||
too_long = False
|
||
if post_p := soup.find_all("p"):
|
||
try:
|
||
for p in post_p:
|
||
t = p.get_text()
|
||
if not t and start:
|
||
continue
|
||
start = False
|
||
for tag in p.contents:
|
||
post_text_ = post_text + parse_tag(tag)
|
||
if len(post_text_) >= (MessageLimit.CAPTION_LENGTH - 55):
|
||
raise RecursionError
|
||
post_text = post_text_
|
||
post_text += "\n"
|
||
except RecursionError:
|
||
too_long = True
|
||
else:
|
||
post_text += f"{escape_markdown(soup.get_text(), version=2)}\n"
|
||
return post_text.strip(), too_long
|
||
|
||
@staticmethod
|
||
def safe_cut(text: str, length: int) -> str:
|
||
text = text[:length]
|
||
right_pattern = r"\[.*?\]\(.*?\)"
|
||
error_pattern = r"\[.*?\]"
|
||
right_length = re.findall(right_pattern, text)
|
||
error_length = re.findall(error_pattern, text)
|
||
if right_length == error_length:
|
||
return text
|
||
error_index = text.rindex(error_length[-1])
|
||
return text[:error_index]
|
||
|
||
@staticmethod
|
||
def input_media(
|
||
media: "ArtworkImage", *args, **kwargs
|
||
) -> Union[None, InputMediaDocument, InputMediaPhoto, InputMediaVideo]:
|
||
file_extension = media.file_extension
|
||
filename = media.file_name
|
||
doc = None
|
||
if file_extension is not None:
|
||
if file_extension in {"jpg", "jpeg", "png", "webp"}:
|
||
doc = InputMediaPhoto(media.data, *args, **kwargs)
|
||
if file_extension in {"gif", "mp4", "mov", "avi", "mkv", "webm", "flv"}:
|
||
doc = InputMediaVideo(media.data, filename=filename, *args, **kwargs)
|
||
if not doc:
|
||
doc = InputMediaDocument(media.data, *args, **kwargs)
|
||
doc._frozen = False
|
||
return doc
|
||
|
||
@staticmethod
|
||
async def execute(command: str) -> Tuple[str, int]:
|
||
process = await create_subprocess_shell(
|
||
command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, stdin=subprocess.PIPE
|
||
)
|
||
stdout, stderr = await process.communicate()
|
||
try:
|
||
result = str(stdout.decode().strip()) + str(stderr.decode().strip())
|
||
except UnicodeDecodeError:
|
||
result = str(stdout.decode("gbk").strip()) + str(stderr.decode("gbk").strip())
|
||
return result, process.returncode
|
||
|
||
@staticmethod
|
||
def get_ffmpeg_command(input_file: str, output_file: str):
|
||
return (
|
||
f'ffmpeg -i "{input_file}" '
|
||
f'-c:v libx264 -crf 20 -vf "fps=30,format=yuv420p,'
|
||
f'scale=trunc(iw/2)*2:trunc(ih/2)*2" -y "{output_file}"'
|
||
)
|
||
|
||
async def gif_to_mp4(self, media: "List[ArtworkImage]"):
|
||
if self.ffmpeg_enable:
|
||
for i in media:
|
||
if i.file_extension == "gif":
|
||
file_path = os.path.join(self.cache_dir, i.file_name)
|
||
file_name, _ = os.path.splitext(i.file_name)
|
||
output_file = file_name + ".mp4"
|
||
output_path = os.path.join(self.cache_dir, output_file)
|
||
if os.path.exists(output_path):
|
||
async with aiofiles.open(output_path, mode="rb") as f:
|
||
i.data = await f.read()
|
||
i.file_name = output_file
|
||
i.file_extension = "mp4"
|
||
continue
|
||
async with aiofiles.open(file_path, mode="wb") as f:
|
||
await f.write(i.data)
|
||
temp_file = sha1(file_name) + ".mp4"
|
||
temp_path = os.path.join(self.cache_dir, temp_file)
|
||
command = self.get_ffmpeg_command(file_path, temp_path)
|
||
result, return_code = await self.execute(command)
|
||
if return_code == 0:
|
||
if os.path.exists(temp_path):
|
||
logger.debug("ffmpeg 执行成功\n%s", result)
|
||
os.rename(temp_path, output_path)
|
||
async with aiofiles.open(output_path, mode="rb") as f:
|
||
i.data = await f.read()
|
||
i.file_name = output_file
|
||
i.file_extension = "mp4"
|
||
else:
|
||
logger.error(
|
||
"输出文件不存在！可能是 ffmpeg 命令执行失败！\n"
|
||
"file_path[%s]\noutput_path[%s]\ntemp_file[%s]\nffmpeg result[%s]",
|
||
file_path,
|
||
output_path,
|
||
temp_path,
|
||
result,
|
||
)
|
||
else:
|
||
logger.error("ffmpeg 执行失败\n%s", result)
|
||
return media
|
||
|
||
@conversation.entry_point
|
||
@handler.callback_query(pattern=r"^post_admin\|", block=False)
|
||
async def callback_query_start(self, update: "Update", context: "ContextTypes.DEFAULT_TYPE") -> int:
|
||
post_handler_data = context.chat_data.get("post_handler_data")
|
||
if post_handler_data is None:
|
||
post_handler_data = PostHandlerData()
|
||
context.chat_data["post_handler_data"] = post_handler_data
|
||
callback_query = update.callback_query
|
||
user = callback_query.from_user
|
||
message = callback_query.message
|
||
logger.info("用户 %s[%s] POST命令请求", user.full_name, user.id)
|
||
|
||
async def get_post_admin_callback(callback_query_data: str) -> Tuple[str, PostTypeEnum, int]:
|
||
_data = callback_query_data.split("|")
|
||
_result = _data[1]
|
||
_post_type = PostTypeEnum(_data[2])
|
||
_post_id = int(_data[3])
|
||
logger.debug(
|
||
"callback_query_data函数返回 result[%s] _post_type[%s] post_id[%s]", _result, _post_type, _post_id
|
||
)
|
||
return _result, _post_type, _post_id
|
||
|
||
result, post_type, post_id = await get_post_admin_callback(callback_query.data)
|
||
|
||
if result == "cancel":
|
||
await message.reply_text("操作已经取消")
|
||
await message.delete()
|
||
elif result == "confirm":
|
||
reply_text = await message.reply_text("正在处理")
|
||
status = await self.send_post_info(post_handler_data, message, post_id, post_type)
|
||
await reply_text.delete()
|
||
return status
|
||
|
||
return ConversationHandler.END
|
||
|
||
@conversation.entry_point
|
||
@handler.command(command="post", block=False, admin=True)
|
||
async def command_start(self, update: "Update", context: "ContextTypes.DEFAULT_TYPE") -> int:
|
||
user = update.effective_user
|
||
message = update.effective_message
|
||
logger.info("用户 %s[%s] POST命令请求", user.full_name, user.id)
|
||
post_handler_data = context.chat_data.get("post_handler_data")
|
||
if post_handler_data is None:
|
||
post_handler_data = PostHandlerData()
|
||
context.chat_data["post_handler_data"] = post_handler_data
|
||
text = f"✿✿ヽ（°▽°）ノ✿ 你好！ {user.username} ，\n" "只需复制URL回复即可 \n" "退出投稿只需回复退出"
|
||
reply_keyboard = [["退出"]]
|
||
await message.reply_text(text, reply_markup=ReplyKeyboardMarkup(reply_keyboard, True, True))
|
||
return CHECK_POST
|
||
|
||
@conversation.state(state=CHECK_POST)
|
||
@handler.message(filters=filters.TEXT & ~filters.COMMAND, block=False)
|
||
async def check_post(self, update: "Update", context: "ContextTypes.DEFAULT_TYPE") -> int:
|
||
post_handler_data: PostHandlerData = context.chat_data.get("post_handler_data")
|
||
message = update.effective_message
|
||
if message.text == "退出":
|
||
await message.reply_text("退出投稿", reply_markup=ReplyKeyboardRemove())
|
||
return ConversationHandler.END
|
||
|
||
post_id, post_type = Hyperion.extract_post_id(update.message.text)
|
||
if post_id == -1:
|
||
await message.reply_text("获取作品ID错误，请检查连接是否合法", reply_markup=ReplyKeyboardRemove())
|
||
return ConversationHandler.END
|
||
return await self.send_post_info(post_handler_data, message, post_id, post_type)
|
||
|
||
async def send_post_info(
|
||
self, post_handler_data: PostHandlerData, message: "Message", post_id: int, post_type: "PostTypeEnum"
|
||
) -> int:
|
||
bbs = self.get_bbs_client(post_type)
|
||
post_info = await bbs.get_post_info(self.gids, post_id)
|
||
post_images = await bbs.get_images_by_post_id(self.gids, post_id)
|
||
await bbs.close()
|
||
post_images = await self.gif_to_mp4(post_images)
|
||
post_data = post_info["post"]["post"]
|
||
post_subject = post_data["subject"]
|
||
post_soup = BeautifulSoup(post_info.content, features="html.parser")
|
||
post_text, too_long = self.parse_post_text(post_soup, post_subject)
|
||
url = post_info.get_url(self.short_name)
|
||
max_len = MessageLimit.CAPTION_LENGTH - 100
|
||
if too_long or len(post_text) >= max_len:
|
||
post_text = self.safe_cut(post_text, max_len)
|
||
await message.reply_text(f"警告！图片字符描述已经超过 {max_len} 个字，已经切割")
|
||
post_text += f"\n\n[source]({url})"
|
||
try:
|
||
if len(post_images) > 1:
|
||
media = [self.input_media(img_info) for img_info in post_images if not img_info.is_error]
|
||
index = (math.ceil(len(media) / 10) - 1) * 10
|
||
media[index].caption = post_text
|
||
media[index].parse_mode = ParseMode.MARKDOWN_V2
|
||
for group in ArkoWrapper(media).group(10): # 每 10 张图片分一个组
|
||
await message.reply_media_group(list(group), write_timeout=len(group) * 5)
|
||
elif len(post_images) == 1:
|
||
image = post_images[0]
|
||
await message.reply_photo(image.data, caption=post_text, parse_mode=ParseMode.MARKDOWN_V2)
|
||
else:
|
||
await message.reply_text(post_text, parse_mode=ParseMode.MARKDOWN_V2)
|
||
except BadRequest as exc:
|
||
await message.reply_text(f"发送图片时发生错误 {exc.message}", reply_markup=ReplyKeyboardRemove())
|
||
logger.error("Post模块发送图片时发生错误 %s", exc.message)
|
||
return ConversationHandler.END
|
||
except TypeError as exc:
|
||
await message.reply_text("发送图片时发生错误，错误信息已经写到日记", reply_markup=ReplyKeyboardRemove())
|
||
logger.error("Post模块发送图片时发生错误", exc_info=exc)
|
||
|
||
return ConversationHandler.END
|
||
post_handler_data.post_text = post_text
|
||
post_handler_data.post_images = post_images
|
||
post_handler_data.delete_photo = []
|
||
post_handler_data.tags = []
|
||
post_handler_data.channel_id = -1
|
||
await message.reply_text("请选择你的操作", reply_markup=self.MENU_KEYBOARD)
|
||
return CHECK_COMMAND
|
||
|
||
@conversation.state(state=CHECK_COMMAND)
|
||
@handler.message(filters=filters.TEXT & ~filters.COMMAND, block=False)
|
||
async def check_command(self, update: "Update", context: "ContextTypes.DEFAULT_TYPE") -> int:
|
||
message = update.effective_message
|
||
if message.text == "退出":
|
||
await message.reply_text("退出任务", reply_markup=ReplyKeyboardRemove())
|
||
return ConversationHandler.END
|
||
if message.text == "推送频道":
|
||
return await self.get_channel(update, context)
|
||
if message.text == "添加TAG":
|
||
return await self.add_tags(update, context)
|
||
if message.text == "编辑文字":
|
||
return await self.edit_text(update, context)
|
||
if message.text == "删除图片":
|
||
return await self.delete_photo(update, context)
|
||
return ConversationHandler.END
|
||
|
||
@staticmethod
|
||
async def delete_photo(update: "Update", context: "ContextTypes.DEFAULT_TYPE") -> int:
|
||
post_handler_data: PostHandlerData = context.chat_data.get("post_handler_data")
|
||
photo_len = len(post_handler_data.post_images)
|
||
message = update.effective_message
|
||
await message.reply_text(
|
||
"请回复你要删除的图片的序列，从1开始，如果删除多张图片回复的序列请以空格作为分隔符，"
|
||
f"当前一共有 {photo_len} 张图片"
|
||
)
|
||
return GTE_DELETE_PHOTO
|
||
|
||
@conversation.state(state=GTE_DELETE_PHOTO)
|
||
@handler.message(filters=filters.TEXT & ~filters.COMMAND, block=False)
|
||
async def get_delete_photo(self, update: "Update", context: "ContextTypes.DEFAULT_TYPE") -> int:
|
||
post_handler_data: PostHandlerData = context.chat_data.get("post_handler_data")
|
||
photo_len = len(post_handler_data.post_images)
|
||
message = update.effective_message
|
||
args = message.text.split(" ")
|
||
index: List[int] = []
|
||
try:
|
||
for temp in args:
|
||
if int(temp) > photo_len:
|
||
raise ValueError
|
||
index.append(int(temp))
|
||
except ValueError:
|
||
await message.reply_text("数据不合法，请重新操作")
|
||
return GTE_DELETE_PHOTO
|
||
post_handler_data.delete_photo = index
|
||
await message.reply_text("删除成功")
|
||
await message.reply_text("请选择你的操作", reply_markup=self.MENU_KEYBOARD)
|
||
return CHECK_COMMAND
|
||
|
||
async def get_channel(self, update: "Update", _: "ContextTypes.DEFAULT_TYPE") -> int:
|
||
message = update.effective_message
|
||
reply_keyboard = []
|
||
try:
|
||
for channel_id in config.channels:
|
||
chat = await self.get_chat(chat_id=channel_id)
|
||
reply_keyboard.append([f"{chat.username}"])
|
||
except KeyError as error:
|
||
logger.error("从配置文件获取频道信息发生错误，退出任务", exc_info=error)
|
||
await message.reply_text("从配置文件获取频道信息发生错误，退出任务", reply_markup=ReplyKeyboardRemove())
|
||
return ConversationHandler.END
|
||
await message.reply_text("请选择你要推送的频道", reply_markup=ReplyKeyboardMarkup(reply_keyboard, True, True))
|
||
return GET_POST_CHANNEL
|
||
|
||
@conversation.state(state=GET_POST_CHANNEL)
|
||
@handler.message(filters=filters.TEXT & ~filters.COMMAND, block=False)
|
||
async def get_post_channel(self, update: "Update", context: "ContextTypes.DEFAULT_TYPE") -> int:
|
||
post_handler_data: PostHandlerData = context.chat_data.get("post_handler_data")
|
||
message = update.effective_message
|
||
channel_id = -1
|
||
try:
|
||
for channel_chat_id in config.channels:
|
||
chat = await self.get_chat(chat_id=channel_chat_id)
|
||
if message.text == chat.username:
|
||
channel_id = channel_chat_id
|
||
except KeyError as exc:
|
||
logger.error("从配置文件获取频道信息发生错误，退出任务", exc_info=exc)
|
||
logger.exception(exc)
|
||
await message.reply_text("从配置文件获取频道信息发生错误，退出任务", reply_markup=ReplyKeyboardRemove())
|
||
return ConversationHandler.END
|
||
if channel_id == -1:
|
||
await message.reply_text("获取频道信息失败，请检查你输入的内容是否正确", reply_markup=ReplyKeyboardRemove())
|
||
return ConversationHandler.END
|
||
post_handler_data.channel_id = channel_id
|
||
reply_keyboard = [["确认", "退出"]]
|
||
await message.reply_text("请核对你修改的信息", reply_markup=ReplyKeyboardMarkup(reply_keyboard, True, True))
|
||
return SEND_POST
|
||
|
||
@staticmethod
|
||
async def add_tags(update: "Update", _: "ContextTypes.DEFAULT_TYPE") -> int:
|
||
message = update.effective_message
|
||
await message.reply_text(
|
||
"请回复添加的tag名称，如果要添加多个tag请以空格作为分隔符，不用添加 # 作为开头，推送时程序会自动添加"
|
||
)
|
||
return GET_TAGS
|
||
|
||
@conversation.state(state=GET_TAGS)
|
||
@handler.message(filters=filters.TEXT & ~filters.COMMAND, block=False)
|
||
async def get_tags(self, update: "Update", context: "ContextTypes.DEFAULT_TYPE") -> int:
|
||
post_handler_data: PostHandlerData = context.chat_data.get("post_handler_data")
|
||
message = update.effective_message
|
||
args = message.text.split(" ")
|
||
post_handler_data.tags = args
|
||
await message.reply_text("添加成功")
|
||
await message.reply_text("请选择你的操作", reply_markup=self.MENU_KEYBOARD)
|
||
return CHECK_COMMAND
|
||
|
||
@staticmethod
|
||
async def edit_text(update: "Update", _: "ContextTypes.DEFAULT_TYPE") -> int:
|
||
message = update.effective_message
|
||
await message.reply_text("请回复替换的文本")
|
||
return GET_TEXT
|
||
|
||
@conversation.state(state=GET_TEXT)
|
||
@handler.message(filters=filters.TEXT & ~filters.COMMAND, block=False)
|
||
async def get_edit_text(self, update: "Update", context: "ContextTypes.DEFAULT_TYPE") -> int:
|
||
post_handler_data: PostHandlerData = context.chat_data.get("post_handler_data")
|
||
message = update.effective_message
|
||
post_handler_data.post_text = message.text_markdown_v2
|
||
await message.reply_text("替换成功")
|
||
await message.reply_text("请选择你的操作", reply_markup=self.MENU_KEYBOARD)
|
||
return CHECK_COMMAND
|
||
|
||
@conversation.state(state=SEND_POST)
|
||
@handler.message(filters=filters.TEXT & ~filters.COMMAND, block=False)
|
||
async def send_post(self, update: "Update", context: "ContextTypes.DEFAULT_TYPE") -> int:
|
||
post_handler_data: PostHandlerData = context.chat_data.get("post_handler_data")
|
||
message = update.effective_message
|
||
if message.text == "退出":
|
||
await message.reply_text(text="退出任务", reply_markup=ReplyKeyboardRemove())
|
||
return ConversationHandler.END
|
||
await message.reply_text("正在推送", reply_markup=ReplyKeyboardRemove())
|
||
channel_id = post_handler_data.channel_id
|
||
channel_name = None
|
||
try:
|
||
for channel_info in config.channels:
|
||
if post_handler_data.channel_id == channel_info:
|
||
chat = await self.get_chat(chat_id=channel_id)
|
||
channel_name = chat.username
|
||
except KeyError as exc:
|
||
logger.error("从配置文件获取频道信息发生错误，退出任务")
|
||
logger.exception(exc)
|
||
await message.reply_text("从配置文件获取频道信息发生错误，退出任务", reply_markup=ReplyKeyboardRemove())
|
||
return ConversationHandler.END
|
||
post_text = post_handler_data.post_text
|
||
post_images = []
|
||
for index, _ in enumerate(post_handler_data.post_images):
|
||
if index + 1 not in post_handler_data.delete_photo:
|
||
post_images.append(post_handler_data.post_images[index])
|
||
post_text += f" @{escape_markdown(channel_name, version=2)}"
|
||
for tag in post_handler_data.tags:
|
||
post_text += f" \\#{tag}"
|
||
try:
|
||
if len(post_images) > 1:
|
||
media = [self.input_media(img_info) for img_info in post_images if not img_info.is_error]
|
||
index = (math.ceil(len(media) / 10) - 1) * 10
|
||
media[index].caption = post_text
|
||
media[index].parse_mode = ParseMode.MARKDOWN_V2
|
||
for group in ArkoWrapper(media).group(10): # 每 10 张图片分一个组
|
||
await context.bot.send_media_group(channel_id, media=list(group), write_timeout=len(group) * 5)
|
||
elif len(post_images) == 1:
|
||
image = post_images[0]
|
||
await context.bot.send_photo(
|
||
channel_id, photo=image.data, caption=post_text, parse_mode=ParseMode.MARKDOWN_V2
|
||
)
|
||
elif not post_images:
|
||
await context.bot.send_message(channel_id, post_text, parse_mode=ParseMode.MARKDOWN_V2)
|
||
else:
|
||
await message.reply_text("图片获取错误", reply_markup=ReplyKeyboardRemove()) # excuse?
|
||
return ConversationHandler.END
|
||
except BadRequest as exc:
|
||
await message.reply_text(f"发送图片时发生错误 {exc.message}", reply_markup=ReplyKeyboardRemove())
|
||
logger.error("Post模块发送图片时发生错误 %s", exc.message)
|
||
return ConversationHandler.END
|
||
except TypeError as exc:
|
||
await message.reply_text("发送图片时发生错误，错误信息已经写到日记", reply_markup=ReplyKeyboardRemove())
|
||
logger.error("Post模块发送图片时发生错误", exc_info=exc)
|
||
await message.reply_text("推送成功", reply_markup=ReplyKeyboardRemove())
|
||
return ConversationHandler.END
|