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.
"""
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':

View File

@ -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'\[([^\]]+)\]\(([^)]+)\)')