Fix in new graphql methods

This commit is contained in:
Kacper Ziubryniewicz
2018-08-05 22:09:03 +02:00
committed by GitHub
parent 0bcccfa65e
commit 043d6b492d

View File

@@ -130,11 +130,11 @@ def graphql_to_attachment(a):
def graphql_to_poll(a): def graphql_to_poll(a):
rtn = Poll( rtn = Poll(
title=a.get('title'), title=a.get('title') if a.get('title') else a.get("text"),
options=[graph_to_poll_option(m) for m in a.get('options')] options=[graphql_to_poll_option(m) for m in a.get('options')]
) )
rtn.uid = a.get("id") rtn.uid = a.get("id")
rnt.options_count = a.get("total_count") rtn.options_count = a.get("total_count")
return rtn return rtn
def graphql_to_poll_option(a): def graphql_to_poll_option(a):
@@ -144,7 +144,7 @@ def graphql_to_poll_option(a):
) )
rtn.uid = a.get('id') rtn.uid = a.get('id')
rtn.voters = list(map(lambda x:x.get('node').get('id'), a.get('voters').get('edges'))) if isinstance(a.get('voters'), dict) else a.get('voters') rtn.voters = list(map(lambda x:x.get('node').get('id'), a.get('voters').get('edges'))) if isinstance(a.get('voters'), dict) else a.get('voters')
rnt.votes_count = a.get('voters').get('count') if isinstance(a.get('voters'), dict) else a.get('total_count') rtn.votes_count = a.get('voters').get('count') if isinstance(a.get('voters'), dict) else a.get('total_count')
return rtn return rtn
def graphql_to_message(message): def graphql_to_message(message):