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

Fix invalid usage of content_type

Browse source
This commit is contained in:
Paul Nameless 2020-05-22 14:00:56 +08:00
parent 981f0c7e65
commit f304ed698f
2 changed files with 10 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tg/models.py
View file

@ -284,9 +284,9 @@ class MsgModel:
if message["id"] != msg_id:
continue
msg = MsgProxy(message)
if msg.type == "voice":
if msg.content_type == "voice":
msg.is_listened = True
elif msg.type == "recording":
elif msg.content_type == "recording":
msg.is_viewed = True
# TODO: start the TTL timer for self-destructing messages
# that is the last case to implement

18
tg/msg.py
View file

@ -114,7 +114,7 @@ class MsgProxy:
@property
def local_path(self):
if self.msg["content"]["@type"] not in self.types:
if self.msg["content"]["@type"] is None:
return None
doc = self.get_doc(self.msg)
return doc["local"]["path"]
@ -126,7 +126,7 @@ class MsgProxy:
@local.setter
def local(self, value):
if self.msg["content"]["@type"] not in self.types:
if self.msg["content"]["@type"] is None:
return None
doc = self.get_doc(self.msg)
doc["local"] = value
@ -146,27 +146,25 @@ class MsgProxy:
@property
def is_listened(self) -> Optional[bool]:
if self.type != "voice":
if self.content_type != "voice":
return None
return self.msg["content"]["is_listened"]
@is_listened.setter
def is_listened(self, value: bool):
if self.type != "voice":
return None
self.msg["content"]["is_listened"] = value
if self.content_type == "voice":
self.msg["content"]["is_listened"] = value
@property
def is_viewed(self) -> Optional[bool]:
if self.type != "recording":
if self.content_type != "recording":
return None
return self.msg["content"]["is_viewed"]
@is_viewed.setter
def is_viewed(self, value: bool):
if self.type != "recording":
return None
self.msg["content"]["is_viewed"] = value
if self.content_type == "recording":
self.msg["content"]["is_viewed"] = value
@property
def msg_id(self) -> int: