From 1e901b0b6ad8de2d221c509af84172bc26bc2615 Mon Sep 17 00:00:00 2001 From: Zaid _ Date: Thu, 20 Jun 2024 03:46:13 +0300 Subject: [PATCH] edit parsers --- telethon/extensions/html.py | 7 ++++--- telethon/extensions/markdown.py | 5 +++-- 2 files changed, 7 insertions(+), 5 deletions(-) diff --git a/telethon/extensions/html.py b/telethon/extensions/html.py index ae45a763..8cbf1a44 100644 --- a/telethon/extensions/html.py +++ b/telethon/extensions/html.py @@ -1,11 +1,10 @@ """ Simple HTML -> Telegram entity parser. """ -import struct from collections import deque from html import escape from html.parser import HTMLParser -from typing import Iterable, Optional, Tuple, List +from typing import Iterable, Tuple, List from ..helpers import add_surrogate, del_surrogate, within_surrogate, strip_text from ..tl import TLObject @@ -14,7 +13,7 @@ from ..tl.types import ( MessageEntityPre, MessageEntityEmail, MessageEntityUrl, MessageEntityTextUrl, MessageEntityMentionName, MessageEntityUnderline, MessageEntityStrike, MessageEntityBlockquote, - TypeMessageEntity + TypeMessageEntity, MessageEntitySpoiler ) @@ -42,6 +41,8 @@ class HTMLToTelegramParser(HTMLParser): EntityType = MessageEntityUnderline elif tag == 'del' or tag == 's': EntityType = MessageEntityStrike + elif tag == 'spoiler': + EntityType = MessageEntitySpoiler elif tag == 'blockquote': EntityType = MessageEntityBlockquote elif tag == 'code': diff --git a/telethon/extensions/markdown.py b/telethon/extensions/markdown.py index 82e90345..75efdc2f 100644 --- a/telethon/extensions/markdown.py +++ b/telethon/extensions/markdown.py @@ -11,7 +11,7 @@ from ..tl import TLObject from ..tl.types import ( MessageEntityBold, MessageEntityItalic, MessageEntityCode, MessageEntityPre, MessageEntityTextUrl, MessageEntityMentionName, - MessageEntityStrike + MessageEntityStrike, MessageEntitySpoiler ) DEFAULT_DELIMITERS = { @@ -19,7 +19,8 @@ DEFAULT_DELIMITERS = { '__': MessageEntityItalic, '~~': MessageEntityStrike, '`': MessageEntityCode, - '```': MessageEntityPre + '```': MessageEntityPre, + '||': MessageEntitySpoiler } DEFAULT_URL_RE = re.compile(r'\[([^\]]+)\]\(([^)]+)\)')