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'))
