edit parsers

This commit is contained in:
Zaid _ 2024-06-20 03:46:13 +03:00
parent 04aea46fe4
commit 1e901b0b6a
2 changed files with 7 additions and 5 deletions

View File

@ -1,11 +1,10 @@
""" """
Simple HTML -> Telegram entity parser. Simple HTML -> Telegram entity parser.
""" """
import struct
from collections import deque from collections import deque
from html import escape from html import escape
from html.parser import HTMLParser 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 ..helpers import add_surrogate, del_surrogate, within_surrogate, strip_text
from ..tl import TLObject from ..tl import TLObject
@ -14,7 +13,7 @@ from ..tl.types import (
MessageEntityPre, MessageEntityEmail, MessageEntityUrl, MessageEntityPre, MessageEntityEmail, MessageEntityUrl,
MessageEntityTextUrl, MessageEntityMentionName, MessageEntityTextUrl, MessageEntityMentionName,
MessageEntityUnderline, MessageEntityStrike, MessageEntityBlockquote, MessageEntityUnderline, MessageEntityStrike, MessageEntityBlockquote,
TypeMessageEntity TypeMessageEntity, MessageEntitySpoiler
) )
@ -42,6 +41,8 @@ class HTMLToTelegramParser(HTMLParser):
EntityType = MessageEntityUnderline EntityType = MessageEntityUnderline
elif tag == 'del' or tag == 's': elif tag == 'del' or tag == 's':
EntityType = MessageEntityStrike EntityType = MessageEntityStrike
elif tag == 'spoiler':
EntityType = MessageEntitySpoiler
elif tag == 'blockquote': elif tag == 'blockquote':
EntityType = MessageEntityBlockquote EntityType = MessageEntityBlockquote
elif tag == 'code': elif tag == 'code':

View File

@ -11,7 +11,7 @@ from ..tl import TLObject
from ..tl.types import ( from ..tl.types import (
MessageEntityBold, MessageEntityItalic, MessageEntityCode, MessageEntityBold, MessageEntityItalic, MessageEntityCode,
MessageEntityPre, MessageEntityTextUrl, MessageEntityMentionName, MessageEntityPre, MessageEntityTextUrl, MessageEntityMentionName,
MessageEntityStrike MessageEntityStrike, MessageEntitySpoiler
) )
DEFAULT_DELIMITERS = { DEFAULT_DELIMITERS = {
@ -19,7 +19,8 @@ DEFAULT_DELIMITERS = {
'__': MessageEntityItalic, '__': MessageEntityItalic,
'~~': MessageEntityStrike, '~~': MessageEntityStrike,
'`': MessageEntityCode, '`': MessageEntityCode,
'```': MessageEntityPre '```': MessageEntityPre,
'||': MessageEntitySpoiler
} }
DEFAULT_URL_RE = re.compile(r'\[([^\]]+)\]\(([^)]+)\)') DEFAULT_URL_RE = re.compile(r'\[([^\]]+)\]\(([^)]+)\)')