Mirrors/tg
2
0
Fork 0
mirror of https://github.com/paul-nameless/tg synced 2024-11-22 03:43:19 +00:00
Code Issues Projects Releases Packages Wiki Activity

Check if msg index exists firstly: causes KeyError in logs (#114)

Browse source
This commit is contained in:
Nameless 2020-07-05 16:53:59 +08:00 committed by GitHub
parent a053fdefec
commit 568e15aecb
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
tg/models.py
View file

@ -351,7 +351,9 @@ class MsgModel:
}
def update_msg_content_opened(self, chat_id: int, msg_id: int) -> None:
index = self.msg_idx[chat_id][msg_id]
index = self.msg_idx[chat_id].get(msg_id)
if not index:
return
msg = MsgProxy(self.msgs[chat_id][index])
if msg.content_type == "voice":
msg.is_listened = True
@ -364,7 +366,9 @@ class MsgModel:
def update_msg(
self, chat_id: int, msg_id: int, **fields: Dict[str, Any]
) -> None:
index = self.msg_idx[chat_id][msg_id]
index = self.msg_idx[chat_id].get(msg_id)
if not index:
return
msg = self.msgs[chat_id][index]
msg.update(fields)