mirror of
https://github.com/wassname/flask-security.git
synced 2026-06-28 16:20:24 +08:00
Matt Wright
89 lines
2.4 KiB
Python
89 lines
2.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
flask.ext.security.forms
|
|
~~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
Flask-Security forms module
|
|
|
|
:copyright: (c) 2012 by Matt Wright.
|
|
:license: MIT, see LICENSE for more details.
|
|
"""
|
|
|
|
from flask import request
|
|
from flask.ext.wtf import Form, TextField, PasswordField, SubmitField, \
|
|
HiddenField, Required, BooleanField, EqualTo, Email
|
|
|
|
|
|
class EmailFormMixin():
|
|
email = TextField("Email Address",
|
|
validators=[Required(message="Email not provided"),
|
|
Email(message="Invalid email address")])
|
|
|
|
|
|
class PasswordFormMixin():
|
|
password = PasswordField("Password",
|
|
validators=[Required(message="Password not provided")])
|
|
|
|
|
|
class PasswordConfirmFormMixin():
|
|
password_confirm = PasswordField("Retype Password",
|
|
validators=[EqualTo('password', message="Passwords do not match")])
|
|
|
|
|
|
class ForgotPasswordForm(Form, EmailFormMixin):
|
|
"""The default forgot password form"""
|
|
|
|
submit = SubmitField("Recover Password")
|
|
|
|
def to_dict(self):
|
|
return dict(email=self.email.data)
|
|
|
|
|
|
class LoginForm(Form, EmailFormMixin, PasswordFormMixin):
|
|
"""The default login form"""
|
|
|
|
remember = BooleanField("Remember Me")
|
|
next = HiddenField()
|
|
submit = SubmitField("Login")
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(LoginForm, self).__init__(*args, **kwargs)
|
|
|
|
if request.method == 'GET':
|
|
self.next.data = request.args.get('next', None)
|
|
|
|
|
|
class RegisterForm(Form,
|
|
EmailFormMixin,
|
|
PasswordFormMixin,
|
|
PasswordConfirmFormMixin):
|
|
"""The default register form"""
|
|
|
|
submit = SubmitField("Register")
|
|
|
|
def to_dict(self):
|
|
return dict(email=self.email.data, password=self.password.data)
|
|
|
|
|
|
class ResetPasswordForm(Form,
|
|
EmailFormMixin,
|
|
PasswordFormMixin,
|
|
PasswordConfirmFormMixin):
|
|
"""The default reset password form"""
|
|
|
|
token = HiddenField(validators=[Required()])
|
|
|
|
submit = SubmitField("Reset Password")
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(ResetPasswordForm, self).__init__(*args, **kwargs)
|
|
|
|
if request.method == 'GET':
|
|
self.token.data = request.args.get('token', None)
|
|
self.email.data = request.args.get('email', None)
|
|
|
|
def to_dict(self):
|
|
return dict(token=self.token.data,
|
|
email=self.email.data,
|
|
password=self.password.data)
|