File size: 2,067 Bytes
6bcf797 |
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 |
from flask import Blueprint, render_template, redirect, url_for, flash
from flask_login import login_user, logout_user, login_required
from werkzeug.security import generate_password_hash, check_password_hash
from models import User, db # Используем SQLAlchemy модели
from forms import LoginForm, RegistrationForm
from datetime import datetime
auth_bp = Blueprint('auth_bp', __name__)
@auth_bp.route('/login', methods=['GET', 'POST'])
def login():
form = LoginForm()
if form.validate_on_submit():
user = User.query.filter_by(email=form.email.data).first() # Запрос через SQLAlchemy
if user and user.check_password(form.password.data):
login_user(user) # Используем модель User из Flask-SQLAlchemy
return redirect(url_for('index'))
flash('Неверный email или пароль', 'danger')
return render_template('auth/login.html', form=form)
@auth_bp.route('/register', methods=['GET', 'POST'])
def register():
form = RegistrationForm()
if form.validate_on_submit():
try:
# Создаем нового пользователя через SQLAlchemy
user = User(
username=form.username.data,
email=form.email.data,
password_hash=generate_password_hash(form.password.data)
)
db.session.add(user)
db.session.commit()
flash('Регистрация прошла успешно! Теперь вы можете войти.', 'success')
return redirect(url_for('auth_bp.login'))
except Exception as e:
db.session.rollback()
flash('Пользователь с таким email или именем уже существует', 'danger')
return render_template('auth/register.html', form=form)
@auth_bp.route('/logout')
@login_required
def logout():
logout_user()
return redirect(url_for('auth_bp.login')) |