diff --git a/.gitignore b/.gitignore index 3b35dc2..d0afe9e 100644 --- a/.gitignore +++ b/.gitignore @@ -114,3 +114,4 @@ dmypy.json .pyre/ toxic/ .idea/ +fuck/ diff --git a/main.py b/main.py index d6196d5..4f9e22e 100644 --- a/main.py +++ b/main.py @@ -9,10 +9,22 @@ unique_old = [] def main(in_args): - # reading CSV file - user_data = read_csv(in_args.csv_input) - discord_ids = user_data['AuthorID'].tolist() + if not os.path.isfile(in_args.csv_input): + for filename in os.listdir(in_args.csv_input): + file_path = os.path.join(in_args.csv_input, filename) + process(in_args, str(file_path)) + else: + # reading CSV file + csv_path = in_args.csv_input + process(in_args, csv_path) + + print("Done.") + + +def process(in_args, csv_path: str): + user_data = read_csv(csv_path) + discord_ids = user_data['AuthorID'].tolist() # check if the output exists to add new IDs too if os.path.isfile(in_args.csv_output) is False: write_csv(["User ID"], in_args.csv_output) @@ -20,20 +32,18 @@ def main(in_args): old_data = read_csv(in_args.csv_output) for x in old_data['User ID'].tolist(): unique_old.append(int(x)) - # check for new IDs 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) - # add users to the CSV for user_id in unique_ids: print("User: ", str(user_id), end="\r", flush=True) update_csv(user_id, in_args.csv_output) - print("Done.") + unique_ids.clear() def write_csv(fields, csv_path):