Flask-Security：自定义注册

Question

我正在尝试使用 Flask-Security 登录和注册用户。但是，我正在尝试做两件事。第一个是将默认的 /registration 更改为 /signup，我相信我做对了。第二个是我要自定义注册表单。我可以加载页面，但是每当我提交表单时都没有任何反应。没有数据被发送到后端或插入到 sqlite 数据库中。

我也不确定是否需要编写自己的 app.route 函数来处理创建新用户 and/or 检查他们是否已经存在。我希望能够使用 flask-security 的寄存器，但只是对其进行调整。如果可能的话。

这是我的主要 python 文件：

from flask import Flask, redirect, flash, session, request, render_template
from flask_sqlalchemy import SQLAlchemy

import os, time


app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///data.db'
app.config['DATABASE'] = 'data.db'
app.config['SECURITY_TRACKABLE'] = True
app.config['SECURITY_REGISTERABLE'] = True
app.config['SECURITY_REGISTER_URL'] = '/signup'
app.config['SECURITY_REGISTER_USER_TEMPLATE'] = 'security/register.html'
# enable this if you want to send email and confirm
# app.config['SECURITY_CONFIRMABLE'] = True
db = SQLAlchemy(app)

from user import Security, User, user_datastore
from forms import logform, signupform, RegisterForm

security = Security(app, user_datastore, register_form=signupform,  login_form=logform,
                    confirm_register_form=signupform)

db.create_all()
db.session.commit()


@app.route('/')
def hello_world():
    log = logform(csrf_enabled=False)
    flash("HELLO.")
    return render_template("index.html", form=log)


@app.route('/login')
def login():
    print "/LOGIN REQUESTED"
    form = logform(csrf_enabled=False)
    return render_template("security/login_user.html", login_user_form=form, security=security)


#not sure if I need this function at all
@app.route('/signup', methods=['GET', 'POST'])
def signup():
    print "/SIGNUP REQUESTED"
    form = signupform(csrf_enabled=False)
    # form = RegisterForm()

    if request.method == 'GET':
        return render_template("security/register.html", register_user_form=form, security=security)

    else:
        #log = logform(csrf_enabled=False)
        if form.validate_on_submit() or form.validate():
            print "HERE NOW SOMEONE SUBMITTED SOMETHING"
            print form.email.data
            print form.username.data
            print form.password.data
            user_datastore.create_user(form)
            db.session.commit()

        return redirect('/')

我不确定我是否需要 /signup 功能，或者 flask-security 是否会默认处理它。接下来是我的日志表单和注册表单类:

from flask_wtf import FlaskForm
from wtforms import StringField, PasswordField, BooleanField
from wtforms.validators import DataRequired, Length, Email, EqualTo
from flask_security.forms import LoginForm, RegisterForm, Form


class logform(LoginForm):
    login = StringField('login', validators=[DataRequired()])
    # password = PasswordField('password', validators=[DataRequired()])


class signupform(RegisterForm):
    # email = StringField('email', validators=[Email(), DataRequired()])
    username = StringField('username', [DataRequired()])
    # password = PasswordField('password', validators=[DataRequired()])

这是我正在尝试使用的当前注册表单。我不确定是否应该使用 url_for('signup')，因为 flask-security 的默认注册表单使用 url_for_security('register')。

<!DOCTYPE html>

{% extends "base.html" %}

{% block content %}

    <div id="signup">
        <h1>Create Account.</h1>
        <h3>TESTING</h3>
        {# not sure what to put as my action. default flask-security uses url_for_security('register') #}
        <form id="logfrm" name='register_form' method="POST" action="{{ url_for('signup') }}">
            {# figure out how to make input required #}

            {{ register_user_form.email(placeholder="email", type="email") }}<br>
            {{ register_user_form.username(placeholder="username") }}<br>
            {{ register_user_form.password(placeholder="password", type="password") }}<br>
            {# register_user_form.confirm(placeholder="confirm password") #}<br>
            {{ register_user_form.submit }}
            <input type="submit" value="SIGNUP" >

        </form>
    </div>

{% endblock %}

任何帮助都会很棒，谢谢！

感谢您的帮助。这确实消除了我遇到的大部分麻烦。唯一错误的是我的表格。我忘记了 RegisterForm 需要密码确认所以我还添加了： {{ register_user_form.password_confirm(...) }}

并且还在我的主文件中包含一个 app.config['SECRET_KEY'] = 'blah'，这样注册就不会运行出错。

Answer 1

您无需为注册定义新路线。

在继承自 RegisterForm 的 signupform class 中，您只需使用自定义逻辑和 return True 或 False 视情况而定。

在您的 html 注册模板中，操作 url 是 {{ url_for_security('register') }}

这同样适用于自定义登录表单 - 尽管 url 操作是 {{ url_for_security('login') }}

Flask-Security：自定义注册

Flask-Security: Customizing Registration

python

flask

flask-sqlalchemy

registration

flask-security