20 用户认证:用户注册
在本篇中,我们将讨论如何在 Flask 应用中实现用户注册功能。这个功能将以之前的数据库迁移为基础,我们也将为后续的用户登录功能做好准备。我们会通过实例进行讲解,以便于理解和应用。
环境准备
首先,确保你已经创建了一些必要的文件和目录,并且将 SQLAlchemy 和 Flask-Migrate 安装到你的虚拟环境中。如果你还没有设置这些,可以参照以下命令:
pip install Flask Flask-SQLAlchemy Flask-Migrate
创建用户模型
在之前的数据库迁移中,我们已经设置了 Flask 应用以及数据库配置。现在,我们将创建一个用户模型,以便存储用户的数据。
from flask_sqlalchemy import SQLAlchemy
db = SQLAlchemy()
class User(db.Model):
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(80), unique=True, nullable=False)
password_hash = db.Column(db.String(128), nullable=False)
def __repr__(self):
return f'<User {self.username}>'
在上述代码中,我们定义了一个 User
类,它继承自 db.Model
。此模型包含三个字段:
id
: 用户的唯一标识。username
: 用户名,要求唯一且不能为空。password_hash
: 用于存储加密后的密码。
创建用户注册表单
接下来,我们将创建一个用于用户注册的表单。我们将使用 Flask-WTF 来简化表单处理。
首先,安装 Flask-WTF:
pip install Flask-WTF
然后,创建一个表单文件 forms.py
:
from flask_wtf import FlaskForm
from wtforms import StringField, PasswordField, SubmitField
from wtforms.validators import DataRequired, EqualTo
class RegistrationForm(FlaskForm):
username = StringField('Username', validators=[DataRequired()])
password = PasswordField('Password', validators=[DataRequired()])
confirm_password = PasswordField('Confirm Password', validators=[DataRequired(), EqualTo('password')])
submit = SubmitField('Register')
在这个表单中,我们创建了三个字段:
username
: 用户名,不能为空。password
: 密码,不能为空。confirm_password
: 确认密码,以确保用户输入相同的密码。
注册视图函数
现在,我们来实现用户注册的视图。我们将在 app.py
文件中添加注册处理逻辑。
from flask import Flask, render_template, redirect, url_for, flash
from forms import RegistrationForm
from models import db, User
from werkzeug.security import generate_password_hash
app = Flask(__name__)
app.config['SECRET_KEY'] = 'your_secret_key'
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///site.db'
db.init_app(app)
@app.route('/register', methods=['GET', 'POST'])
def register():
form = RegistrationForm()
if form.validate_on_submit():
hashed_password = generate_password_hash(form.password.data, method='sha256')
new_user = User(username=form.username.data, password_hash=hashed_password)
db.session.add(new_user)
db.session.commit()
flash('Registration successful! You can now log in.', 'success')
return redirect(url_for('login')) # 假设我们有一个登录视图
return render_template('register.html', form=form)
if __name__ == '__main__':
app.run(debug=True)
在 register
视图中,我们首先实例化 RegistrationForm
,然后检查表单是否通过验证。如果验证通过,使用 generate_password_hash
对用户的密码进行加密,并创建一个新用户对象。最后,将用户存入数据库,并重定向到登录页。
创建注册模板
最后,我们需要一个 HTML 模板来渲染注册表单。创建文件 register.html
,并添加以下内容:
<!doctype html>
<html lang="en">
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<title>Register</title>
</head>
<body>
<h1>Register</h1>
<form method="POST">
{{ form.hidden_tag() }}
<p>
{{ form.username.label }}<br>
{{ form.username(size=32) }}<br>
{% for error in form.username.errors %}
<span style="color: red;">[{{ error }}]</span>
{% endfor %}
</p>
<p>
{{ form.password.label }}<br>
{{ form.password(size=32) }}<br>
{% for error in form.password.errors %}
<span style="color: red;">[{{ error }}]</span>
{% endfor %}
</p>
<p>
{{ form.confirm_password.label }}<br>
{{ form.confirm_password(size=32) }}<br>
{% for error in form.confirm_password.errors %}
<span style="color: red;">[{{ error }}]</span>
{% endfor %}
</p>
<p>{{ form.submit() }}</p>
</form>
</body>
</html>
在这个模板中,我们使用了 Flask-WTF 提供的表单渲染,确保显示任何验证错误。
小结
在本篇中,我们完成了用户注册功能的实现。用户输入的密码会通过 SHA-256 算法进行加密,增强了安全性。用户注册后会被重定向到登录页面,接下来我们将在下篇中实现用户登录功能。
如果你有任何问题或需要更多帮助,请随时提问!