Source code for invenio_accounts.forms

# -*- coding: utf-8 -*-
# This file is part of Invenio.
# Copyright (C) 2016-2018 CERN.
# Invenio is free software; you can redistribute it and/or modify it
# under the terms of the MIT License; see LICENSE file for more details.

"""Additional non-userprofile fields used during registration.

Currently supported: recaptcha
from flask import request
from flask_security.forms import NextFormMixin
from flask_wtf import FlaskForm, Recaptcha, RecaptchaField
from invenio_i18n import gettext as _
from wtforms import FormField, HiddenField

from invenio_accounts.proxies import current_datastore

[docs]class RegistrationFormRecaptcha(FlaskForm): """Form for editing user profile.""" recaptcha = RecaptchaField( validators=[Recaptcha(message=_("Please complete the reCAPTCHA."))] )
[docs]class RevokeForm(FlaskForm): """Form for revoking a session.""" sid_s = HiddenField()
[docs]def confirm_register_form_factory(Form, app): """Return confirmation for extended registration form.""" class ConfirmRegisterForm(Form, NextFormMixin): def __init__(self, *args, **kwargs): """Adds next parameter from URL.""" super(ConfirmRegisterForm, self).__init__(*args, **kwargs) if not = request.args.get("next", "") if app.config.get("RECAPTCHA_PUBLIC_KEY") and app.config.get( "RECAPTCHA_PRIVATE_KEY" ): class ConfirmRegisterWithCaptchaForm(ConfirmRegisterForm): recaptcha = FormField(RegistrationFormRecaptcha, separator=".") return ConfirmRegisterWithCaptchaForm return ConfirmRegisterForm
[docs]def register_form_factory(Form, app): """Return extended registration form.""" if app.config.get("RECAPTCHA_PUBLIC_KEY") and app.config.get( "RECAPTCHA_PRIVATE_KEY" ): class RegisterForm(Form): recaptcha = FormField(RegistrationFormRecaptcha, separator=".") return RegisterForm return Form
[docs]def login_form_factory(Form, app): """Return extended login form.""" class LoginForm(Form): pass return LoginForm
[docs]def send_confirmation_form_factory(Form, app): """Return extended login form.""" class SendConfirmationEmailView(Form): """Form which sends confirmation instructions. If user not in the system, do not raise but ignore. The overriden behaviour exists because the endpoint is public and users can take an insight on which emails exist in the system otherwise. """ def validate(self, extra_validators=None): self.user = current_datastore.get_user(["email"]) # Form is valid if user exists and they are not yet confirmed if self.user is not None and self.user.confirmed_at is None: return True return False return SendConfirmationEmailView