Fix most warnings
This commit is contained in:
@@ -106,7 +106,7 @@ class AppService:
|
|||||||
|
|
||||||
try:
|
try:
|
||||||
response = self.query_user(user_id)
|
response = self.query_user(user_id)
|
||||||
except:
|
except Exception:
|
||||||
self.log.exception("Exception in user query handler")
|
self.log.exception("Exception in user query handler")
|
||||||
return web.Response(status=500)
|
return web.Response(status=500)
|
||||||
|
|
||||||
@@ -122,7 +122,7 @@ class AppService:
|
|||||||
|
|
||||||
try:
|
try:
|
||||||
response = self.query_alias(alias)
|
response = self.query_alias(alias)
|
||||||
except:
|
except Exception:
|
||||||
self.log.exception("Exception in alias query handler")
|
self.log.exception("Exception in alias query handler")
|
||||||
return web.Response(status=500)
|
return web.Response(status=500)
|
||||||
|
|
||||||
@@ -164,7 +164,7 @@ class AppService:
|
|||||||
for handler in self.event_handlers:
|
for handler in self.event_handlers:
|
||||||
try:
|
try:
|
||||||
handler(event)
|
handler(event)
|
||||||
except:
|
except Exception:
|
||||||
self.log.exception("Exception in Matrix event handler")
|
self.log.exception("Exception in Matrix event handler")
|
||||||
|
|
||||||
def matrix_event_handler(self, func):
|
def matrix_event_handler(self, func):
|
||||||
|
|||||||
@@ -25,11 +25,8 @@ from matrix_client.errors import MatrixRequestError
|
|||||||
class HTTPAPI(MatrixHttpApi):
|
class HTTPAPI(MatrixHttpApi):
|
||||||
def __init__(self, base_url, domain=None, bot_mxid=None, token=None, identity=None, log=None,
|
def __init__(self, base_url, domain=None, bot_mxid=None, token=None, identity=None, log=None,
|
||||||
state_store=None):
|
state_store=None):
|
||||||
self.base_url = base_url
|
super().__init__(base_url, token, identity)
|
||||||
self.domain = domain
|
self.domain = domain
|
||||||
self.token = token
|
|
||||||
self.identity = identity
|
|
||||||
self.txn_id = 0
|
|
||||||
self.bot_mxid = bot_mxid
|
self.bot_mxid = bot_mxid
|
||||||
self.intent_log = log.getChild("intent")
|
self.intent_log = log.getChild("intent")
|
||||||
self.log = log.getChild("api")
|
self.log = log.getChild("api")
|
||||||
@@ -51,25 +48,17 @@ class HTTPAPI(MatrixHttpApi):
|
|||||||
def intent(self, user):
|
def intent(self, user):
|
||||||
return IntentAPI(user, self.user(user), self, self.state_store, self.intent_log)
|
return IntentAPI(user, self.user(user), self, self.state_store, self.intent_log)
|
||||||
|
|
||||||
def _send(self, method, path, content=None, query_params={}, headers={},
|
def _send(self, method, path, content=None, query_params=None, headers=None,
|
||||||
api_path="/_matrix/client/r0"):
|
api_path="/_matrix/client/r0"):
|
||||||
if not query_params:
|
if not query_params:
|
||||||
query_params = {}
|
query_params = {}
|
||||||
query_params["user_id"] = self.identity
|
query_params["user_id"] = self.identity
|
||||||
log_content = content if not isinstance(content, bytes) else f"<{len(content)} bytes>"
|
log_content = content if not isinstance(content, bytes) else f"<{len(content)} bytes>"
|
||||||
self.log.debug("%s %s %s", method, path, log_content)
|
self.log.debug("%s %s %s", method, path, log_content)
|
||||||
return super()._send(method, path, content, query_params, headers, api_path=api_path)
|
return super()._send(method, path, content, query_params, headers or {}, api_path=api_path)
|
||||||
|
|
||||||
def create_room(self, alias=None, is_public=False, name=None, topic=None, is_direct=False,
|
def create_room(self, alias=None, is_public=False, name=None, topic=None, is_direct=False,
|
||||||
invitees=(), initial_state=[]):
|
invitees=(), initial_state=None):
|
||||||
"""Perform /createRoom.
|
|
||||||
Args:
|
|
||||||
alias (str): Optional. The room alias name to set for this room.
|
|
||||||
is_public (bool): Optional. The public/private visibility.
|
|
||||||
name (str): Optional. The name for the room.
|
|
||||||
topic (str): Optional. The topic for the room.
|
|
||||||
invitees (list<str>): Optional. The list of user IDs to invite.
|
|
||||||
"""
|
|
||||||
content = {
|
content = {
|
||||||
"visibility": "public" if is_public else "private"
|
"visibility": "public" if is_public else "private"
|
||||||
}
|
}
|
||||||
@@ -106,9 +95,10 @@ class HTTPAPI(MatrixHttpApi):
|
|||||||
|
|
||||||
class ChildHTTPAPI(HTTPAPI):
|
class ChildHTTPAPI(HTTPAPI):
|
||||||
def __init__(self, user, parent):
|
def __init__(self, user, parent):
|
||||||
self.identity = user
|
|
||||||
self.token = parent.token
|
|
||||||
self.base_url = parent.base_url
|
self.base_url = parent.base_url
|
||||||
|
self.token = parent.token
|
||||||
|
self.identity = user
|
||||||
|
self.validate_cert = True
|
||||||
self.validate_cert = parent.validate_cert
|
self.validate_cert = parent.validate_cert
|
||||||
self.log = parent.log
|
self.log = parent.log
|
||||||
self.domain = parent.domain
|
self.domain = parent.domain
|
||||||
@@ -133,7 +123,7 @@ def matrix_error_code(err):
|
|||||||
try:
|
try:
|
||||||
data = json.loads(err.content)
|
data = json.loads(err.content)
|
||||||
return data["errcode"]
|
return data["errcode"]
|
||||||
except:
|
except Exception:
|
||||||
return err.content
|
return err.content
|
||||||
|
|
||||||
|
|
||||||
@@ -189,10 +179,10 @@ class IntentAPI:
|
|||||||
# region Room actions
|
# region Room actions
|
||||||
|
|
||||||
def create_room(self, alias=None, is_public=False, name=None, topic=None, is_direct=False,
|
def create_room(self, alias=None, is_public=False, name=None, topic=None, is_direct=False,
|
||||||
invitees=(), initial_state=[]):
|
invitees=(), initial_state=None):
|
||||||
self.ensure_registered()
|
self.ensure_registered()
|
||||||
return self.client.create_room(alias, is_public, name, topic, is_direct, invitees,
|
return self.client.create_room(alias, is_public, name, topic, is_direct, invitees,
|
||||||
initial_state)
|
initial_state or {})
|
||||||
|
|
||||||
def invite(self, room_id, user_id):
|
def invite(self, room_id, user_id):
|
||||||
self.ensure_joined(room_id)
|
self.ensure_joined(room_id)
|
||||||
@@ -251,31 +241,31 @@ class IntentAPI:
|
|||||||
def send_emote(self, room_id, text, html=None):
|
def send_emote(self, room_id, text, html=None):
|
||||||
return self.send_text(room_id, text, html, "m.emote")
|
return self.send_text(room_id, text, html, "m.emote")
|
||||||
|
|
||||||
def send_image(self, room_id, url, info={}, text=None):
|
def send_image(self, room_id, url, info=None, text=None):
|
||||||
return self.send_file(room_id, url, info, text, "m.image")
|
return self.send_file(room_id, url, info or {}, text, "m.image")
|
||||||
|
|
||||||
def send_file(self, room_id, url, info={}, text=None, type="m.file"):
|
def send_file(self, room_id, url, info=None, text=None, file_type="m.file"):
|
||||||
return self.send_message(room_id, {
|
return self.send_message(room_id, {
|
||||||
"msgtype": type,
|
"msgtype": file_type,
|
||||||
"url": url,
|
"url": url,
|
||||||
"body": text or "Uploaded file",
|
"body": text or "Uploaded file",
|
||||||
"info": info,
|
"info": info or {},
|
||||||
})
|
})
|
||||||
|
|
||||||
def send_text(self, room_id, text, html=None, type="m.text"):
|
def send_text(self, room_id, text, html=None, msgtype="m.text"):
|
||||||
if html:
|
if html:
|
||||||
if not text:
|
if not text:
|
||||||
text = html
|
text = html
|
||||||
return self.send_message(room_id, {
|
return self.send_message(room_id, {
|
||||||
"body": text,
|
"body": text,
|
||||||
"msgtype": type,
|
"msgtype": msgtype,
|
||||||
"format": "org.matrix.custom.html",
|
"format": "org.matrix.custom.html",
|
||||||
"formatted_body": html or text,
|
"formatted_body": html or text,
|
||||||
})
|
})
|
||||||
else:
|
else:
|
||||||
return self.send_message(room_id, {
|
return self.send_message(room_id, {
|
||||||
"body": text,
|
"body": text,
|
||||||
"msgtype": type,
|
"msgtype": msgtype,
|
||||||
})
|
})
|
||||||
|
|
||||||
def send_message(self, room_id, body):
|
def send_message(self, room_id, body):
|
||||||
@@ -290,15 +280,15 @@ class IntentAPI:
|
|||||||
self.ensure_joined(room_id)
|
self.ensure_joined(room_id)
|
||||||
return self.client.kick_user(room_id, user_id, message)
|
return self.client.kick_user(room_id, user_id, message)
|
||||||
|
|
||||||
def send_event(self, room_id, type, body, txn_id=None):
|
def send_event(self, room_id, event_type, body, txn_id=None):
|
||||||
self.ensure_joined(room_id)
|
self.ensure_joined(room_id)
|
||||||
self._ensure_has_power_level_for(room_id, type)
|
self._ensure_has_power_level_for(room_id, event_type)
|
||||||
return self.client.send_message_event(room_id, type, body, txn_id)
|
return self.client.send_message_event(room_id, event_type, body, txn_id)
|
||||||
|
|
||||||
def send_state_event(self, room_id, type, body, state_key=""):
|
def send_state_event(self, room_id, event_type, body, state_key=""):
|
||||||
self.ensure_joined(room_id)
|
self.ensure_joined(room_id)
|
||||||
self._ensure_has_power_level_for(room_id, type)
|
self._ensure_has_power_level_for(room_id, event_type)
|
||||||
return self.client.send_state_event(room_id, type, body, state_key)
|
return self.client.send_state_event(room_id, event_type, body, state_key)
|
||||||
|
|
||||||
def join_room(self, room_id):
|
def join_room(self, room_id):
|
||||||
return self.ensure_joined(room_id, ignore_cache=True)
|
return self.ensure_joined(room_id, ignore_cache=True)
|
||||||
@@ -359,7 +349,8 @@ class IntentAPI:
|
|||||||
elif not self.bot:
|
elif not self.bot:
|
||||||
self.log.warning(
|
self.log.warning(
|
||||||
f"Power level of {self.mxid} is not enough for {event_type} in {room_id}")
|
f"Power level of {self.mxid} is not enough for {event_type} in {room_id}")
|
||||||
# raise IntentError(f"Power level of {self.mxid} is not enough for {event_type} in {room_id}")
|
# raise IntentError(f"Power level of {self.mxid} is not enough"
|
||||||
|
# + f"for {event_type} in {room_id}")
|
||||||
return
|
return
|
||||||
# TODO implement
|
# TODO implement
|
||||||
|
|
||||||
|
|||||||
@@ -42,13 +42,13 @@ class StateStore:
|
|||||||
def load(self, file):
|
def load(self, file):
|
||||||
if isinstance(file, str):
|
if isinstance(file, str):
|
||||||
try:
|
try:
|
||||||
input = open(file, "r")
|
input_source = open(file, "r")
|
||||||
except FileNotFoundError:
|
except FileNotFoundError:
|
||||||
return
|
return
|
||||||
else:
|
else:
|
||||||
input = file
|
input_source = file
|
||||||
|
|
||||||
data = json.load(input)
|
data = json.load(input_source)
|
||||||
if "registrations" in data:
|
if "registrations" in data:
|
||||||
self.registrations = set(data["registrations"])
|
self.registrations = set(data["registrations"])
|
||||||
if "memberships" in data:
|
if "memberships" in data:
|
||||||
@@ -57,7 +57,7 @@ class StateStore:
|
|||||||
self.power_levels = data["power_levels"]
|
self.power_levels = data["power_levels"]
|
||||||
|
|
||||||
if isinstance(file, str):
|
if isinstance(file, str):
|
||||||
input.close()
|
input_source.close()
|
||||||
|
|
||||||
def _autosave(self):
|
def _autosave(self):
|
||||||
if self.autosave_file:
|
if self.autosave_file:
|
||||||
@@ -104,7 +104,7 @@ class StateStore:
|
|||||||
return has >= required
|
return has >= required
|
||||||
|
|
||||||
def set_power_level(self, room, user, level):
|
def set_power_level(self, room, user, level):
|
||||||
if not room in self.power_levels:
|
if room not in self.power_levels:
|
||||||
self.power_levels[room] = {
|
self.power_levels[room] = {
|
||||||
"users": {},
|
"users": {},
|
||||||
"events": {},
|
"events": {},
|
||||||
|
|||||||
@@ -47,7 +47,7 @@ class CommandHandler:
|
|||||||
with self.handler(sender, room, command, args, is_management, is_portal) as handle_command:
|
with self.handler(sender, room, command, args, is_management, is_portal) as handle_command:
|
||||||
try:
|
try:
|
||||||
handle_command(self, sender, args)
|
handle_command(self, sender, args)
|
||||||
except:
|
except Exception:
|
||||||
self.reply("Fatal error while handling command. Check logs for more details.")
|
self.reply("Fatal error while handling command. Check logs for more details.")
|
||||||
self.log.exception(f"Fatal error handling command "
|
self.log.exception(f"Fatal error handling command "
|
||||||
+ f"'$cmdprefix {command} {''.join(args)}' from {sender.mxid}")
|
+ f"'$cmdprefix {command} {''.join(args)}' from {sender.mxid}")
|
||||||
@@ -158,7 +158,7 @@ class CommandHandler:
|
|||||||
}
|
}
|
||||||
return self.reply("Your account has two-factor authentication."
|
return self.reply("Your account has two-factor authentication."
|
||||||
"Please send your password here.")
|
"Please send your password here.")
|
||||||
except:
|
except Exception:
|
||||||
self.log.exception()
|
self.log.exception()
|
||||||
return self.reply("Unhandled exception while sending code."
|
return self.reply("Unhandled exception while sending code."
|
||||||
"Check console for more details.")
|
"Check console for more details.")
|
||||||
@@ -177,7 +177,7 @@ class CommandHandler:
|
|||||||
return self.reply(f"Successfully logged in as @{user.username}")
|
return self.reply(f"Successfully logged in as @{user.username}")
|
||||||
except PasswordHashInvalidError:
|
except PasswordHashInvalidError:
|
||||||
return self.reply("Incorrect password.")
|
return self.reply("Incorrect password.")
|
||||||
except:
|
except Exception:
|
||||||
self.log.exception()
|
self.log.exception()
|
||||||
return self.reply("Unhandled exception while sending password. "
|
return self.reply("Unhandled exception while sending password. "
|
||||||
"Check console for more details.")
|
"Check console for more details.")
|
||||||
@@ -199,9 +199,9 @@ class CommandHandler:
|
|||||||
return self.reply("**Usage:** `$cmdprefix+sp search [-r|--remote] <query>")
|
return self.reply("**Usage:** `$cmdprefix+sp search [-r|--remote] <query>")
|
||||||
elif not sender.logged_in:
|
elif not sender.logged_in:
|
||||||
return self.reply("This command requires you to be logged in.")
|
return self.reply("This command requires you to be logged in.")
|
||||||
force_remote = False
|
# force_remote = False
|
||||||
if args[0] in {"-r", "--remote"}:
|
if args[0] in {"-r", "--remote"}:
|
||||||
force_remote = True
|
# force_remote = True
|
||||||
args.pop(0)
|
args.pop(0)
|
||||||
query = " ".join(args)
|
query = " ".join(args)
|
||||||
if len(query) < 5:
|
if len(query) < 5:
|
||||||
@@ -298,7 +298,7 @@ class CommandHandler:
|
|||||||
if arg.startswith("joinchat/"):
|
if arg.startswith("joinchat/"):
|
||||||
invite_hash = arg[len("joinchat/"):]
|
invite_hash = arg[len("joinchat/"):]
|
||||||
try:
|
try:
|
||||||
check = sender.client(CheckChatInviteRequest(invite_hash))
|
sender.client(CheckChatInviteRequest(invite_hash))
|
||||||
except InviteHashInvalidError:
|
except InviteHashInvalidError:
|
||||||
return self.reply("Invalid invite link.")
|
return self.reply("Invalid invite link.")
|
||||||
except InviteHashExpiredError:
|
except InviteHashExpiredError:
|
||||||
|
|||||||
@@ -14,16 +14,16 @@
|
|||||||
#
|
#
|
||||||
# You should have received a copy of the GNU General Public License
|
# You should have received a copy of the GNU General Public License
|
||||||
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
import ruamel.yaml
|
from ruamel.yaml import YAML
|
||||||
import random
|
import random
|
||||||
import string
|
import string
|
||||||
|
|
||||||
yaml = ruamel.yaml.YAML()
|
yaml = YAML()
|
||||||
|
|
||||||
|
|
||||||
class DictWithRecursion:
|
class DictWithRecursion:
|
||||||
def __init__(self, data={}):
|
def __init__(self, data=None):
|
||||||
self._data = data
|
self._data = data or {}
|
||||||
|
|
||||||
def _recursive_get(self, data, key, default_value):
|
def _recursive_get(self, data, key, default_value):
|
||||||
if '.' in key:
|
if '.' in key:
|
||||||
|
|||||||
@@ -53,12 +53,12 @@ class MatrixParser(HTMLParser):
|
|||||||
self._open_tags.appendleft(tag)
|
self._open_tags.appendleft(tag)
|
||||||
self._open_tags_meta.appendleft(0)
|
self._open_tags_meta.appendleft(0)
|
||||||
attrs = dict(attrs)
|
attrs = dict(attrs)
|
||||||
EntityType = None
|
entity_type = None
|
||||||
args = {}
|
args = {}
|
||||||
if tag == "strong" or tag == "b":
|
if tag == "strong" or tag == "b":
|
||||||
EntityType = MessageEntityBold
|
entity_type = MessageEntityBold
|
||||||
elif tag == "em" or tag == "i":
|
elif tag == "em" or tag == "i":
|
||||||
EntityType = MessageEntityItalic
|
entity_type = MessageEntityItalic
|
||||||
elif tag == "code":
|
elif tag == "code":
|
||||||
try:
|
try:
|
||||||
pre = self._building_entities["pre"]
|
pre = self._building_entities["pre"]
|
||||||
@@ -67,9 +67,9 @@ class MatrixParser(HTMLParser):
|
|||||||
except KeyError:
|
except KeyError:
|
||||||
pass
|
pass
|
||||||
except KeyError:
|
except KeyError:
|
||||||
EntityType = MessageEntityCode
|
entity_type = MessageEntityCode
|
||||||
elif tag == "pre":
|
elif tag == "pre":
|
||||||
EntityType = MessageEntityPre
|
entity_type = MessageEntityPre
|
||||||
args["language"] = ""
|
args["language"] = ""
|
||||||
elif tag == "a":
|
elif tag == "a":
|
||||||
try:
|
try:
|
||||||
@@ -86,13 +86,13 @@ class MatrixParser(HTMLParser):
|
|||||||
if not user:
|
if not user:
|
||||||
return
|
return
|
||||||
if user.username:
|
if user.username:
|
||||||
EntityType = MessageEntityMention
|
entity_type = MessageEntityMention
|
||||||
url = f"@{user.username}"
|
url = f"@{user.username}"
|
||||||
else:
|
else:
|
||||||
EntityType = MessageEntityMentionName
|
entity_type = MessageEntityMentionName
|
||||||
args["user_id"] = user.tgid
|
args["user_id"] = user.tgid
|
||||||
elif reply and self._user_id and (
|
elif reply and self._user_id and (len(self.entities) == 0
|
||||||
len(self.entities) == 0 and len(self._building_entities) == 0):
|
and len(self._building_entities) == 0):
|
||||||
room_id = reply.group(1)
|
room_id = reply.group(1)
|
||||||
message_id = reply.group(2)
|
message_id = reply.group(2)
|
||||||
message = DBMessage.query.filter(DBMessage.mxid == message_id
|
message = DBMessage.query.filter(DBMessage.mxid == message_id
|
||||||
@@ -100,25 +100,25 @@ class MatrixParser(HTMLParser):
|
|||||||
and DBMessage.user == self._user_id).one_or_none()
|
and DBMessage.user == self._user_id).one_or_none()
|
||||||
if not message:
|
if not message:
|
||||||
return
|
return
|
||||||
EntityType = MessageEntityReply
|
entity_type = MessageEntityReply
|
||||||
args["msg_id"] = message.tgid
|
args["msg_id"] = message.tgid
|
||||||
self._building_reply = True
|
self._building_reply = True
|
||||||
url = None
|
url = None
|
||||||
elif url.startswith("mailto:"):
|
elif url.startswith("mailto:"):
|
||||||
url = url[len("mailto:"):]
|
url = url[len("mailto:"):]
|
||||||
EntityType = MessageEntityEmail
|
entity_type = MessageEntityEmail
|
||||||
else:
|
else:
|
||||||
if self.get_starttag_text() == url:
|
if self.get_starttag_text() == url:
|
||||||
EntityType = MessageEntityUrl
|
entity_type = MessageEntityUrl
|
||||||
else:
|
else:
|
||||||
EntityType = MessageEntityTextUrl
|
entity_type = MessageEntityTextUrl
|
||||||
args["url"] = url
|
args["url"] = url
|
||||||
url = None
|
url = None
|
||||||
self._open_tags_meta.popleft()
|
self._open_tags_meta.popleft()
|
||||||
self._open_tags_meta.appendleft(url)
|
self._open_tags_meta.appendleft(url)
|
||||||
|
|
||||||
if EntityType and tag not in self._building_entities:
|
if entity_type and tag not in self._building_entities:
|
||||||
self._building_entities[tag] = EntityType(offset=len(self.text), length=0, **args)
|
self._building_entities[tag] = entity_type(offset=len(self.text), length=0, **args)
|
||||||
|
|
||||||
def _list_depth(self):
|
def _list_depth(self):
|
||||||
depth = 0
|
depth = 0
|
||||||
@@ -183,7 +183,7 @@ def matrix_to_telegram(html, user_id=None):
|
|||||||
parser = MatrixParser(user_id)
|
parser = MatrixParser(user_id)
|
||||||
parser.feed(html)
|
parser.feed(html)
|
||||||
return parser.text, parser.entities
|
return parser.text, parser.entities
|
||||||
except:
|
except Exception:
|
||||||
log.exception("Failed to convert Matrix format:\nhtml=%s", html)
|
log.exception("Failed to convert Matrix format:\nhtml=%s", html)
|
||||||
|
|
||||||
|
|
||||||
@@ -197,18 +197,20 @@ def telegram_event_to_matrix(evt, source):
|
|||||||
if evt.fwd_from:
|
if evt.fwd_from:
|
||||||
if not html:
|
if not html:
|
||||||
html = escape(text)
|
html = escape(text)
|
||||||
id = evt.fwd_from.from_id
|
from_id = evt.fwd_from.from_id
|
||||||
user = u.User.get_by_tgid(id)
|
user = u.User.get_by_tgid(from_id)
|
||||||
if user:
|
if user:
|
||||||
fwd_from = f"<a href='https://matrix.to/#/{user.mxid}'>{user.mxid}</a>"
|
fwd_from = f"<a href='https://matrix.to/#/{user.mxid}'>{user.mxid}</a>"
|
||||||
else:
|
else:
|
||||||
puppet = p.Puppet.get(id, create=False)
|
puppet = p.Puppet.get(from_id, create=False)
|
||||||
if puppet and puppet.displayname:
|
if puppet and puppet.displayname:
|
||||||
fwd_from = f"<a href='https://matrix.to/#/{puppet.mxid}'>{puppet.displayname}</a>"
|
fwd_from = f"<a href='https://matrix.to/#/{puppet.mxid}'>{puppet.displayname}</a>"
|
||||||
else:
|
else:
|
||||||
user = source.client.get_entity(id)
|
user = source.client.get_entity(from_id)
|
||||||
if user:
|
if user:
|
||||||
fwd_from = p.Puppet.get_displayname(user, format=False)
|
fwd_from = p.Puppet.get_displayname(user, format=False)
|
||||||
|
else:
|
||||||
|
fwd_from = None
|
||||||
if not fwd_from:
|
if not fwd_from:
|
||||||
fwd_from = "Unknown user"
|
fwd_from = "Unknown user"
|
||||||
html = (f"Forwarded message from <b>{fwd_from}</b><br/>"
|
html = (f"Forwarded message from <b>{fwd_from}</b><br/>"
|
||||||
@@ -232,7 +234,7 @@ def telegram_event_to_matrix(evt, source):
|
|||||||
def telegram_to_matrix(text, entities):
|
def telegram_to_matrix(text, entities):
|
||||||
try:
|
try:
|
||||||
return _telegram_to_matrix(text, entities)
|
return _telegram_to_matrix(text, entities)
|
||||||
except:
|
except Exception:
|
||||||
log.exception("Failed to convert Telegram format:\n"
|
log.exception("Failed to convert Telegram format:\n"
|
||||||
"message=%s\n"
|
"message=%s\n"
|
||||||
"entities=%s",
|
"entities=%s",
|
||||||
|
|||||||
+19
-16
@@ -87,12 +87,14 @@ class Portal:
|
|||||||
self._main_intent = puppet.intent if direct else self.az.intent
|
self._main_intent = puppet.intent if direct else self.az.intent
|
||||||
return self._main_intent
|
return self._main_intent
|
||||||
|
|
||||||
def invite_matrix(self, users=[]):
|
def invite_matrix(self, users):
|
||||||
if isinstance(users, str):
|
if isinstance(users, str):
|
||||||
self.main_intent.invite(self.mxid, users)
|
self.main_intent.invite(self.mxid, users)
|
||||||
else:
|
elif isinstance(users, list):
|
||||||
for user in users:
|
for user in users:
|
||||||
self.main_intent.invite(self.mxid, user)
|
self.main_intent.invite(self.mxid, user)
|
||||||
|
else:
|
||||||
|
raise ValueError("Invalid invite identifier given to invite_matrix()")
|
||||||
|
|
||||||
def update_after_create(self, user, entity, direct, puppet=None):
|
def update_after_create(self, user, entity, direct, puppet=None):
|
||||||
if not direct:
|
if not direct:
|
||||||
@@ -106,7 +108,7 @@ class Portal:
|
|||||||
puppet.update_info(user, entity)
|
puppet.update_info(user, entity)
|
||||||
puppet.intent.join_room(self.mxid)
|
puppet.intent.join_room(self.mxid)
|
||||||
|
|
||||||
def create_matrix_room(self, user, entity=None, invites=[], update_if_exists=True):
|
def create_matrix_room(self, user, entity=None, invites=None, update_if_exists=True):
|
||||||
if not entity:
|
if not entity:
|
||||||
entity = user.client.get_entity(self.peer)
|
entity = user.client.get_entity(self.peer)
|
||||||
self.log.debug("Fetched data: %s", entity)
|
self.log.debug("Fetched data: %s", entity)
|
||||||
@@ -115,7 +117,7 @@ class Portal:
|
|||||||
if self.mxid:
|
if self.mxid:
|
||||||
if update_if_exists:
|
if update_if_exists:
|
||||||
self.update_after_create(user, entity, direct)
|
self.update_after_create(user, entity, direct)
|
||||||
self.invite_matrix(invites)
|
self.invite_matrix(invites or [])
|
||||||
return self.mxid
|
return self.mxid
|
||||||
|
|
||||||
self.log.debug(f"Creating room for {self.tgid_log}")
|
self.log.debug(f"Creating room for {self.tgid_log}")
|
||||||
@@ -141,8 +143,8 @@ class Portal:
|
|||||||
if alias:
|
if alias:
|
||||||
# TODO properly handle existing room aliases
|
# TODO properly handle existing room aliases
|
||||||
intent.remove_room_alias(alias)
|
intent.remove_room_alias(alias)
|
||||||
room = intent.create_room(alias=alias, is_public=public, invitees=invites, name=self.title,
|
room = intent.create_room(alias=alias, is_public=public, invitees=invites or [],
|
||||||
is_direct=direct)
|
name=self.title, is_direct=direct)
|
||||||
if not room:
|
if not room:
|
||||||
raise Exception(f"Failed to create room for {self.tgid_log}")
|
raise Exception(f"Failed to create room for {self.tgid_log}")
|
||||||
|
|
||||||
@@ -166,7 +168,7 @@ class Portal:
|
|||||||
return config.get("bridge.alias_template", "telegram_{groupname}").format(
|
return config.get("bridge.alias_template", "telegram_{groupname}").format(
|
||||||
groupname=username)
|
groupname=username)
|
||||||
|
|
||||||
def sync_telegram_users(self, source, users=[]):
|
def sync_telegram_users(self, source, users):
|
||||||
for entity in users:
|
for entity in users:
|
||||||
puppet = p.Puppet.get(entity.id)
|
puppet = p.Puppet.get(entity.id)
|
||||||
puppet.update_info(source, entity)
|
puppet.update_info(source, entity)
|
||||||
@@ -243,8 +245,8 @@ class Portal:
|
|||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def _get_largest_photo_size(photo):
|
def _get_largest_photo_size(photo):
|
||||||
return max(photo.sizes, key=(lambda photo: (
|
return max(photo.sizes, key=(lambda photo2: (
|
||||||
len(photo.bytes) if isinstance(photo, PhotoCachedSize) else photo.size)))
|
len(photo2.bytes) if isinstance(photo2, PhotoCachedSize) else photo2.size)))
|
||||||
|
|
||||||
def update_avatar(self, user, photo):
|
def update_avatar(self, user, photo):
|
||||||
photo_id = f"{photo.volume_id}-{photo.local_id}"
|
photo_id = f"{photo.volume_id}-{photo.local_id}"
|
||||||
@@ -293,8 +295,8 @@ class Portal:
|
|||||||
# endregion
|
# endregion
|
||||||
# region Matrix event handling
|
# region Matrix event handling
|
||||||
|
|
||||||
def _get_file_meta(self, body, mime):
|
@staticmethod
|
||||||
file_name = None
|
def _get_file_meta(body, mime):
|
||||||
try:
|
try:
|
||||||
current_extension = body[body.rindex("."):]
|
current_extension = body[body.rindex("."):]
|
||||||
if mimetypes.types_map[current_extension] == mime:
|
if mimetypes.types_map[current_extension] == mime:
|
||||||
@@ -510,14 +512,14 @@ class Portal:
|
|||||||
sender.intent.set_typing(self.mxid, is_typing=False)
|
sender.intent.set_typing(self.mxid, is_typing=False)
|
||||||
return sender.intent.send_image(self.mxid, uploaded["content_uri"], info=info, text=name)
|
return sender.intent.send_image(self.mxid, uploaded["content_uri"], info=info, text=name)
|
||||||
|
|
||||||
@staticmethod
|
def convert_webp(self, file, to="png"):
|
||||||
def convert_webp(file, to="png"):
|
|
||||||
try:
|
try:
|
||||||
image = Image.open(BytesIO(file)).convert("RGBA")
|
image = Image.open(BytesIO(file)).convert("RGBA")
|
||||||
new_file = BytesIO()
|
new_file = BytesIO()
|
||||||
image.save(new_file, to)
|
image.save(new_file, to)
|
||||||
return f"image/{to}", new_file.getvalue()
|
return f"image/{to}", new_file.getvalue()
|
||||||
except:
|
except Exception:
|
||||||
|
self.log.exception(f"Failed to convert webp to {to}")
|
||||||
return "image/webp", file
|
return "image/webp", file
|
||||||
|
|
||||||
def handle_telegram_document(self, source, sender, media):
|
def handle_telegram_document(self, source, sender, media):
|
||||||
@@ -551,7 +553,7 @@ class Portal:
|
|||||||
type = "m.image"
|
type = "m.image"
|
||||||
sender.intent.set_typing(self.mxid, is_typing=False)
|
sender.intent.set_typing(self.mxid, is_typing=False)
|
||||||
return sender.intent.send_file(self.mxid, uploaded["content_uri"], info=info, text=name,
|
return sender.intent.send_file(self.mxid, uploaded["content_uri"], info=info, text=name,
|
||||||
type=type)
|
file_type=type)
|
||||||
|
|
||||||
def handle_telegram_location(self, source, sender, location):
|
def handle_telegram_location(self, source, sender, location):
|
||||||
long = location.long
|
long = location.long
|
||||||
@@ -566,7 +568,8 @@ class Portal:
|
|||||||
url = f"https://maps.google.com/?q={lat},{long}"
|
url = f"https://maps.google.com/?q={lat},{long}"
|
||||||
|
|
||||||
formatted_body = f"Location: <a href='{url}'>{body}</a>"
|
formatted_body = f"Location: <a href='{url}'>{body}</a>"
|
||||||
# At least Riot ignores formatting in m.location messages, so we'll add a plaintext link.
|
# At least riot-web ignores formatting in m.location messages,
|
||||||
|
# so we'll add a plaintext link.
|
||||||
body = f"Location: {body}\n{url}"
|
body = f"Location: {body}\n{url}"
|
||||||
|
|
||||||
return sender.intent.send_message(self.mxid, {
|
return sender.intent.send_message(self.mxid, {
|
||||||
|
|||||||
@@ -75,6 +75,7 @@ class Puppet:
|
|||||||
}
|
}
|
||||||
preferences = config.get("bridge.displayname_preference",
|
preferences = config.get("bridge.displayname_preference",
|
||||||
["full name", "username", "phone"])
|
["full name", "username", "phone"])
|
||||||
|
name = None
|
||||||
for preference in preferences:
|
for preference in preferences:
|
||||||
name = data[preference]
|
name = data[preference]
|
||||||
if name:
|
if name:
|
||||||
|
|||||||
@@ -62,7 +62,7 @@ class User:
|
|||||||
return self.logged_in and self.whitelisted
|
return self.logged_in and self.whitelisted
|
||||||
|
|
||||||
def get_input_entity(self, user):
|
def get_input_entity(self, user):
|
||||||
return user.client.get_input_entity(InputUser(user_id=self.tgid))
|
return user.client.get_input_entity(InputUser(user_id=self.tgid, access_hash=0))
|
||||||
|
|
||||||
# region Database conversion
|
# region Database conversion
|
||||||
|
|
||||||
@@ -202,7 +202,7 @@ class User:
|
|||||||
def update_catch(self, update):
|
def update_catch(self, update):
|
||||||
try:
|
try:
|
||||||
self.update(update)
|
self.update(update)
|
||||||
except:
|
except Exception:
|
||||||
self.log.exception("Failed to handle Telegram update")
|
self.log.exception("Failed to handle Telegram update")
|
||||||
|
|
||||||
def update(self, update):
|
def update(self, update):
|
||||||
|
|||||||
Reference in New Issue
Block a user