def login(): # just display a login page if request.method == 'GET': return render_template('login.html') # get credentials username = request.form['username'] password = request.form['password'] # validate credentials and load user user_obj = User.load_by_username_password(username, password, config) if user_obj is None: flash('Username or Password is invalid', 'error') return redirect(url_for('login')) # log in user login_user(user_obj) flash('Logged in successfully') return redirect(url_for('admin'))
args = vars(parser.parse_args()) config = Config() # Display all users if args["command"] == "list": users = User.load_all(config) if users: print "Found %d users:" % len(users) for user in users: print "%s - %s" % (user['username'], user['password']) else: print "No users found" # Add a new user if args["command"] == "add": users = User.store_username_password(args['username'], args['password'], config) print "Added user %s to the database" % args['username'] # Delete a user if args["command"] == "delete": users = User.remove_by_username(args['username'], config) print "Removed user %s from the database" % args['username'] # Test login if args["command"] == "login": user = User.load_by_username_password(args['username'], args['password'], config) if user: print "Login successful with username '%s' and password '%s'!" % (args['username'], args['password']) else: print "Login failed with username '%s'!" % args['username']