Update login
This commit is contained in:
@@ -1,17 +0,0 @@
|
|||||||
__author__ = 'RemiZOffAlex'
|
|
||||||
__copyright__ = '(c) RemiZOffAlex'
|
|
||||||
__license__ = 'MIT'
|
|
||||||
__email__ = 'remizoffalex@mail.ru'
|
|
||||||
__url__ = 'http://remizoffalex.ru'
|
|
||||||
|
|
||||||
from wtforms import (
|
|
||||||
validators,
|
|
||||||
Form,
|
|
||||||
StringField,
|
|
||||||
PasswordField
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
class LoginForm(Form):
|
|
||||||
username = StringField('Логин', [validators.Length(min=4, max=25)])
|
|
||||||
password = PasswordField('Пароль', [validators.DataRequired()])
|
|
||||||
@@ -7,33 +7,14 @@ __url__ = 'https://remizoffalex.ru'
|
|||||||
from flask import render_template, escape, request, session, redirect
|
from flask import render_template, escape, request, session, redirect
|
||||||
|
|
||||||
from .. import app, lib, models
|
from .. import app, lib, models
|
||||||
from . import forms
|
|
||||||
|
|
||||||
@app.route('/login', methods=['GET', 'POST'])
|
|
||||||
|
@app.route('/login')
|
||||||
def login():
|
def login():
|
||||||
"""
|
"""
|
||||||
Логин
|
Логин
|
||||||
"""
|
"""
|
||||||
pagedata = {}
|
pagedata = {}
|
||||||
pagedata['form'] = forms.LoginForm(request.form)
|
|
||||||
|
|
||||||
if request.method == 'POST':
|
|
||||||
if pagedata['form'].validate():
|
|
||||||
user = models.db_session.query(
|
|
||||||
models.User
|
|
||||||
).filter(
|
|
||||||
models.User.name == escape(pagedata['form'].username.data),
|
|
||||||
models.User.password == lib.get_hash_password(
|
|
||||||
escape(pagedata['form'].password.data),
|
|
||||||
app.config['SECRET_KEY']
|
|
||||||
),
|
|
||||||
models.User.disabled == False
|
|
||||||
).first()
|
|
||||||
if user:
|
|
||||||
session['logged_in'] = True
|
|
||||||
session['user_id'] = user.id
|
|
||||||
return redirect('/', code=302)
|
|
||||||
|
|
||||||
body = render_template('login.html', pagedata=pagedata)
|
body = render_template('login.html', pagedata=pagedata)
|
||||||
return body
|
return body
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user