create-tasks: skript na zakládání série úloh

bin/create-tasks

+#!/usr/bin/env python3
+
+import argparse
+
+import mo.db as db
+import mo.util
+from mo.util import die, init_standalone
+
+parser = argparse.ArgumentParser(description='Založí úlohy pro dané kolo')
+parser.add_argument(dest='round', type=str, metavar='YY-C-S[p]', help='kód kola')
+parser.add_argument(dest='count', type=int, help='počet úloh')
+parser.add_argument('-p', '--points', type=int, default=None, help='maximální počet bodů')
+
+args = parser.parse_args()
+
+init_standalone()
+sess = db.get_session()
+
+round_code = mo.util.RoundCode.parse(args.round)
+if round_code is None:
+    die("Chybná syntaxe kódu kola")
+round = mo.util.get_round_by_code(round_code)
+if round is None:
+    die("Kolo s tímto kódem neexistuje!")
+
+if round.state != db.RoundState.preparing:
+    die("Kolo musí být ve stavu 'připravuje se'")
+
+for i in range(1, args.count + 1):
+    code = f'{round.category}-{round.code or round.seq}-{i}'
+    task = sess.query(db.Task).filter_by(round=round, code=code).one_or_none()
+    if task:
+        print(f'{code}: již existuje')
+    else:
+        task = db.Task(round=round, code=code, name=f'Úloha {i}', max_points=args.points)
+        sess.add(task)
+        print(f'{code}: zakládám')
+
+sess.commit()

setup.py

@@ -11,6 +11,7 @@ setuptools.setup(
         'bin/add-role',
         'bin/create-contests',
         'bin/create-round',
+        'bin/create-tasks',
         'bin/create-user',
         'bin/export-orgs',
         'bin/fix-submits',