Xtao-Labs/MTPyroger
3
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Revamp HTML style parser

Browse Source
This commit is contained in:
Dan 2018-01-23 14:43:12 +01:00
parent 0e3d08ae75
commit ae92c92c06
1 changed files with 30 additions and 57 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

77
pyrogram/extensions/html.py
View File

@ -7,23 +7,15 @@ from pyrogram.api.types import (
MessageEntityCode as Code,
MessageEntityTextUrl as Url,
MessageEntityPre as Pre,
InputMessageEntityMentionName as Mention
MessageEntityMentionName as MentionInvalid,
InputMessageEntityMentionName as Mention,
)
class HTML:
SMP_RE = re.compile(r"[\U00010000-\U0010FFFF]")
BOLD_RE = r"(?P<b><b>(?P<b_body>.*?)</b>)"
STRONG_RE = r"(?P<strong><strong>(?P<strong_body>.*?)</strong>)"
ITALIC_RE = r"(?P<i><i>(?P<i_body>.*?)</i>)"
EMPATHIZE_RE = r"(?P<em><em>(?P<em_body>.*?)</em>)"
CODE_RE = r"(?P<code><code>(?P<code_body>.*?)</code>)"
PRE_RE = r"(?P<pre><pre>(?P<pre_body>.*?)</pre>)"
MENTION_RE = r"(?P<mention><a href=\"tg://user\?id=(?P<user_id>\d+?)\">(?P<mention_text>.*?)</a>)"
URL_RE = r"(?P<url><a href=\"(?P<url_path>.*?)\">(?P<url_text>.*?)</a>)"
HTML_RE = re.compile("|".join([BOLD_RE, STRONG_RE, ITALIC_RE, EMPATHIZE_RE, CODE_RE, PRE_RE, MENTION_RE, URL_RE]))
HTML_RE = re.compile(r"<(\w+)(?: href=\"(.*)\")?>(.*)</\1>")
MENTION_RE = re.compile(r"tg://user\?id=(\d+)")
@classmethod
def add_surrogates(cls, text):
@ -45,57 +37,38 @@ class HTML:
text = self.add_surrogates(text)
offset = 0
# TODO: Beautify ifs
for match in self.HTML_RE.finditer(text):
start = match.start() - offset
style, url, body = match.groups()
if match.group("b"):
pattern = match.group("b")
body = match.group("b_body")
if url:
mention = self.MENTION_RE.match(url)
if mention:
user_id = int(mention.group(1))
input_user = self.peers_by_id.get(user_id, None)
entity = (
Mention(start, len(body), input_user)
if input_user else MentionInvalid(start, len(body), user_id)
)
else:
entity = Url(start, len(body), url)
else:
if style == "b" or style == "strong":
entity = Bold(start, len(body))
offset += 7
elif match.group("strong"):
pattern = match.group("strong")
body = match.group("strong_body")
entity = Bold(start, len(body))
offset += 17
elif match.group("i"):
pattern = match.group("i")
body = match.group("i_body")
elif style == "i" or style == "em":
entity = Italic(start, len(body))
offset += 7
elif match.group("em"):
pattern = match.group("em")
body = match.group("em_body")
entity = Italic(start, len(body))
offset += 9
elif match.group("code"):
pattern = match.group("code")
body = match.group("code_body")
elif style == "code":
entity = Code(start, len(body))
offset += 13
elif match.group("pre"):
pattern = match.group("pre")
body = match.group("pre_body")
elif style == "pre":
entity = Pre(start, len(body), "")
offset += 11
elif match.group("mention"):
pattern = match.group("mention")
body = match.group("mention_text")
user_id = match.group("user_id")
entity = Mention(start, len(body), self.peers_by_id[int(user_id)])
offset += len(user_id) + 28
elif match.group("url"):
pattern = match.group("url")
body = match.group("url_text")
path = match.group("url_path")
entity = Url(start, len(body), path)
offset += len(path) + 15
else:
continue
entities.append(entity)
text = text.replace(pattern, body)
text = text.replace(match.group(), body)
offset += len(style) * 2 + 5 + (len(url) + 8 if url else 0)
return dict(
message=self.remove_surrogates(text),