iartamonov/mautrix-telegram
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Fix possible errors

Browse Source
This commit is contained in:
Tulir Asokan
2018-02-21 23:35:59 +02:00
parent 03734a6745
commit 2d13c30a26
2 changed files with 11 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mautrix_telegram/bot.py
+6
View File
@@ -65,6 +65,12 @@ class Bot(AbstractUser):
except (ChannelPrivateError, ChannelInvalidError): except (ChannelPrivateError, ChannelInvalidError):
self.remove_chat(id.channel_id) self.remove_chat(id.channel_id)
def register_portal(self, portal):
self.add_chat(portal.tgid, portal.peer_type)
def unregister_portal(self, portal):
self.remove_chat(portal.tgid)
def add_chat(self, id, type): def add_chat(self, id, type):
if id not in self.chats: if id not in self.chats:
self.chats[id] = type self.chats[id] = type
mautrix_telegram/portal.py
+5 -2
View File
@@ -491,8 +491,11 @@ class Portal:
if self.peer_type == "user": if self.peer_type == "user":
await self.main_intent.leave_room(self.mxid) await self.main_intent.leave_room(self.mxid)
self.delete() self.delete()
del self.by_tgid[self.tgid_full] try:
del self.by_mxid[self.mxid] del self.by_tgid[self.tgid_full]
del self.by_mxid[self.mxid]
except KeyError:
pass
elif source and source.tgid != user.tgid: elif source and source.tgid != user.tgid:
if self.peer_type == "chat": if self.peer_type == "chat":
await source.client(DeleteChatUserRequest(chat_id=self.tgid, user_id=user.tgid)) await source.client(DeleteChatUserRequest(chat_id=self.tgid, user_id=user.tgid))