コード例 #1
0
def confirm(confirm_code):
    user = User.find_one({"confirm_code": confirm_code})
    if not user:
        return redirect(url_for("index"))
    user.confirm_code = ""
    user.commit()
    return redirect(url_for("auth.login", success=True))
コード例 #2
0
def signup():
    if request.method == "GET":
        return render("signup.html", title="Зарегистрироваться", error_msg="")
    name = request.form["name"]
    email = request.form["email"]
    password = request.form["password"]
    if User.find_one({"email": email}) is not None:
        return render(
            "signup.html",
            title="Зарегистрироваться",
            error_msg="Пользователь с таким email уже существует!",
        )
    confirm_code = generate_confirm_code(email)
    user = User(
        name=name,
        email=email,
        password=hash_password(password),
        confirm_code=confirm_code,
    )
    user.commit()
    msg = Message(
        subject="Подтверждение аккаунта",
        sender=config.MAIL_DEFAULT_SENDER,
        recipients=[email],
        body="Ваша ссылка для подтверждения: "
        + config.APP_URL
        + url_for("auth.confirm", confirm_code=confirm_code)[1:],
    )
    mail.send(msg)
    return render("finishSignup.html", title="Завершение регистрации")
コード例 #3
0
def login():
    if get_current_user() is not None:
        return redirect(url_for("index"))
    if request.method == "GET":
        success_on_creation = request.args.get("success")
        if success_on_creation:
            return render(
                "login.html",
                title="Войти",
                msg="Аккаунт успешно создан!",
                msg_color="#01cd6c",
            )
        else:
            return render("login.html", title="Войти")
    email = request.form["email"]
    raw_password = request.form["password"]
    password = hash_password(raw_password)
    user = User.find_one({"email": email, "password": password})
    if user:
        if not user.confirm_code:
            session["email"] = email
            session["password"] = password
            return redirect(url_for("index"))
        return render(
            "login.html", title="Войти", msg="Подтвердите почту!", msg_color="#f74a5a"
        )
    return render(
        "login.html",
        title="Войти",
        msg="Произошла ошибка, проверьте введённые данные!",
        msg_color="#f74a5a",
    )
コード例 #4
0
def reset():
    email = request.form["reestablishEmail"]
    user = User.find_one({"email": email})
    if user is None:
        return render(
            "login.html",
            title="Войти",
            msg="Пользователя с таким e-mail не существует!",
            msg_color="#f74a5a",
        )
    confirm_code = generate_confirm_code(email + str(random.randint(0, 10000)))
    user.reset_code = confirm_code
    user.commit()
    msg = Message(
        subject="Восстановление пароля",
        sender=config.MAIL_DEFAULT_SENDER,
        recipients=[email],
        body="Ваша ссылка для восстановления: "
        + config.APP_URL
        + url_for("auth.reset_finish", confirm_code=confirm_code)[1:],
    )
    mail.send(msg)
    return render(
        "login.html",
        title="Войти",
        msg="Ссылка на восстановление была успешно отправлена!",
        msg_color="#01cd6c",
    )
コード例 #5
0
def reset_finish(confirm_code):
    user = User.find_one({"reset_code": confirm_code})
    if user is None:
        return render(
            "login.html",
            title="Войти",
            msg="Ссылка на восстановление пароля недействителена!",
            msg_color="#f74a5a",
        )
    if request.method == "GET":
        return render("acceptReset.html", confirm_code=confirm_code)
    password = request.form["password"]
    user.password = hash_password(password)
    user.confirm_code = ""
    user.commit()
    return redirect(url_for("index"))
コード例 #6
0
ファイル: lib.py プロジェクト: lllumineux/EGEBasis
def get_current_user():
    email = session.get("email", None)
    password = session.get("password", None)
    user = User.find_one({"email": email, "password": password})
    return user