mirror of
https://github.com/LonamiWebs/Telethon.git
synced 2024-11-22 09:26:37 +03:00
Updated Sending more than just messages (markdown)
parent
7d4ea3ce77
commit
7f691dc74b
|
@ -120,3 +120,223 @@ client.send_message('me', 'hello this is a [hidden text](spoiler), with custom e
|
||||||
```
|
```
|
||||||
|
|
||||||
You may have noticed the emoji URL is followed by a number. This number is a `document_id`. To find it, the easiest way is to send a message to your own chat with the premium emoji you want to use using an official client, and then use Telethon to print the `message.entities`. It will contain the `document_id` you need to use.
|
You may have noticed the emoji URL is followed by a number. This number is a `document_id`. To find it, the easiest way is to send a message to your own chat with the premium emoji you want to use using an official client, and then use Telethon to print the `message.entities`. It will contain the `document_id` you need to use.
|
||||||
|
|
||||||
|
<details>
|
||||||
|
<summary>Full html parser/unparser example </summary>
|
||||||
|
|
||||||
|
A full example of html parser and unparser that have both spoiler and custom emoji.\
|
||||||
|
you can use this code exactly like how would use the custom markdown parser mentioned above, just set `parse_mode` to `CustomHtmlParser()` and you are done.
|
||||||
|
```python
|
||||||
|
"""
|
||||||
|
Simple HTML -> Telegram entity parser.
|
||||||
|
"""
|
||||||
|
from collections import deque
|
||||||
|
from html import escape
|
||||||
|
from html.parser import HTMLParser
|
||||||
|
from typing import Iterable, Tuple, List
|
||||||
|
|
||||||
|
from telethon.helpers import add_surrogate, del_surrogate, within_surrogate, strip_text
|
||||||
|
from telethon.tl import TLObject
|
||||||
|
from telethon.types import (
|
||||||
|
MessageEntityBold, MessageEntityItalic, MessageEntityCode,
|
||||||
|
MessageEntityPre, MessageEntityEmail, MessageEntityUrl,
|
||||||
|
MessageEntityTextUrl, MessageEntityMentionName,
|
||||||
|
MessageEntityUnderline, MessageEntityStrike, MessageEntityBlockquote,
|
||||||
|
MessageEntityCustomEmoji, MessageEntitySpoiler, TypeMessageEntity
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class HTMLToTelegramParser(HTMLParser):
|
||||||
|
def __init__(self):
|
||||||
|
super().__init__()
|
||||||
|
self.text = ''
|
||||||
|
self.entities = []
|
||||||
|
self._building_entities = {}
|
||||||
|
self._open_tags = deque()
|
||||||
|
self._open_tags_meta = deque()
|
||||||
|
|
||||||
|
def handle_starttag(self, tag, attrs):
|
||||||
|
self._open_tags.appendleft(tag)
|
||||||
|
self._open_tags_meta.appendleft(None)
|
||||||
|
|
||||||
|
attrs = dict(attrs)
|
||||||
|
EntityType = None
|
||||||
|
args = {}
|
||||||
|
match tag:
|
||||||
|
case 'strong' | 'b':
|
||||||
|
EntityType = MessageEntityBold
|
||||||
|
case 'em' | 'i':
|
||||||
|
EntityType = MessageEntityItalic
|
||||||
|
case 'u':
|
||||||
|
EntityType = MessageEntityUnderline
|
||||||
|
case 'del' | 's':
|
||||||
|
EntityType = MessageEntityStrike
|
||||||
|
case 'blockquote':
|
||||||
|
EntityType = MessageEntityBlockquote
|
||||||
|
case 'tg-spoiler':
|
||||||
|
EntityType = MessageEntitySpoiler
|
||||||
|
case 'code':
|
||||||
|
try:
|
||||||
|
# If we're in the middle of a <pre> tag, this <code> tag is
|
||||||
|
# probably intended for syntax highlighting.
|
||||||
|
#
|
||||||
|
# Syntax highlighting is set with
|
||||||
|
# <code class='language-...'>codeblock</code>
|
||||||
|
# inside <pre> tags
|
||||||
|
pre = self._building_entities['pre']
|
||||||
|
try:
|
||||||
|
pre.language = attrs['class'][len('language-'):] # type: ignore
|
||||||
|
except KeyError:
|
||||||
|
pass
|
||||||
|
except KeyError:
|
||||||
|
EntityType = MessageEntityCode
|
||||||
|
case 'pre':
|
||||||
|
EntityType = MessageEntityPre
|
||||||
|
args['language'] = ''
|
||||||
|
case 'a':
|
||||||
|
try:
|
||||||
|
url = attrs['href']
|
||||||
|
if not url:
|
||||||
|
raise KeyError
|
||||||
|
except KeyError:
|
||||||
|
return
|
||||||
|
if url.startswith('mailto:'):
|
||||||
|
url = url[len('mailto:'):]
|
||||||
|
EntityType = MessageEntityEmail
|
||||||
|
else:
|
||||||
|
if self.get_starttag_text() == url:
|
||||||
|
EntityType = MessageEntityUrl
|
||||||
|
else:
|
||||||
|
EntityType = MessageEntityTextUrl
|
||||||
|
args['url'] = del_surrogate(url)
|
||||||
|
url = None
|
||||||
|
self._open_tags_meta.popleft()
|
||||||
|
self._open_tags_meta.appendleft(url)
|
||||||
|
case 'tg-emoji':
|
||||||
|
try:
|
||||||
|
emoji_id = attrs['emoji-id']
|
||||||
|
if not emoji_id:
|
||||||
|
raise ValueError
|
||||||
|
emoji_id = int(emoji_id)
|
||||||
|
except (KeyError, ValueError):
|
||||||
|
return
|
||||||
|
EntityType = MessageEntityCustomEmoji
|
||||||
|
args['document_id'] = emoji_id
|
||||||
|
|
||||||
|
if EntityType and tag not in self._building_entities:
|
||||||
|
self._building_entities[tag] = EntityType(
|
||||||
|
offset=len(self.text),
|
||||||
|
# The length will be determined when closing the tag.
|
||||||
|
length=0,
|
||||||
|
**args)
|
||||||
|
|
||||||
|
def handle_data(self, data):
|
||||||
|
previous_tag = self._open_tags[0] if len(self._open_tags) > 0 else ''
|
||||||
|
if previous_tag == 'a':
|
||||||
|
url = self._open_tags_meta[0]
|
||||||
|
if url:
|
||||||
|
data = url
|
||||||
|
|
||||||
|
for tag, entity in self._building_entities.items():
|
||||||
|
entity.length += len(data)
|
||||||
|
|
||||||
|
self.text += data
|
||||||
|
|
||||||
|
def handle_endtag(self, tag):
|
||||||
|
try:
|
||||||
|
self._open_tags.popleft()
|
||||||
|
self._open_tags_meta.popleft()
|
||||||
|
except IndexError:
|
||||||
|
pass
|
||||||
|
entity = self._building_entities.pop(tag, None)
|
||||||
|
if entity:
|
||||||
|
self.entities.append(entity)
|
||||||
|
|
||||||
|
ENTITY_TO_FORMATTER = {
|
||||||
|
MessageEntityBold: ('<strong>', '</strong>'),
|
||||||
|
MessageEntityItalic: ('<em>', '</em>'),
|
||||||
|
MessageEntityCode: ('<code>', '</code>'),
|
||||||
|
MessageEntityUnderline: ('<u>', '</u>'),
|
||||||
|
MessageEntityStrike: ('<del>', '</del>'),
|
||||||
|
MessageEntityBlockquote: ('<blockquote>', '</blockquote>'),
|
||||||
|
MessageEntitySpoiler: ('<tg-spoiler>', '</tg-spoiler>'),
|
||||||
|
MessageEntityPre: lambda e, _: (
|
||||||
|
"<pre>\n"
|
||||||
|
" <code class='language-{}'>\n"
|
||||||
|
" ".format(e.language), "{}\n"
|
||||||
|
" </code>\n"
|
||||||
|
"</pre>"
|
||||||
|
),
|
||||||
|
MessageEntityEmail: lambda _, t: ('<a href="mailto:{}">'.format(t), '</a>'),
|
||||||
|
MessageEntityUrl: lambda _, t: ('<a href="{}">'.format(t), '</a>'),
|
||||||
|
MessageEntityTextUrl: lambda e, _: ('<a href="{}">'.format(escape(e.url)), '</a>'),
|
||||||
|
MessageEntityMentionName: lambda e, _: ('<a href="tg://user?id={}">'.format(e.user_id), '</a>'),
|
||||||
|
MessageEntityCustomEmoji: lambda e, _: ('<tg-emoji emoji-id="{}">'.format(e.document_id), '</tg-emoji>')
|
||||||
|
}
|
||||||
|
|
||||||
|
class CustomHtmlParser:
|
||||||
|
@staticmethod
|
||||||
|
def parse(html: str) -> Tuple[str, List[TypeMessageEntity]]:
|
||||||
|
"""
|
||||||
|
Parses the given HTML message and returns its stripped representation
|
||||||
|
plus a list of the MessageEntity's that were found.
|
||||||
|
|
||||||
|
:param html: the message with HTML to be parsed.
|
||||||
|
:return: a tuple consisting of (clean message, [message entities]).
|
||||||
|
"""
|
||||||
|
if not html:
|
||||||
|
return html, []
|
||||||
|
|
||||||
|
parser = HTMLToTelegramParser()
|
||||||
|
parser.feed(add_surrogate(html))
|
||||||
|
text = strip_text(parser.text, parser.entities)
|
||||||
|
parser.entities.reverse()
|
||||||
|
parser.entities.sort(key=lambda entity: entity.offset)
|
||||||
|
return del_surrogate(text), parser.entities
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def unparse(text: str, entities: Iterable[TypeMessageEntity]) -> str:
|
||||||
|
"""
|
||||||
|
Performs the reverse operation to .parse(), effectively returning HTML
|
||||||
|
given a normal text and its MessageEntity's.
|
||||||
|
|
||||||
|
:param text: the text to be reconverted into HTML.
|
||||||
|
:param entities: the MessageEntity's applied to the text.
|
||||||
|
:return: a HTML representation of the combination of both inputs.
|
||||||
|
"""
|
||||||
|
if not text:
|
||||||
|
return text
|
||||||
|
elif not entities:
|
||||||
|
return escape(text)
|
||||||
|
|
||||||
|
if isinstance(entities, TLObject):
|
||||||
|
entities = (entities,) # type: ignore
|
||||||
|
|
||||||
|
text = add_surrogate(text)
|
||||||
|
insert_at = []
|
||||||
|
for i, entity in enumerate(entities):
|
||||||
|
s = entity.offset
|
||||||
|
e = entity.offset + entity.length
|
||||||
|
delimiter = ENTITY_TO_FORMATTER.get(type(entity), None) # type: ignore
|
||||||
|
if delimiter:
|
||||||
|
if callable(delimiter):
|
||||||
|
delimiter = delimiter(entity, text[s:e])
|
||||||
|
insert_at.append((s, i, delimiter[0]))
|
||||||
|
insert_at.append((e, -i, delimiter[1]))
|
||||||
|
|
||||||
|
insert_at.sort(key=lambda t: (t[0], t[1]))
|
||||||
|
next_escape_bound = len(text)
|
||||||
|
while insert_at:
|
||||||
|
# Same logic as markdown.py
|
||||||
|
at, _, what = insert_at.pop()
|
||||||
|
while within_surrogate(text, at):
|
||||||
|
at += 1
|
||||||
|
|
||||||
|
text = text[:at] + what + escape(text[at:next_escape_bound]) + text[next_escape_bound:]
|
||||||
|
next_escape_bound = at
|
||||||
|
|
||||||
|
text = escape(text[:next_escape_bound]) + text[next_escape_bound:]
|
||||||
|
|
||||||
|
return del_surrogate(text)
|
||||||
|
```
|
||||||
|
</details>
|
Loading…
Reference in New Issue
Block a user