From e35cc71cf439cc9b417f3a84edb2c12090487b7b Mon Sep 17 00:00:00 2001 From: Kacper Ziubryniewicz Date: Thu, 23 Aug 2018 12:17:22 +0200 Subject: [PATCH] Fix plan fetching from threads --- fbchat/graphql.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/fbchat/graphql.py b/fbchat/graphql.py index d77b51b..f69089d 100644 --- a/fbchat/graphql.py +++ b/fbchat/graphql.py @@ -214,7 +214,7 @@ def graphql_to_user(user): if user.get('profile_picture') is None: user['profile_picture'] = {} c_info = get_customization_info(user) - plan = graphql_to_plan(user['event_reminders']['nodes'][0]) if user.get('event_reminders') else None + plan = graphql_to_plan(user['event_reminders']['nodes'][0]) if user.get('event_reminders', dict()).get('nodes') else None return User( user['id'], url=user.get('url'), @@ -252,7 +252,7 @@ def graphql_to_thread(thread): else: last_name = user.get('name').split(first_name, 1).pop().strip() - plan = graphql_to_plan(thread['event_reminders']['nodes'][0]) if thread.get('event_reminders') else None + plan = graphql_to_plan(thread['event_reminders']['nodes'][0]) if thread.get('event_reminders', dict()).get('nodes') else None return User( user['id'], @@ -282,7 +282,7 @@ def graphql_to_group(group): last_message_timestamp = None if 'last_message' in group: last_message_timestamp = group['last_message']['nodes'][0]['timestamp_precise'] - plan = graphql_to_plan(group['event_reminders']['nodes'][0]) if group.get('event_reminders') else None + plan = graphql_to_plan(group['event_reminders']['nodes'][0]) if group.get('event_reminders', dict()).get('nodes') else None return Group( group['thread_key']['thread_fbid'], participants=set([node['messaging_actor']['id'] for node in group['all_participants']['nodes']]), @@ -300,7 +300,7 @@ def graphql_to_room(room): if room.get('image') is None: room['image'] = {} c_info = get_customization_info(room) - plan = graphql_to_plan(room['event_reminders']['nodes'][0]) if room.get('event_reminders') else None + plan = graphql_to_plan(room['event_reminders']['nodes'][0]) if room.get('event_reminders', dict()).get('nodes') else None return Room( room['thread_key']['thread_fbid'], participants=set([node['messaging_actor']['id'] for node in room['all_participants']['nodes']]), @@ -323,7 +323,7 @@ def graphql_to_page(page): page['profile_picture'] = {} if page.get('city') is None: page['city'] = {} - plan = graphql_to_plan(page['event_reminders']['nodes'][0]) if page.get('event_reminders') else None + plan = graphql_to_plan(page['event_reminders']['nodes'][0]) if page.get('event_reminders', dict()).get('nodes') else None return Page( page['id'], url=page.get('url'),