diff --git a/bin/add-role b/bin/add-role index 5c6d79c2a771a65c9bd9099a428ec7e34bf63234..206e36826ac8b256919f356d74944a1ec1bf44e2 100755 --- a/bin/add-role +++ b/bin/add-role @@ -3,6 +3,7 @@ import argparse import mo.db as db +import mo.users import mo.util from mo.util import die, init_standalone @@ -22,9 +23,9 @@ session = db.get_session() if args.email and args.uid: parser.error('--email a --uid nesmí být uvedeny současně') elif args.email: - user = session.query(db.User).filter_by(email=args.email).first() + user = mo.user.user_by_email(args.email) elif args.uid: - user = session.query(db.User).filter_by(user_id=args.uid).first() + user = mo.user.user_by_uid(args.uid) else: parser.error('Je nutné vybrat uživatele pomocí --email nebo --uid')