Update login.py
This commit is contained in:
parent
1d0418d9eb
commit
a6567c0ecb
62
login.py
62
login.py
@ -1,40 +1,34 @@
|
|||||||
from flask import Flask
|
<link rel="stylesheet" href="/static/style.css" type="text/css">
|
||||||
from flask import Flask, flash, redirect, render_template, request, session, abort
|
{% block body %}
|
||||||
import os
|
{% if session['logged_in'] %}
|
||||||
from sqlalchemy.orm import sessionmaker
|
<p>You're logged in already!</p>
|
||||||
from tabledef import *
|
{% else %}
|
||||||
engine = create_engine('sqlite:///tutorial.db', echo=True)
|
|
||||||
|
|
||||||
app = Flask(__name__)
|
|
||||||
|
|
||||||
@app.route('/')
|
<form action="/login" method="POST">
|
||||||
def home():
|
<div class="login">
|
||||||
if not session.get('logged_in'):
|
<div class="login-screen">
|
||||||
return render_template('login.html')
|
<div class="app-title">
|
||||||
else:
|
<h1>Login</h1>
|
||||||
return "Hello Boss! <a href='/logout'>Logout</a>"
|
</div>
|
||||||
|
|
||||||
@app.route('/login', methods=['POST'])
|
<div class="login-form">
|
||||||
def do_admin_login():
|
<div class="control-group">
|
||||||
|
<input type="text" class="login-field" value="" placeholder="username" name="username">
|
||||||
|
<label class="login-field-icon fui-user" for="login-name"></label>
|
||||||
|
</div>
|
||||||
|
|
||||||
POST_USERNAME = str(request.form['username'])
|
<div class="control-group">
|
||||||
POST_PASSWORD = str(request.form['password'])
|
<input type="password" class="login-field" value="" placeholder="password" name="password">
|
||||||
|
<label class="login-field-icon fui-lock" for="login-pass"></label>
|
||||||
|
</div>
|
||||||
|
|
||||||
Session = sessionmaker(bind=engine)
|
<input type="submit" value="Log in" class="btn btn-primary btn-large btn-block" >
|
||||||
s = Session()
|
<br>
|
||||||
query = s.query(User).filter(User.username.in_([POST_USERNAME]), User.password.in_([POST_PASSWORD]) )
|
</div>
|
||||||
result = query.first()
|
</div>
|
||||||
if result:
|
</div>
|
||||||
session['logged_in'] = True
|
</form>
|
||||||
else:
|
|
||||||
flash('wrong password!')
|
|
||||||
return home()
|
|
||||||
|
|
||||||
@app.route("/logout")
|
{% endif %}
|
||||||
def logout():
|
{% endblock %}
|
||||||
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