batch icpc team

VNOI-Admin · Oct 18, 2024 · 3e53101 · 3e53101
1 parent 74be67b
commit 3e53101
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/judge/management/commands/batch_add_icpc_team.py b/judge/management/commands/batch_add_icpc_team.py
@@ -75,7 +75,7 @@ def handle(self, *args, **options):
         prefix = options['prefix']
 
         reader = csv.DictReader(fin)
-        writer = csv.DictWriter(fout, fieldnames=['username', 'teamname', 'password', 'org', 'email'])
+        writer = csv.DictWriter(fout, fieldnames=['id', 'username', 'teamname', 'password', 'org', 'email'])
         writer.writeheader()
 
         done_team_ids = set()
@@ -98,6 +98,7 @@ def handle(self, *args, **options):
             add_user(username, teamname, password, org, org_group, internalid)
 
             writer.writerow({
+                'id': internalid,
                 'username': username,
                 'teamname': teamname,
                 'password': password,