Skript na slučování účtů

bin/merge-users

+#!/usr/bin/env python3
+
+import argparse
+
+import mo.db as db
+import mo.users
+import mo.util
+
+parser = argparse.ArgumentParser(description='Sloučí dva uživatele do jednoho')
+parser.add_argument(dest='src', help='e-mailová adresa zdrojového účtu')
+parser.add_argument(dest='dest', help='e-mailová adresa cílového účtu')
+parser.add_argument('-n', '--dry-run', default=False, action='store_true', help='vyzkouší sloučení nanečisto (necommitne)')
+
+args = parser.parse_args()
+mo.util.init_standalone()
+
+su = mo.users.user_by_email(args.src)
+if su is None:
+    mo.util.die(f'Uživatel <{args.src}> nenalezen')
+
+du = mo.users.user_by_email(args.dest)
+if du is None:
+    mo.util.die(f'Uživatel <{args.dest}> nenalezen')
+
+if (su.is_org, su.is_admin) != (du.is_org, du.is_admin):
+    mo.util.die('Neumím sloučit účty různého typu')
+
+suid = su.user_id
+duid = du.user_id
+
+print(f"Slučuji UID {suid} do UID {duid}")
+
+sess = db.get_session()
+conn = sess.connection()
+
+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__.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))
+conn.execute(db.UserRole.__table__.update().where(db.UserRole.assigned_by == suid).values(assigned_by=duid))
+
+conn.execute(db.Paper.__table__.update().where(db.Paper.for_user == suid).values(for_user=duid))
+conn.execute(db.Paper.__table__.update().where(db.Paper.uploaded_by == suid).values(uploaded_by=duid))
+
+conn.execute(db.PointsHistory.__table__.update().where(db.PointsHistory.participant_id == suid).values(participant_id=duid))
+conn.execute(db.PointsHistory.__table__.update().where(db.PointsHistory.points_by == suid).values(points_by=duid))
+
+conn.execute(db.Solution.__table__.update().where(db.Solution.user_id == suid).values(user_id=duid))
+
+conn.execute(db.Job.__table__.update().where(db.Job.user_id == suid).values(user_id=duid))
+
+mo.util.log(db.LogType.user, suid, {
+    'action': 'merge',
+    'to': duid,
+})
+
+mo.util.log(db.LogType.user, duid, {
+    'action': 'merge',
+    'from': suid,
+})
+
+if not args.dry_run:
+    sess.commit()
+else:
+    sess.rollback()