def register(): if request.method == "POST": username = request.form['username'] email = request.form['email'] password = request.form['password'] confirm_password = request.form['confirm_password'] msg = "" if not username or not password or not email: msg = {"status": {"type": "failure", "message": "Missing Data!"}} return jsonify(msg) if confirm_password != password: msg ={"status": {"type": "failure", "message": "Please make sure your passwords match."}} return jsonify(msg) if User.query.filter_by(username=username).count() == 1: msg = {"status": {"type": "failure", "message": "Username already taken!"}} return jsonify(msg) if User.query.filter_by(email=email).count() == 1: msg = {"status": {"type": "failure", "message": "Email already taken!"}} return jsonify(msg) u = User() u.username = username u.email = email u.set_password(password) db.session.add(u) db.session.commit(u) msg = {"status": {"type": "success", "message": "You have registered successfully!"}} return jsonify(msg)
def register(): form = RegistrationForm() if form.validate_on_submit(): user = User(username=form.username.data, email=form.email.data) user.set_password(form.password.data) db.session.add(user) db.session.commit() flash(f'Account created for {form.username.data}!', 'success') return redirect(url_for('users.login')) return render_template('register.html', title='Register', form=form)
def test_password_hashing(self): u = User(username='******') u.set_password('cat') self.assertFalse(u.check_password('dog')) self.assertTrue(u.check_password('cat'))