Fix plan fetching from threads

This commit is contained in:
Kacper Ziubryniewicz
2018-08-23 12:17:22 +02:00
committed by GitHub
parent 7aa774b4ef
commit e35cc71cf4

View File

@@ -214,7 +214,7 @@ def graphql_to_user(user):
if user.get('profile_picture') is None: if user.get('profile_picture') is None:
user['profile_picture'] = {} user['profile_picture'] = {}
c_info = get_customization_info(user) 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( return User(
user['id'], user['id'],
url=user.get('url'), url=user.get('url'),
@@ -252,7 +252,7 @@ def graphql_to_thread(thread):
else: else:
last_name = user.get('name').split(first_name, 1).pop().strip() 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( return User(
user['id'], user['id'],
@@ -282,7 +282,7 @@ def graphql_to_group(group):
last_message_timestamp = None last_message_timestamp = None
if 'last_message' in group: if 'last_message' in group:
last_message_timestamp = group['last_message']['nodes'][0]['timestamp_precise'] 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( return Group(
group['thread_key']['thread_fbid'], group['thread_key']['thread_fbid'],
participants=set([node['messaging_actor']['id'] for node in group['all_participants']['nodes']]), 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: if room.get('image') is None:
room['image'] = {} room['image'] = {}
c_info = get_customization_info(room) 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( return Room(
room['thread_key']['thread_fbid'], room['thread_key']['thread_fbid'],
participants=set([node['messaging_actor']['id'] for node in room['all_participants']['nodes']]), 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'] = {} page['profile_picture'] = {}
if page.get('city') is None: if page.get('city') is None:
page['city'] = {} 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( return Page(
page['id'], page['id'],
url=page.get('url'), url=page.get('url'),