Add files via upload
This commit is contained in:
parent
7b70202b3a
commit
1d0418d9eb
40
login.py
Normal file
40
login.py
Normal file
@ -0,0 +1,40 @@
|
|||||||
|
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! <a href='/logout'>Logout</a>"
|
||||||
|
|
||||||
|
@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)
|
Loading…
Reference in New Issue
Block a user