Advertisement
Guest User

Untitled

a guest
Aug 31st, 2016
97
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.18 KB | None | 0 0
  1. @app.route('/login', methods=('GET', 'POST'))
  2. def login():
  3. form = LoginForm(request.form)
  4. if form.validate_on_submit():
  5.  
  6. user = User.query.filter_by(email=form.email.data).first()
  7. if user and check_password_hash(user.password, form.password.data):
  8. session['user.id'] = user.id
  9. flash('Welcome %s' % user.username)
  10. return redirect(url_for('index'))
  11. return render_template("login.html", form=form)
  12.  
  13. class User(UserMixin,db.Model):
  14. id = db.Column(db.Integer, primary_key = True, autoincrement=True)
  15. username = db.Column(db.String(64), index = True, unique = True)
  16. email = db.Column(db.String(120), index = True, unique = True)
  17. password = db.Column(db.String(120), index = True)
  18. role = db.Column(db.SmallInteger, default=ROLE_USER)
  19.  
  20. def __init__(self, id, username, email, password, role):
  21. self.id = id
  22. self.username = username
  23. self.password = password
  24. self.email = email
  25. self.role = role
  26.  
  27. def set_password(self, password):
  28. self.pw_hash = generate_password_hash(password)
  29.  
  30. def check_password(self, password):
  31. return check_password_hash(self.pw_hash, password)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement