From dcc9525246f18929cbd6603f22b11843a7b8eb08 Mon Sep 17 00:00:00 2001
From: Martin Mares <mj@ucw.cz>
Date: Sun, 30 Oct 2022 00:17:50 +0200
Subject: [PATCH] =?UTF-8?q?import-points:=20Import=20v=C5=A1ech=20=C3=BAlo?=
=?UTF-8?q?h=20najednou?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit
---
bin/import-points | 13 ++++++++-----
1 file changed, 8 insertions(+), 5 deletions(-)
diff --git a/bin/import-points b/bin/import-points
index 14227e98..4309d892 100755
--- a/bin/import-points
+++ b/bin/import-points
@@ -10,9 +10,9 @@ import mo.users
import mo.util
from mo.util import die
-parser = argparse.ArgumentParser(description='Importuje body za úlohu (bez --import vypíše šablonu).')
+parser = argparse.ArgumentParser(description='Importuje body za jednu nebo více úloh (bez --import vypíše šablonu).')
parser.add_argument('--round', '-r', type=str, required=True, metavar='YY-C-S[p]', help='kód kola')
-parser.add_argument('--task', '-t', type=str, required=True, metavar='ID', help='kód úlohy')
+parser.add_argument('--task', '-t', type=str, metavar='ID', help='kód úlohy (default: všechny úlohy kola)')
parser.add_argument('--user', '-u', type=str, required=True, metavar='EMAIL', help='uživatel, ktery provádí import')
parser.add_argument('--import', '-i', dest='import_file', type=str, metavar='NAME', help='importuje data ze souboru (TSV)')
parser.add_argument('--add-del', default=False, action='store_true', help='povolí zakládat/rušit řešení')
@@ -29,9 +29,12 @@ round = mo.util.get_round_by_code(round_code)
if round is None:
die("Kolo s tímto kódem neexistuje!")
-task = sess.query(db.Task).filter_by(round=round, code=args.task).one_or_none()
-if task is None:
- die("Úloha s tímto kódem neexistuje")
+if args.task is not None:
+ task = sess.query(db.Task).filter_by(round=round, code=args.task).one_or_none()
+ if task is None:
+ die("Úloha s tímto kódem neexistuje")
+else:
+ task = None
user = mo.users.user_by_email(args.user)
if user is None:
--
GitLab