Guest User

Untitled

a guest
Dec 5th, 2017
119
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.89 KB | None | 0 0
  1. password = form.password.data.encode('utf-8')
  2. hashed_password = bcrypt.generate_password_hash(password).decode('utf-8')
  3.  
  4. password = form.password.data.encode('utf-8')
  5. if bcrypt.check_password_hash(user.password, password):
  6.  
  7. @app.route('/<role>/signup/', methods=['GET', 'POST'])
  8. def signUp(role):
  9. form = RegisterForm()
  10.  
  11. if form.validate_on_submit():
  12. password = form.password.data.encode('utf-8')
  13. hashed_password = bcrypt.generate_password_hash(password).decode('utf-8')
  14. new_user = User(role=role, password=hashed_password, visits=1)
  15. form.populate_obj(new_user)
  16.  
  17. db.session.add(new_user)
  18. db.session.commit()
  19.  
  20. login_user(new_user)
  21.  
  22. return redirect(url_for('createContactInfo', user_id=current_user.id))
  23.  
  24. return render_template('signup.html', form=form, role=role)
  25.  
  26. @app.route('/login/', methods=['GET', 'POST'])
  27. def logIn():
  28. form = LoginForm()
  29.  
  30. if form.validate_on_submit():
  31. password = form.password.data.encode('utf-8')
  32. user = User.query.filter_by(email=form.email.data).first()
  33. if user:
  34. if bcrypt.check_password_hash(user.password, password):
  35. login_user(user, remember=form.remember.data)
  36. User.query.filter_by(id=current_user.id).update({'visits': User.visits + 1})
  37. db.session.commit()
  38.  
  39. if current_user.role == 'admin':
  40. return redirect(url_for('adminMainPage', user_id=current_user.id))
  41. elif current_user.role == 'guest':
  42. return redirect(url_for('guestMainPage', user_id=current_user.id))
  43. else:
  44. return "wrong role!"
  45.  
  46. flash('Invalid login')
  47. return render_template('login.html', form=form)
  48.  
  49. flash('Invalid login')
  50. return render_template('login.html', form=form)
  51.  
  52. return render_template('login.html', form=form)
Add Comment
Please, Sign In to add comment