iframe-proxy | Sunbelt Computer Software

History

127 lines (109 loc) · 4.57 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

# Pyrogram - Telegram MTProto API Client Library for Python

# This file is part of Pyrogram.

# Pyrogram is free software: you can redistribute it and/or modify

# it under the terms of the GNU Lesser General Public License as published

# by the Free Software Foundation, either version 3 of the License, or

# (at your option) any later version.

# Pyrogram is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

# GNU Lesser General Public License for more details.

# You should have received a copy of the GNU Lesser General Public License

# along with Pyrogram. If not, see <http://www.gnu.org/licenses/>.

import re

from collections import OrderedDict

import pyrogram

from pyrogram.api.types import (

MessageEntityBold as Bold,

MessageEntityItalic as Italic,

MessageEntityCode as Code,

MessageEntityTextUrl as Url,

MessageEntityPre as Pre,

MessageEntityMentionName as MentionInvalid,

InputMessageEntityMentionName as Mention,

)

from pyrogram.errors import PeerIdInvalid

from . import utils

class HTML:

HTML_RE = re.compile(r"<(\w+)(?: href=([\"'])([^<]+)\2)?>([^>]+)</\1>")

MENTION_RE = re.compile(r"tg://user\?id=(\d+)")

def __init__(self, client: "pyrogram.BaseClient" = None):

self.client = client

def parse(self, message: str):

entities = []

message = utils.add_surrogates(str(message or ""))

offset = 0

for match in self.HTML_RE.finditer(message):

start = match.start() - offset

style, url, body = match.group(1, 3, 4)

if url:

mention = self.MENTION_RE.match(url)

if mention:

user_id = int(mention.group(1))

try:

input_user = self.client.resolve_peer(user_id)

except PeerIdInvalid:

input_user = None

entity = (

Mention(offset=start, length=len(body), user_id=input_user)

if input_user else MentionInvalid(offset=start, length=len(body), user_id=user_id)

)

else:

entity = Url(offset=start, length=len(body), url=url)

else:

if style == "b" or style == "strong":

entity = Bold(offset=start, length=len(body))

elif style == "i" or style == "em":

entity = Italic(offset=start, length=len(body))

elif style == "code":

entity = Code(offset=start, length=len(body))

elif style == "pre":

entity = Pre(offset=start, length=len(body), language="")

else:

continue

entities.append(entity)

message = message.replace(match.group(), body)

offset += len(style) * 2 + 5 + (len(url) + 8 if url else 0)

# TODO: OrderedDict to be removed in Python3.6

return OrderedDict([

("message", utils.remove_surrogates(message)),

("entities", entities)

])

def unparse(self, message: str, entities: list):

message = utils.add_surrogates(message).strip()

offset = 0

for entity in entities:

start = entity.offset + offset

type = entity.type

url = entity.url

user = entity.user

sub = message[start: start + entity.length]

if type == "bold":

style = "b"

elif type == "italic":

style = "i"

elif type == "code":

style = "code"

elif type == "pre":

style = "pre"

elif type == "text_link":

offset += 15 + len(url)

message = message[:start] + message[start:].replace(

sub, "<a href=\"{}\">{}</a>".format(url, sub), 1)

continue

elif type == "text_mention":

offset += 28 + len(str(user.id))

message = message[:start] + message[start:].replace(

sub, "<a href=\"tg://user?id={}\">{}</a>".format(user.id, sub), 1)

continue

else:

continue

offset += len(style) * 2 + 5

message = message[:start] + message[start:].replace(

sub, "<{0}>{1}</{0}>".format(style, sub), 1)

return utils.remove_surrogates(message)

Sunbelt Computer Software

PL/B Language Development and Support

Navigation Menu

Sunbelt Computer Software

PL/B Language Development and Support

FilesExpand file tree

html.py

Latest commit

History

html.py

File metadata and controls