58 lines
1.7 KiB
Python
58 lines
1.7 KiB
Python
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)
|