edit parsers

This commit is contained in:
Zaid _ 2024-06-20 04:30:32 +03:00
parent 94048d9102
commit 32c10eafe7
2 changed files with 4 additions and 2 deletions

View File

@ -48,7 +48,7 @@ class HTMLToTelegramParser(HTMLParser):
EntityType = MessageEntityStrike EntityType = MessageEntityStrike
elif tag == "blockquote": elif tag == "blockquote":
EntityType = MessageEntityBlockquote EntityType = MessageEntityBlockquote
elif tag == "details": elif tag == "spoiler":
EntityType = MessageEntitySpoiler EntityType = MessageEntitySpoiler
elif tag == "code": elif tag == "code":
try: try:
@ -142,7 +142,7 @@ ENTITY_TO_FORMATTER: dict[
MessageEntityUnderline: ("<u>", "</u>"), MessageEntityUnderline: ("<u>", "</u>"),
MessageEntityStrike: ("<del>", "</del>"), MessageEntityStrike: ("<del>", "</del>"),
MessageEntityBlockquote: ("<blockquote>", "</blockquote>"), MessageEntityBlockquote: ("<blockquote>", "</blockquote>"),
MessageEntitySpoiler: ("<details>", "</details>"), MessageEntitySpoiler: ("<spoiler>", "</spoiler>"),
MessageEntityPre: lambda e, _: ( MessageEntityPre: lambda e, _: (
'<pre><code class="language-{}">'.format(e.language) if e.language else "<pre>", '<pre><code class="language-{}">'.format(e.language) if e.language else "<pre>",
"</code></pre>" if e.language else "</pre>", "</code></pre>" if e.language else "</pre>",

View File

@ -16,6 +16,7 @@ from ...tl.types import (
MessageEntityStrike, MessageEntityStrike,
MessageEntityTextUrl, MessageEntityTextUrl,
MessageEntityUnderline, MessageEntityUnderline,
MessageEntitySpoiler
) )
from .strings import add_surrogate, del_surrogate, within_surrogate from .strings import add_surrogate, del_surrogate, within_surrogate
@ -27,6 +28,7 @@ DELIMITERS: dict[Type[MessageEntity], tuple[str, str]] = {
MessageEntityItalic: ("_", "_"), MessageEntityItalic: ("_", "_"),
MessageEntityStrike: ("~~", "~~"), MessageEntityStrike: ("~~", "~~"),
MessageEntityUnderline: ("# ", ""), MessageEntityUnderline: ("# ", ""),
MessageEntitySpoiler: ("||", "||")
} }
# Not trying to be complete; just enough to have an alternative (mostly for inline underline). # Not trying to be complete; just enough to have an alternative (mostly for inline underline).