Fix possible errors

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
+6
View File
@@ -65,6 +65,12 @@ class Bot(AbstractUser):
except (ChannelPrivateError, ChannelInvalidError):
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):
if id not in self.chats:
self.chats[id] = type
+5 -2
View File
@@ -491,8 +491,11 @@ class Portal:
if self.peer_type == "user":
await self.main_intent.leave_room(self.mxid)
self.delete()
del self.by_tgid[self.tgid_full]
del self.by_mxid[self.mxid]
try:
del self.by_tgid[self.tgid_full]
del self.by_mxid[self.mxid]
except KeyError:
pass
elif source and source.tgid != user.tgid:
if self.peer_type == "chat":
await source.client(DeleteChatUserRequest(chat_id=self.tgid, user_id=user.tgid))