当前位置: 代码迷 >> python >> Flask-Security:自定义注册
  详细解决方案

Flask-Security:自定义注册

热度:83   发布时间:2023-06-13 15:30:35.0

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

我也不确定是否需要编写自己的app.route函数来处理创建新用户和/或检查它们是否已经存在。 我希望能够使用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是否会处理它。 接下来是我的logform和signupform类:

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' ,以便注册不会出错。

您不需要为注册定义新的路线。

在从RegisterForm继承的signupform类中,您只需要使用自定义逻辑覆盖validate方法,并根据需要返回TrueFalse

在用于注册的html模板中,操作网址为{{ url_for_security('register') }}

自定义登录表单也是{{ url_for_security('login') }}操作为{{ url_for_security('login') }}

  相关解决方案