Dyxless/dyxless/auth.py

import datetime

from flask import (
    Blueprint,
    render_template,
    redirect,
    url_for,
    request,
    flash,
    current_app,
    Markup,
)
from werkzeug.security import check_password_hash
from flask_login import current_user, login_user, logout_user
from itsdangerous import URLSafeTimedSerializer

from . import db
from .models import User
from .mails import send_async_email

auth = Blueprint("auth", __name__)


def generate_confirmation_token(email):
    serializer = URLSafeTimedSerializer(current_app.config["SECRET_KEY"])
    return serializer.dumps(
        email, salt=current_app.config["SECURITY_PASSWORD_SALT"]
    )


def send_confirmation_email(email):

    token = generate_confirmation_token(email)
    confirm_url = url_for("auth.confirm_email", token=token, _external=True)

    try:
        send_async_email(
            subject="Подтверждение регистрации",
            recipients=[email],
            html=render_template(
                "mail/confirmation_mail.html", confirm_url=confirm_url
            ),
        )
    except:
        return False

    return True


@auth.route("/signup", methods=["GET", "POST"])
def signup():

    if current_user.is_authenticated:
        return redirect(url_for("main.index"))

    elif request.method == "GET":
        return render_template("signup.html")

    elif request.method == "POST":
        email = request.form.get("email")
        username = request.form.get("username")
        password = request.form.get("password")
        password_confirmation = request.form.get("password_confirmation")

        if password != password_confirmation:
            flash(
                "Пароли не совпадают",
                "is-danger",
            )
            return redirect(url_for("auth.signup"))

        user = User.query.filter_by(email=email).first()

        if user:
            flash(
                Markup(
                    f"Указанная почта уже используется. <a href='{url_for('auth.login')}'>Перейти к странице входа</a>"
                ),
                "is-warning",
            )
            return redirect(url_for("auth.signup"))

        user = User.query.filter_by(username=username).first()

        if user:
            flash(
                "Указанное имя уже используется. <a href='{url_for('auth.login')}'>Перейти к странице входа</a>",
                "is-warning",
            )
            return redirect(url_for("auth.signup"))

        new_user = User(
            email=email,
            password=password,
            username=username,
        )

        db.session.add(new_user)
        db.session.commit()

        confirmation_email_sending = send_confirmation_email(new_user.email)

        if confirmation_email_sending:
            flash(
                "На вашу почту была выслана ссылка для подтверждения регистрации",
                "is-success",
            )
        else:
            flash(
                "Что-то пошло не так, свяжитесь с администратором",
                "is-danger",
            )

        return redirect(url_for("auth.login"))


@auth.route("/resend_confirmation", methods=["GET", "POST"])
def resend_confirmation():

    if current_user.is_authenticated and current_user.is_confirmed:
        return redirect(url_for("main.index"))

    elif request.method == "GET":
        return render_template("resend_confirmation.html")

    elif request.method == "POST":
        email = request.form.get("email")

        user = User.query.filter_by(email=email).first()
        if not user or user.is_confirmed:
            flash(
                "Аккаунт уже подтвержден или не зарегистрирован", "is-warning"
            )
            return redirect(url_for("auth.login"))

        confirmation_email_sending = send_confirmation_email(email)

        if confirmation_email_sending:
            flash(
                "На вашу почту была выслана новая ссылка для подтверждения регистрации",
                "is-success",
            )
        else:
            flash(
                "Что-то пошло не так, свяжитесь с администратором",
                "is-danger",
            )

        return redirect(url_for("auth.login"))


def confirm_token(token, expiration):
    serializer = URLSafeTimedSerializer(current_app.config["SECRET_KEY"])
    try:
        email = serializer.loads(
            token,
            salt=current_app.config["SECURITY_PASSWORD_SALT"],
            max_age=expiration,
        )
    except:
        return False
    return email


@auth.route("/confirm/<token>")
def confirm_email(token):

    email = confirm_token(token, int(current_app.config["TOKEN_EXPIRATION"]))
    if not email:
        flash(
            Markup(
                f"Ссылка подтверждения невалидна или устарела. <a href='{url_for('auth.resend_confirmation')}'>Повторить отправку</a>"
            ),
            "is-danger",
        )
        return redirect(url_for("auth.login"))

    user = User.query.filter_by(email=email).first()
    if user.is_confirmed:
        flash("Аккаунт уже подтвержден", "is-warning")
    else:
        user.is_confirmed = True
        db.session.commit()
        flash("Ваш аккаунт подвтержден!", "is-success")
    return redirect(url_for("auth.login"))


@auth.route("/login", methods=["GET", "POST"])
def login():

    if current_user.is_authenticated:
        return redirect(url_for("main.index"))

    elif request.method == "GET":
        return render_template("login.html")

    elif request.method == "POST":
        email = request.form.get("email")
        password = request.form.get("password")
        remember = True if request.form.get("remember") else False

        user = User.query.filter_by(email=email).first()

        if not user or not check_password_hash(user.password, password):
            flash("Неверный логин или пароль", "is-danger")
            return redirect(url_for("auth.login"))
        elif not user.is_confirmed:
            flash(
                Markup(
                    f"Аккаунт еще не активирован. Пожалуйста, проверьте вашу почту. <a href='{url_for('auth.resend_confirmation')}'>Повторить отправку</a>"
                ),
                "is-warning",
            )
            return redirect(url_for("auth.login"))

        login_user(user, remember=remember)

        user.last_login = datetime.datetime.now()

        db.session.commit()

        return redirect(url_for("main.profile"))


@auth.route("/logout")
def logout():
    logout_user()
    flash("Вы успешно вышли!", "is-success")
    return redirect(url_for("main.index"))