Fix name edge case

This commit is contained in:
Will Vesey
2018-06-27 13:54:45 -04:00
parent 39adc646e6
commit e57ae069a7

View File

@@ -185,12 +185,18 @@ def graphql_to_thread(thread):
if 'last_message' in thread: if 'last_message' in thread:
last_message_timestamp = thread['last_message']['nodes'][0]['timestamp_precise'] last_message_timestamp = thread['last_message']['nodes'][0]['timestamp_precise']
first_name = user.get('short_name')
if first_name is None:
last_name = None
else:
last_name = user.get('name').split(first_name, 1).pop().strip()
return User( return User(
user['id'], user['id'],
url=user.get('url'), url=user.get('url'),
name=user.get('name'), name=user.get('name'),
first_name=user.get('short_name'), first_name=first_name,
last_name=user.get('name').split(user.get('short_name'),1).pop().strip(), last_name=last_name,
is_friend=user.get('is_viewer_friend'), is_friend=user.get('is_viewer_friend'),
gender=GENDERS.get(user.get('gender')), gender=GENDERS.get(user.get('gender')),
affinity=user.get('affinity'), affinity=user.get('affinity'),