Guest User

Untitled

a guest
Jun 25th, 2018
115
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 2.65 KB | None | 0 0
  1. app= Flask(__name__)
  2. app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://root:admin@localhost/sys'
  3. app.config['SECRET_KEY'] = 'mysecret'
  4.  
  5. db = SQLAlchemy(app)
  6. login = LoginManager(app)
  7. login.init_app(app)
  8. login.login_view = 'login'
  9. LOGIN_URL='/login'
  10. @login.user_loader
  11. def load_user(user_id):
  12. return Users.query.get(user_id)
  13.  
  14. class Users(db.Model, UserMixin):
  15. id = db.Column(db.Integer, primary_key=True)
  16. username = db.Column(db.String(20))
  17. password = db.Column(db.String(200))
  18. roles = db.Column(db.String(100))
  19. def is_authenticated(self):
  20. return True
  21. def is_active(self):
  22. return True
  23. def has_roles(self):
  24. return True
  25.  
  26. @app.route('/vendorreview')
  27. @login_required
  28. @roles_required('admin')
  29. def vendorreview():
  30. return render_template('review.html')
  31. @app.route('/register')
  32. @login_required
  33. def register():
  34. return render_template('register.html')
  35. @app.route('/submit')
  36. def submit():
  37. if not session.get('logged_in'):
  38. return render_template('login.html')
  39. else:
  40. return render_template('submit.html')
  41. return render_template('submit.html')
  42. @app.route('/error')
  43. def error():
  44. return render_template('error.html')
  45.  
  46. @app.route("/login", methods=['GET', 'POST'])
  47. def login():
  48. error = None
  49. x='customer'
  50. if request.method == 'POST':
  51. if request.form['username'] == 'admin' and request.form['password'] == 'admin' and x == 'business':
  52. session['logged_in'] = True
  53. # get_role = 'admin'
  54. return redirect(url_for('index'))
  55. #for customers
  56. elif request.form['username'] == 'admin' and request.form['password'] == 'admin' and x == 'customer':
  57. session['logged_in'] = True
  58. return redirect(url_for('index'))
  59. else:
  60. error = 'Invalid Credentials. Please try again.'
  61. return render_template('login.html', error=error)
  62.  
  63.  
  64. if __name__ == '__main__':
  65. app.secret_key = os.urandom(12)
  66. app.run(debug=True)
Add Comment
Please, Sign In to add comment