import csv

team_mates = {}

with open("CS384 Project Group (Responses) - Form Responses 1.csv") as f:
    reader = csv.DictReader(f)
    data = [dict(row) for row in reader]

for row in data:
    if row["Partner 1 Name"] not in team_mates:
        team_mates[row["Partner 1 Name"]] = set()
    if row["Partner 2 Name"] not in team_mates:
        team_mates[row["Partner 2 Name"]] = set()
    if row["Partner 2 Name"] not in team_mates[row["Partner 1 Name"]]:
        team_mates[row["Partner 1 Name"]].add(row["Partner 2 Name"])
    if row["Partner 1 Name"] not in team_mates[row["Partner 2 Name"]]:
        team_mates[row["Partner 2 Name"]].add(row["Partner 1 Name"])


with open("MultipleTeamMates.csv", "w") as f:
    writer = csv.DictWriter(f, fieldnames=["Name", "Team mates"])
    writer.writeheader()
    for key, value in team_mates.items():
        if len(value) != 1:
            writer.writerow({"Name": key, "Team mates": ",".join(value)})

with open("contact_info_for_andriod.csv", "r") as f:
    reader = csv.DictReader(f)
    all_names = [dict(row) for row in reader]


with open("NotRegistered.csv", "w") as f:
    writer = csv.DictWriter(f, fieldnames=["Name", "Email1", "Email2"])
    writer.writeheader()
    for name in all_names:
        name["Name"] = name["Name"].strip()
        if name["Name"] not in team_mates:
            writer.writerow(
                {"Name": name['Name'], "Email1": name["Email1"], "Email2": name["Email2"]}
            )