diff --git a/judge/management/commands/batch_add_icpc_team.py b/judge/management/commands/batch_add_icpc_team.py index 95da9b3f3..38bb98874 100755 --- 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,