new new version

1 files changed, 94 insertions, 0 deletions

diff --git a/process-polls.py b/process-polls.py
new file mode 100644
index 0000000..6400e15
--- /dev/null
+++ b/process-polls.py
@@ -0,0 +1,94 @@
+from os import environ
+import argparse
+import discord
+import sqlite3
+import time
+
+ap = argparse.ArgumentParser()
+ap.add_argument('database', help='the file with the database')
+
+a = ap.parse_args()
+
+if 'TOKEN' not in environ:
+  print('please set an environment variable TOKEN with the bot token')
+  exit(1)
+
+token = environ['TOKEN']
+
+dbcon = sqlite3.connect(a.database)
+
+cur = dbcon.cursor()
+
+rows = cur.execute('''
+  SELECT entry_1, entry_2, poll_number, expires, channel_id, message_id
+  FROM polls
+  WHERE voters_1 IS NULL
+  ORDER BY expires
+''').fetchall()
+
+s = discord.session(token)
+
+def get_all_voters(s, channel, message, answer):
+  endpoint_base = f'/channels/{channel}/polls/{message}/answers/{answer}?limit=100'
+  users = discord.get(s, endpoint_base)['users']
+  if len(users) < 100:
+    return users
+  while True:
+    new_users = discord.get(s, endpoint_base + f'&after={users[-1]['id']}')['users']
+    users += new_users
+    if len(new_users) < 100:
+      return users
+
+for row in rows:
+
+  expires = row[3]
+  sleep_for = expires - time.time()
+  if sleep_for > 0:
+    time.sleep(sleep_for)
+
+  result = None
+
+  time_to_sleep = 1
+  while True:
+    result = discord.get(s, f'/channels/{row[4]}/messages/{row[5]}')
+    if result['poll']['results']['is_finalized']:
+      break
+    time.sleep(time_to_sleep)
+    time_to_sleep *= 2
+
+  answer_list_1 = list(filter(lambda x: x['poll_media']['text'] == row[0], result['poll']['answers']))
+  answer_list_2 = list(filter(lambda x: x['poll_media']['text'] == row[1], result['poll']['answers']))
+
+  if len(answer_list_1) != 1 or len(answer_list_2) != 1:
+    exit(1)
+
+  the_time = time.time()
+
+  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'])
+
+  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)))
+
+  cur.execute('''
+    UPDATE polls
+    SET voters_1 = ?, voters_2 = ?
+    WHERE poll_number = ?
+    LIMIT 1
+  ''', (
+    ','.join(map(lambda x: x['id'], voters_1)),
+    ','.join(map(lambda x: x['id'], voters_2)),
+    row[2]
+  ))
+
+  cur.executemany('''
+    UPDATE entries
+    SET status = ?
+    WHERE name = ?
+    LIMIT 1
+  ''', [
+    (0 if len(voters_1) >= len(voters_2) else 2, row[0]),
+    (0 if len(voters_2) >= len(voters_1) else 2, row[1])
+  ])
+
+  dbcon.commit()