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

Merge branch 'master' into docs

Browse Source
This commit is contained in:
Dan 2018-02-15 18:25:44 +01:00
commit 7b73b98d78
4 changed files with 7864 additions and 63 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pyrogram/__init__.py
View File

@ -30,3 +30,4 @@ from .client import ChatAction
from .client import Client from .client import Client
from .client import ParseMode from .client import ParseMode
from .client.input_media import InputMedia from .client.input_media import InputMedia
from .client import Emoji

1
pyrogram/client/__init__.py
View File

@ -19,3 +19,4 @@
from .chat_action import ChatAction from .chat_action import ChatAction
from .client import Client from .client import Client
from .parse_mode import ParseMode from .parse_mode import ParseMode
from .emoji import Emoji

7810
pyrogram/client/emoji.py Normal file
View File

File diff suppressed because it is too large Load Diff

115
pyrogram/client/style/markdown.py
View File

@ -24,95 +24,84 @@ from pyrogram.api.types import (
MessageEntityCode as Code, MessageEntityCode as Code,
MessageEntityTextUrl as Url, MessageEntityTextUrl as Url,
MessageEntityPre as Pre, MessageEntityPre as Pre,
MessageEntityMentionName as MentionInvalid,
InputMessageEntityMentionName as Mention InputMessageEntityMentionName as Mention
) )
from . import utils from . import utils
class Markdown: class Markdown:
INLINE_DELIMITERS = { BOLD_DELIMITER = "**"
"**": Bold, ITALIC_DELIMITER = "__"
"__": Italic, CODE_DELIMITER = "`"
"`": Code PRE_DELIMITER = "```"
}
# ``` python MARKDOWN_RE = re.compile(r"```([\w ]*)\n([\w\W]*)(?:\n|)```|\[([^[(]+)\]\(([^])]+)\)|({d})(.+?)\5".format(
# for i in range(10):
# print(i)
# ```
PRE_RE = r"(?P<pre>```(?P<lang>.*)\n(?P<code>(.|\n)*)\n```)"
# [url](github.com)
URL_RE = r"(?P<url>(\[(?P<url_text>.+?)\]\((?P<url_path>.+?)\)))"
# [name](tg://user?id=123456789)
MENTION_RE = r"(?P<mention>(\[(?P<mention_text>.+?)\]\(tg:\/\/user\?id=(?P<user_id>\d+?)\)))"
# **bold**
# __italic__
# `code`
INLINE_RE = r"(?P<inline>(?P<start_delimiter>{d})(?P<body>.+?)(?P<end_delimiter>{d}))".format(
d="|".join( d="|".join(
["".join(i) for i in [ ["".join(i) for i in [
["\{}".format(j) for j in i] ["\{}".format(j) for j in i]
for i in sorted( # Sort delimiters by length for i in [
INLINE_DELIMITERS.keys(), PRE_DELIMITER,
key=lambda k: len(k), # Or: key=len CODE_DELIMITER,
reverse=True ITALIC_DELIMITER,
) BOLD_DELIMITER
]
]] ]]
) )
) ))
MENTION_RE = re.compile(r"tg://user\?id=(\d+)")
MARKDOWN_RE = re.compile("|".join([PRE_RE, MENTION_RE, URL_RE, INLINE_RE])) def __init__(self, peers_by_id: dict):
def __init__(self, peers_by_id):
self.peers_by_id = peers_by_id self.peers_by_id = peers_by_id
def parse(self, text): def parse(self, message: str):
entities = [] entities = []
text = utils.add_surrogates(text) message = utils.add_surrogates(message).strip()
offset = 0 offset = 0
for match in self.MARKDOWN_RE.finditer(text): for match in self.MARKDOWN_RE.finditer(message):
start = match.start() - offset start = match.start() - offset
lang, pre, text, url, style, body = match.groups()
if match.group("pre"): if pre:
pattern = match.group("pre") body = pre = pre.strip()
lang = match.group("lang") entity = Pre(start, len(pre), lang.strip() or "")
replace = match.group("code") offset += len(lang) + len(self.PRE_DELIMITER) * 2
entity = Pre(start, len(replace), lang.strip()) elif url:
offset += len(lang) + 8 mention = self.MENTION_RE.match(url)
elif match.group("url"):
pattern = match.group("url")
replace = match.group("url_text")
path = match.group("url_path")
entity = Url(start, len(replace), path)
offset += len(path) + 4
elif match.group("mention"):
pattern = match.group("mention")
replace = match.group("mention_text")
user_id = match.group("user_id")
entity = Mention(start, len(replace), self.peers_by_id[int(user_id)])
offset += len(user_id) + 17
elif match.group("inline"):
pattern = match.group("inline")
replace = match.group("body")
start_delimiter = match.group("start_delimiter")
end_delimiter = match.group("end_delimiter")
if start_delimiter != end_delimiter: if mention:
user_id = int(mention.group(1))
input_user = self.peers_by_id.get(user_id, None)
entity = (
Mention(start, len(text), input_user)
if input_user
else MentionInvalid(start, len(text), user_id)
)
else:
entity = Url(start, len(text), url)
body = text
offset += len(url) + 4
else:
if style == self.BOLD_DELIMITER:
entity = Bold(start, len(body))
elif style == self.ITALIC_DELIMITER:
entity = Italic(start, len(body))
elif style == self.CODE_DELIMITER:
entity = Code(start, len(body))
elif style == self.PRE_DELIMITER:
entity = Pre(start, len(body), "")
else:
continue continue
entity = self.INLINE_DELIMITERS[start_delimiter](start, len(replace)) offset += len(style) * 2
offset += len(start_delimiter) * 2
else:
continue
entities.append(entity) entities.append(entity)
text = text.replace(pattern, replace) message = message.replace(match.group(), body)
return dict( return dict(
message=utils.remove_surrogates(text), message=utils.remove_surrogates(message),
entities=entities entities=entities
) )