From 22476bfb7b83d1fe8f00f5b4730133df5e25a807 Mon Sep 17 00:00:00 2001 From: crudo Date: Fri, 24 Mar 2017 15:54:08 +0100 Subject: [PATCH] Implement 'user set' command in cli.py. --- cli.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/cli.py b/cli.py index e2cd7e4..4d55ba1 100755 --- a/cli.py +++ b/cli.py @@ -63,6 +63,26 @@ def user_list(): print("No users found.") +@user.command('set') +@click.option('-p', '--password') +@click.argument('user_uid', type=click.INT) +def user_set(user_uid, password): + with db.get_session() as session: + user = session.query(User).get(user_uid) + + if not user: + print("No user found with id #{}.".format(user_uid)) + return + + if password: + user.password = password + + with db.get_session() as session: + session.add(user) + + print("User successfully edited.") + + @cli.group('event') def event(): pass