From bfd29e1b8f2ad93449a07039579a89b043ff9472 Mon Sep 17 00:00:00 2001
From: Pierre Theo Klein <pierre.klein@mail.mcgill.ca>
Date: Tue, 22 Jan 2019 11:54:29 -0500
Subject: [PATCH] Add batch script to update hacker statuses

---
 scripts/batch_update.py | 74 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 74 insertions(+)
 create mode 100644 scripts/batch_update.py

diff --git a/scripts/batch_update.py b/scripts/batch_update.py
new file mode 100644
index 00000000..c6586cce
--- /dev/null
+++ b/scripts/batch_update.py
@@ -0,0 +1,74 @@
+import requests
+import json
+import getpass
+
+print("Enter target API")
+DEFAULT_API = 'https://api.mchacks.ca'
+
+API_URL = input('Target API (Default '+DEFAULT_API+'):')
+
+if API_URL == '':
+    API_URL = DEFAULT_API
+
+# Get credentials
+print("Enter credentials for", API_URL)
+username = input("Email: ")
+
+s = requests.Session()
+
+logged_in = False
+
+while not logged_in:
+    password = getpass.getpass("Password: ")
+    CREDENTIALS = {
+        'email': username,
+        'password': password
+    }
+    r = s.post('{0}/api/auth/login'.format(API_URL), CREDENTIALS)
+
+    if r.status_code != 200:
+        print("Incorrect password, please try again.")
+    else:
+        print('Logged in as ', username)
+        logged_in = True
+
+# Get information about batch actions
+
+ACCEPT_ID_FILE = input("Path to file with hackerIDs:")
+
+VALID_ACTIONS = ['Applied', 'Accepted', 'Waitlisted',
+                 'Confirmed', 'Cancelled', 'Checked-in']
+
+INITIAL_STATUS = input(
+    "Inital status required " + str(VALID_ACTIONS) + ":")
+
+NEW_STATUS = input(
+    "Status to update to " + str(VALID_ACTIONS) + ":")
+
+
+hackerIds = []
+with open(ACCEPT_ID_FILE, 'r') as f:
+    rows = f.readlines()
+    for r in rows:
+        hackerIds.append(r.strip())
+
+# remove duplicates
+hackerIds = list(set(hackerIds))
+
+for index, ID in enumerate(hackerIds):
+    # so that we aren't 0-based index
+    index = index + 1
+    r1 = s.get('{0}/api/hacker/{1}'.format(API_URL, ID))
+    if r1.status_code != 200:
+        print('({0}/{1}) ERROR cannot get {2}'.format(index, len(hackerIds), ID))
+    else:
+        hackerInfo = json.loads(r1.content)
+        if hackerInfo['data']['status'] == INITIAL_STATUS:
+            r2 = s.patch('{0}/api/hacker/status/{1}'.format(API_URL, ID),
+                         {"status": NEW_STATUS})
+            if r2.status_code != 200:
+                print(
+                    '({0}/{1}) ERROR cannot update status for {2}'.format(index, len(hackerIds), ID))
+            else:
+                print(
+                    '({0}/{1}) {2} {3}'.format(index, len(hackerIds), NEW_STATUS, ID))