From 72cddf98e303f1edd0e6fc4981d188df9c1738cd Mon Sep 17 00:00:00 2001
From: Owen Quinlan <46948241+BuyMyMojo@users.noreply.github.com>
Date: Wed, 4 Aug 2021 13:50:25 +1000
Subject: [PATCH] v1.0 let's god

---
 .gitignore |  1 +
 main.py    | 58 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 59 insertions(+)
 create mode 100644 main.py

diff --git a/.gitignore b/.gitignore
index 510c73d..3639b5c 100644
--- a/.gitignore
+++ b/.gitignore
@@ -112,3 +112,4 @@ dmypy.json
 
 # Pyre type checker
 .pyre/
+!/toxic/
diff --git a/main.py b/main.py
new file mode 100644
index 0000000..ae2a82c
--- /dev/null
+++ b/main.py
@@ -0,0 +1,58 @@
+import argparse as argp
+import csv
+import os
+
+from pandas import read_csv
+
+unique_ids = []
+unique_old = []
+
+
+def main(in_args):
+    # reading CSV file
+    user_data = read_csv(in_args.csv_input)
+    discord_ids = user_data['AuthorID'].tolist()
+
+    if os.path.isfile(in_args.csv_output) is False:
+        write_csv(["User ID"], in_args.csv_output)
+    else:
+        old_data = read_csv(in_args.csv_output)
+        for x in old_data['User ID'].tolist():
+            unique_old.append(int(x))
+
+    for user_id in discord_ids:
+        int_id = int(user_id)
+        print("User: ", str(int_id), end="\r", flush=True)
+        if user_id not in unique_ids and user_id not in unique_old:
+            unique_ids.append(user_id)
+
+    for user_id in unique_ids:
+        print("User: ", str(user_id), end="\r", flush=True)
+        update_csv(user_id, in_args.csv_output)
+
+
+def write_csv(fields, csv_path):
+    with open(csv_path, 'w', newline='') as csv_file:
+        writer = csv.DictWriter(csv_file, fieldnames=fields)
+
+        writer.writeheader()
+
+
+def update_csv(uids, csv_path):
+    # print("FPS: ", str(FPS), end="\r", flush=True)
+    with open(csv_path, 'a', newline='') as csv_file:
+        writer = csv.writer(csv_file)
+        writer.writerow([uids])
+
+
+# setup argparse
+parser = argp.ArgumentParser(description='Make a list of all unique user IDs from a discord chat exporter CSV',
+                             allow_abbrev=False)
+
+# add arguments
+parser.add_argument('csv_input', metavar='csv_input', type=str, help='The path of your exported CSV file')
+parser.add_argument('csv_output', metavar='csv_output', type=str, help='The path your new CSV with unique IDs')
+
+args = parser.parse_args()
+
+main(args)