-
Notifications
You must be signed in to change notification settings - Fork 0
/
auth_pages.py
63 lines (51 loc) · 2.17 KB
/
auth_pages.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
import flask
from flask import render_template, redirect
from flask_login import login_user, logout_user, login_required
from data.auth_forms import RegistrationForm, LoginForm
from data.users_model import User
from data import db_session
import logging
blueprint = flask.Blueprint(
'auth_pages',
__name__,
template_folder='templates'
)
@blueprint.route('/register', methods=['GET', 'POST'])
def register():
form = RegistrationForm()
if form.validate_on_submit():
db_sess = db_session.create_session()
if db_sess.query(User).filter(User.login == form.login.data).first():
logging.info(f'[auth_pages.py, register] New user {form.login.data} register failed')
return render_template('registration.html',
form=form,
message="Такой пользователь уже есть")
user = User(
login=form.login.data,
)
user.set_user_password(form.password.data)
db_sess.add(user)
db_sess.commit()
logging.info(f'[auth_pages.py, register] New user {form.login.data} registered')
return redirect('/login')
return render_template('registration.html', title='Регистрация', form=form)
@blueprint.route('/login', methods=['GET', 'POST'])
def login():
form = LoginForm()
if form.validate_on_submit():
db_sess = db_session.create_session()
user = db_sess.query(User).filter(User.login == form.login.data).first()
if user and user.check_user_password(form.password.data):
login_user(user, remember=form.remember_me.data)
logging.info(f'[auth_pages.py, login] User {form.login.data} logged in')
return redirect('/')
logging.info(f'[auth_pages.py, login] User {form.login.data} log in failed')
return render_template('login.html',
message="Неправильный логин или пароль",
form=form)
return render_template('login.html', form=form)
@blueprint.route('/logout')
@login_required
def logout():
logout_user()
return redirect('/')