PaiGramTeam/PamGram
3
0
Fork 0
mirror of https://github.com/PaiGramTeam/PamGram.git synced 2024-11-16 03:55:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

🐛 Fix safe cut post text

Browse Source
This commit is contained in:
xtaodada 2024-08-06 20:46:56 +08:00
parent 663445e39b
commit ee2e07717f
Signed by: xtaodada
GPG Key ID: 4CBB3F4FA8C85659
1 changed files with 17 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
plugins/admin/post.py
View File

@ -1,5 +1,6 @@
import math import math
import os import os
import re
from asyncio import create_subprocess_shell, subprocess from asyncio import create_subprocess_shell, subprocess
from functools import partial from functools import partial
from typing import List, Optional, Tuple, TYPE_CHECKING, Union, Dict from typing import List, Optional, Tuple, TYPE_CHECKING, Union, Dict
@ -199,6 +200,18 @@ class Post(Plugin.Conversation):
post_text += f"{escape_markdown(soup.get_text(), version=2)}\n" post_text += f"{escape_markdown(soup.get_text(), version=2)}\n"
return post_text.strip(), too_long 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 @staticmethod
def input_media( def input_media(
media: "ArtworkImage", *args, **kwargs media: "ArtworkImage", *args, **kwargs
@ -359,10 +372,11 @@ class Post(Plugin.Conversation):
post_soup = BeautifulSoup(post_info.content, features="html.parser") post_soup = BeautifulSoup(post_info.content, features="html.parser")
post_text, too_long = self.parse_post_text(post_soup, post_subject) post_text, too_long = self.parse_post_text(post_soup, post_subject)
url = post_info.get_url(self.short_name) 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})" post_text += f"\n\n[source]({url})"
if too_long or len(post_text) >= MessageLimit.CAPTION_LENGTH:
post_text = post_text[: MessageLimit.CAPTION_LENGTH]
await message.reply_text(f"警告！图片字符描述已经超过 {MessageLimit.CAPTION_LENGTH} 个字，已经切割")
if post_info.video_urls: if post_info.video_urls:
await message.reply_text("检测到视频，需要单独下载，视频链接：" + "\n".join(post_info.video_urls)) await message.reply_text("检测到视频，需要单独下载，视频链接：" + "\n".join(post_info.video_urls))
try: try: