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))
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="Завершение регистрации")
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",
)
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",
)
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"))
def get_current_user():
email = session.get("email", None)
password = session.get("password", None)
user = User.find_one({"email": email, "password": password})
return user
