Numerum/Incepted
Numerum/Incepted
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Добавил авторизацию по логину, в страницу регистрации добавил кнопку перехода на страницу авторизации, на главной странице изменил кнопку регистрации на кнопку авторизации

Browse Source
This commit is contained in:
Andrei 2022-12-20 13:10:16 +05:00
parent e7f147f223
commit 4a8a967d86
4 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
forms/login.py
View File

@ -1,10 +1,10 @@
from flask_wtf import FlaskForm from flask_wtf import FlaskForm
from wtforms import EmailField, PasswordField, BooleanField, SubmitField from wtforms import StringField, PasswordField, BooleanField, SubmitField
from wtforms.validators import DataRequired from wtforms.validators import DataRequired
class LoginForm(FlaskForm): class LoginForm(FlaskForm):
login = EmailField('Почта', validators=[DataRequired()]) login = StringField('Почта', validators=[DataRequired()])
password = PasswordField('Пароль', validators=[DataRequired()]) password = PasswordField('Пароль', validators=[DataRequired()])
remember_me = BooleanField('Запомнить меня') remember_me = BooleanField('Запомнить меня')
submit = SubmitField('Войти') submit = SubmitField('Войти')

7
main.py
View File

@ -97,8 +97,11 @@ def login():
danger = request.args.get('danger') if request.args.get('danger') else False danger = request.args.get('danger') if request.args.get('danger') else False
form = LoginForm() form = LoginForm()
if form.validate_on_submit(): if form.validate_on_submit():
db_sess = db_session.create_session() data_session = db_session.create_session()
user = db_sess.query(User).filter(User.email == form.login.data).first() user = data_session.query(User).filter(User.email == form.login.data).first()
if not user:
user = data_session.query(User).filter(User.login == form.login.data).first()
data_session.close()
if user and user.check_password(form.password.data): if user and user.check_password(form.password.data):
if user.activated: if user.activated:
login_user(user, remember=form.remember_me.data) login_user(user, remember=form.remember_me.data)

4
templates/main.html
View File

@ -13,9 +13,9 @@
<strong class="header_title_2">Самый удобный сайт для создания проектов</strong> <strong class="header_title_2">Самый удобный сайт для создания проектов</strong>
</div> </div>
<div class="header_buttons"> <div class="header_buttons">
<a class="header_button" href="/register"> <a class="header_button" href="/login">
<div class="header_button_text"> <div class="header_button_text">
<p>Регистрация</p> <p>Авторизация</p>
</div> </div>
</a> </a>
<a class="header_button" id="link_to_about" href="#about_block"> <a class="header_button" id="link_to_about" href="#about_block">

3
templates/register.html
View File

@ -37,6 +37,9 @@
</div> </div>
<div class="form_data_button"> <div class="form_data_button">
{{ form.submit(type="submit", class="register_button") }} {{ form.submit(type="submit", class="register_button") }}
<a class="auth_button" type="submit" href="/login">
<div class="auth"><strong>Уже есть аккаунт?</strong></div>
</a>
</div> </div>
</div> </div>
<div class="message_block"> <div class="message_block">