diff options
| -rw-r--r-- | process-polls.py | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/process-polls.py b/process-polls.py index 63d3692..cd64764 100644 --- a/process-polls.py +++ b/process-polls.py @@ -67,8 +67,14 @@ for row in rows: voters_1 = get_all_voters(s, row[4], row[5], answer_list_1[0]['answer_id']) voters_2 = get_all_voters(s, row[4], row[5], answer_list_2[0]['answer_id']) + def user_to_params(user): + name = user['global_name'] + if name is None: + name = user['username'] + return (user['id'], name, user['avatar'], the_time) + cur.executemany('INSERT OR REPLACE INTO users VALUES(?, ?, ?, ?)', - list(map(lambda x: (x['id'], x['global_name'], x['avatar'], the_time), voters_1 + voters_2))) + list(map(user_to_params, voters_1 + voters_2))) cur.execute(''' UPDATE polls |