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 .. import app, lib, models
|
||||
from . import forms
|
||||
|
||||
@app.route('/login', methods=['GET', 'POST'])
|
||||
|
||||
@app.route('/login')
|
||||
def login():
|
||||
"""
|
||||
Логин
|
||||
"""
|
||||
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)
|
||||
return body
|
||||
|
||||
|
||||
Reference in New Issue
Block a user