allow duplicate team in csv

VNOI-Admin · Oct 20, 2023 · 40791e1 · 40791e1
1 parent 0bdfd26
commit 40791e1
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/judge/management/commands/batch_add_icpc_team.py b/judge/management/commands/batch_add_icpc_team.py
@@ -71,12 +71,17 @@ def handle(self, *args, **options):
         writer = csv.DictWriter(fout, fieldnames=['username', 'teamname', 'password'])
         writer.writeheader()
 
+        done_team_ids = set()
+
         for cnt, row in enumerate(reader, start=1):
             username = f'{prefix}{cnt}'
             teamname = row['name']
             org = get_org(row['instName'])
             password = generate_password()
             internalid = row['id']
+            if internalid in done_team_ids:
+                continue
+            done_team_ids.add(internalid)
 
             add_user(username, teamname, password, org, internalid)