Automatické přihlašování účastníků do testovací soutěže

etc/config.py.example

@@ -47,3 +47,7 @@ JOB_GC_PERIOD = 60
 
 # Za jak dlouho expiruje dokončená dávka [min]
 JOB_EXPIRATION = 5
+
+# Automatické přihlašování účastníků do testovací soutěže
+# (kolo aktuální_ročník-T-1, celostátní soutěž)
+AUTO_REGISTER_TEST = True

mo/web/user.py

@@ -3,9 +3,11 @@ from flask_wtf import FlaskForm
 import flask_wtf.file
 from sqlalchemy import and_
 from sqlalchemy.orm import joinedload
+from typing import List, Tuple
 import werkzeug.exceptions
 import wtforms
 
+import mo
 import mo.config as config
 import mo.db as db
 import mo.submit
@@ -18,9 +20,19 @@ import mo.web.util
 
 @app.route('/user')
 def user_index():
-    sess = db.get_session()
+    pcrs = load_pcrs()
+    if getattr(config, 'AUTO_REGISTER_TEST', False) and not any(round.category == 'T' for pion, contest, round in pcrs):
+        if register_to_test():
+            pcrs = load_pcrs()
+
+    return render_template(
+        'user_index.html',
+        pions=pcrs,
+    )
+
 
-    pions = (sess.query(db.Participation, db.Contest, db.Round)
+def load_pcrs() -> List[Tuple[db.Participation, db.Contest, db.Round]]:
+    return (db.get_session().query(db.Participation, db.Contest, db.Round)
             .select_from(db.Participation)
             .join(db.Contest, db.Contest.master_contest_id == db.Participation.contest_id)
             .join(db.Round)
@@ -29,10 +41,35 @@ def user_index():
             .order_by(db.Round.year.desc(), db.Round.category, db.Round.seq, db.Round.part)
             .all())
 
-    return render_template(
-        'user_index.html',
-        pions=pions,
+
+def register_to_test() -> bool:
+    sess = db.get_session()
+    round = sess.query(db.Round).filter_by(year=mo.current_year, category='T', seq=1, part=0).one_or_none()
+    if not round:
+        app.logger.error(f'Nemohu najít kolo {mo.current_year}-T-1')
+        return False
+
+    if round.level != 0:
+        app.logger.error(f'Kolo {round.round_code_short()} není na celostátní úrovni')
+        return False
+
+    contest = sess.query(db.Contest).filter_by(round=round).limit(1).one_or_none()
+    if not contest:
+        app.logger.error(f'Kolo {round.round_code_short()} nemá soutěž')
+        return False
+
+    pion = db.Participation(user=g.user, contest=contest, place=contest.place, state=db.PartState.registered)
+    sess.add(pion)
+    sess.flush()
+    mo.util.log(
+        type=db.LogType.participant,
+        what=g.user.user_id,
+        details={'action': 'add-to-contest', 'new': db.row2dict(pion)},
     )
+    sess.commit()
+
+    app.logger.info(f'Účastník #{g.user.user_id} automaticky registrován do soutěže #{contest.contest_id}')
+    return True
 
 
 def get_contest(id: int) -> db.Contest: