mirror of
https://github.com/lucaspalomodevelop/eventcally.git
synced 2026-03-13 08:09:37 +00:00
84 lines
2.6 KiB
Python
84 lines
2.6 KiB
Python
from flask_babel import gettext, lazy_gettext
|
|
from flask_security import url_for_security
|
|
from flask_security.forms import (
|
|
ConfirmRegisterForm,
|
|
EqualTo,
|
|
ForgotPasswordForm,
|
|
LoginForm,
|
|
get_form_field_label,
|
|
)
|
|
from flask_security.utils import localize_callback
|
|
from flask_wtf import FlaskForm
|
|
from wtforms import BooleanField, PasswordField, SubmitField
|
|
from wtforms.validators import DataRequired
|
|
|
|
from project.forms.common import get_accept_tos_markup
|
|
from project.views.utils import flash_message
|
|
|
|
|
|
class ExtendedConfirmRegisterForm(ConfirmRegisterForm):
|
|
password = PasswordField(
|
|
get_form_field_label("password"), validators=[DataRequired()]
|
|
)
|
|
password_confirm = PasswordField(
|
|
get_form_field_label("retype_password"),
|
|
validators=[
|
|
EqualTo("password", message="RETYPE_PASSWORD_MISMATCH"),
|
|
DataRequired(),
|
|
],
|
|
)
|
|
accept_tos = BooleanField(validators=[DataRequired()])
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(ExtendedConfirmRegisterForm, self).__init__(*args, **kwargs)
|
|
self._fields["email"].description = lazy_gettext("register_email_desc")
|
|
self._fields["accept_tos"].label.text = get_accept_tos_markup()
|
|
|
|
|
|
class ExtendedLoginForm(LoginForm):
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self._fields["email"].flags.required = True
|
|
|
|
def validate(self, **kwargs):
|
|
result = super().validate(**kwargs)
|
|
|
|
if not result and self.requires_confirmation:
|
|
flash_message(
|
|
gettext("login_confirmation_required"),
|
|
url_for_security("send_confirmation"),
|
|
localize_callback("Resend confirmation instructions"),
|
|
"danger",
|
|
)
|
|
|
|
return result
|
|
|
|
|
|
class ExtendedForgotPasswordForm(ForgotPasswordForm):
|
|
def validate(self, **kwargs):
|
|
result = super().validate(**kwargs)
|
|
|
|
if not result and self.requires_confirmation:
|
|
flash_message(
|
|
gettext("login_confirmation_required"),
|
|
url_for_security("send_confirmation"),
|
|
localize_callback("Resend confirmation instructions"),
|
|
"danger",
|
|
)
|
|
|
|
return result
|
|
|
|
|
|
class AuthorizeForm(FlaskForm):
|
|
allow = SubmitField(lazy_gettext("Allow"))
|
|
deny = SubmitField(lazy_gettext("Deny"))
|
|
|
|
|
|
class AcceptTosForm(FlaskForm):
|
|
accept_tos = BooleanField(validators=[DataRequired()])
|
|
submit = SubmitField(lazy_gettext("Confirm"))
|
|
|
|
def __init__(self, **kwargs):
|
|
super(AcceptTosForm, self).__init__(**kwargs)
|
|
self._fields["accept_tos"].label.text = get_accept_tos_markup()
|