Update login

This commit is contained in:
RemiZOffAlex
2020-02-16 22:58:14 +03:00
parent 98180767eb
commit c4f4295dcc
2 changed files with 2 additions and 38 deletions

View File

@@ -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()])

View File

@@ -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