diff --git a/bin/merge-users b/bin/merge-users
index 539268117a362064c4d04cfef149799321c5d2da..6a06e5bd466f42dbb18050425ecf280660426a3e 100755
--- a/bin/merge-users
+++ b/bin/merge-users
@@ -1,6 +1,7 @@
#!/usr/bin/env python3
import argparse
+from sqlalchemy import and_
import mo.db as db
import mo.users
@@ -33,10 +34,31 @@ print(f"Slučuji UID {suid} do UID {duid}")
sess = db.get_session()
conn = sess.connection()
+test_round = sess.query(db.Round).filter_by(category='T').one_or_none()
+if test_round is not None:
+ test_submits = (sess
+ .query(db.Solution)
+ .join(db.Task)
+ .filter(db.Solution.user_id == suid)
+ .filter(db.Task.round == test_round)
+ .filter(db.Round.category == 'T')
+ .all())
+ if test_submits:
+ mo.util.die("Zdrojový účastník něco odevzdal v testovací soutěži, nutno vyřešit ručně")
+ test_contest = sess.query(db.Contest).filter_by(round=test_round).one_or_none()
+ test_contest_id = test_contest.contest_id if test_contest is not None else None
+else:
+ test_contest = None
+ test_contest_id = None
+
+sess.flush()
+
conn.execute(db.Log.__table__.update().where(db.Log.changed_by == suid).values(changed_by=duid))
conn.execute(db.Participant.__table__.delete().where(db.Participant.user_id == suid))
+conn.execute(db.Participation.__table__.delete().where(and_(db.Participation.user_id == suid, db.Participation.contest_id == test_contest_id)))
+
conn.execute(db.Participation.__table__.update().where(db.Participation.user_id == suid).values(user_id=duid))
conn.execute(db.UserRole.__table__.update().where(db.UserRole.user_id == suid).values(user_id=duid))