diff --git a/login.py b/login.py index bc02d8e..70c4f16 100644 --- a/login.py +++ b/login.py @@ -1,40 +1,34 @@ -from flask import Flask -from flask import Flask, flash, redirect, render_template, request, session, abort -import os -from sqlalchemy.orm import sessionmaker -from tabledef import * -engine = create_engine('sqlite:///tutorial.db', echo=True) - -app = Flask(__name__) - -@app.route('/') -def home(): - if not session.get('logged_in'): - return render_template('login.html') - else: - return "Hello Boss! Logout" - -@app.route('/login', methods=['POST']) -def do_admin_login(): - - POST_USERNAME = str(request.form['username']) - POST_PASSWORD = str(request.form['password']) - - Session = sessionmaker(bind=engine) - s = Session() - query = s.query(User).filter(User.username.in_([POST_USERNAME]), User.password.in_([POST_PASSWORD]) ) - result = query.first() - if result: - session['logged_in'] = True - else: - flash('wrong password!') - return home() - -@app.route("/logout") -def logout(): - session['logged_in'] = False - return home() - -if __name__ == "__main__": - app.secret_key = os.urandom(12) - app.run(debug=True,host='0.0.0.0', port=4000) + +{% block body %} +{% if session['logged_in'] %} +
You're logged in already!
+{% else %} + + + + +{% endif %} +{% endblock %}