fix error with optional data

This commit is contained in:
Zomatree
2021-08-30 16:25:09 +01:00
parent 013885dfb4
commit 37e3da5ec7
2 changed files with 7 additions and 5 deletions
+1 -3
View File
@@ -10,10 +10,8 @@ Relation = Literal["Blocked", "BlockedOther", "Friend", "Incoming", "None", "Out
class UserBot(TypedDict):
owner: str
class _OptionalStatus(TypedDict, total=False):
class Status(TypedDict, total=False):
text: str
class Status(_OptionalStatus):
presence: Literal["Busy", "Idle", "Invisible", "Online"]
class UserRelation(TypedDict):
+6 -2
View File
@@ -15,7 +15,7 @@ class Relation(NamedTuple):
class Status(NamedTuple):
text: Optional[str]
presence: PresenceType
presence: Optional[PresenceType]
class User:
"""Represents a user
@@ -66,4 +66,8 @@ class User:
self.relationship = RelationshipType(relationship) if relationship else None
status = data.get("status")
self.status = Status(status.get("text"), PresenceType(status["presence"])) if status else None
if status:
presence = status.get("presence")
self.status = Status(status.get("text"), PresenceType(presence) if presence else None) if status else None
else:
self.status = None