make prefix optional

VNOI-Admin · Dec 10, 2024 · 5e640b9 · 5e640b9
1 parent ba417b9
commit 5e640b9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/judge/management/commands/batch_add_icpc_team.py b/judge/management/commands/batch_add_icpc_team.py
@@ -62,7 +62,7 @@ class Command(BaseCommand):
     def add_arguments(self, parser):
         parser.add_argument('input', help='csv file containing username and teamname')
         parser.add_argument('output', help='where to store output csv file')
-        parser.add_argument('prefix', help='prefix for username')
+        parser.add_argument('prefix', help='prefix for username', type=str, nargs='?', default='')
 
     def handle(self, *args, **options):
         fin = open(options['input'], 'r', encoding='utf-8')

diff --git a/judge/management/commands/batch_add_olp.py b/judge/management/commands/batch_add_olp.py
@@ -46,7 +46,7 @@ class Command(BaseCommand):
     def add_arguments(self, parser):
         parser.add_argument('input', help='csv file containing username and teamname')
         parser.add_argument('output', help='where to store output csv file')
-        parser.add_argument('prefix', help='prefix for username')
+        parser.add_argument('prefix', help='prefix for username', type=str, nargs='?', default='')
 
     def handle(self, *args, **options):
         fin = open(options['input'], 'r', encoding='utf-8')