diff --git a/fbchat/_delta_class.py b/fbchat/_delta_class.py index 2bd82a5..c864dfa 100644 --- a/fbchat/_delta_class.py +++ b/fbchat/_delta_class.py @@ -181,7 +181,7 @@ class ThreadFolder(Event): def parse_delta(session, data): - class_ = data.get("class") + class_ = data["class"] if class_ == "ParticipantsAddedToGroupThread": return PeopleAdded._parse(session, data) elif class_ == "ParticipantLeftGroupThread": diff --git a/fbchat/_delta_type.py b/fbchat/_delta_type.py index 2ea78f1..7011b40 100644 --- a/fbchat/_delta_type.py +++ b/fbchat/_delta_type.py @@ -279,7 +279,7 @@ class PlanResponded(ThreadEvent): def parse_delta(session, data): - type_ = data.get("type") + type_ = data["type"] if type_ == "change_thread_theme": return ColorSet._parse(session, data) elif type_ == "change_thread_icon": diff --git a/fbchat/_event.py b/fbchat/_event.py index c55fb8b..86a0135 100644 --- a/fbchat/_event.py +++ b/fbchat/_event.py @@ -74,7 +74,7 @@ class Presence(Event): def parse_delta(session, data): try: - class_ = data.get("class") + class_ = data["class"] if class_ == "ClientPayload": yield from _client_payload.parse_client_payloads(session, data) elif class_ == "AdminTextMessage": @@ -93,8 +93,8 @@ def parse_events(session, topic, data): # See Mqtt._configure_connect_options for information about these topics try: if topic == "/t_ms": - if "deltas" not in data: - return + # `deltas` will always be available, since we're filtering out the things + # that don't have it earlier in the MQTT listener for delta in data["deltas"]: yield from parse_delta(session, delta) @@ -105,7 +105,7 @@ def parse_events(session, topic, data): yield Typing._parse_orca(session, data) elif topic == "/legacy_web": - if data.get("type") == "jewel_requests_add": + if data["type"] == "jewel_requests_add": yield FriendRequest._parse(session, data) else: yield UnknownEvent(source="/legacy_web", data=data)