基于flask框架的注册登录

基于 Flask 框架的注册登录功能一般需要以下几个步骤：

1. 创建 Flask 应用
2. 配置数据库信息
3. 创建用户模型
4. 创建注册页面和视图函数
5. 创建登录页面和视图函数
6. 创建登出视图函数

具体实现方法如下：

1. 创建 Flask 应用

from flask import Flask

app = Flask(__name__)

2. 配置数据库信息

可以使用 SQLAlchemy 进行数据库操作，需要安装相关依赖并配置数据库信息：

from flask_sqlalchemy import SQLAlchemy

app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///database.db'
app.config['SECRET_KEY'] = 'your-secret-key'
db = SQLAlchemy(app)

3. 创建用户模型

可以创建一个 User 模型来表示用户，使用 SQLAlchemy 来进行数据库操作：

class User(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(20), unique=True, nullable=False)
    email = db.Column(db.String(120), unique=True, nullable=False)
    password = db.Column(db.String(60), nullable=False)

    def __repr__(self):
        return f"User('{self.username}', '{self.email}')"

4. 创建注册页面和视图函数

可以使用 Flask-WTF 和 Flask-Bootstrap 来实现注册页面的表单验证和美化，需要安装相关依赖：

from flask_wtf import FlaskForm
from wtforms import StringField, PasswordField, SubmitField
from wtforms.validators import DataRequired, Email, EqualTo, ValidationError
from flask_bootstrap import Bootstrap

bootstrap = Bootstrap(app)

class RegistrationForm(FlaskForm):
    username = StringField('Username', validators=[DataRequired()])
    email = StringField('Email', validators=[DataRequired(), Email()])
    password = PasswordField('Password', validators=[DataRequired()])
    confirm_password = PasswordField('Confirm Password', validators=[DataRequired(), EqualTo('password')])
    submit = SubmitField('Sign Up')

@app.route('/register', methods=['GET', 'POST'])
def register():
    form = RegistrationForm()
    if form.validate_on_submit():
        hashed_password = bcrypt.generate_password_hash(form.password.data).decode('utf-8')
        user = User(username=form.username.data, email=form.email.data, password=hashed_password)
        db.session.add(user)
        db.session.commit()
        flash('Your account has been created! You are now able to log in', 'success')
        return redirect(url_for('login'))
    return render_template('register.html', title='Register', form=form)

5. 创建登录页面和视图函数

可以使用 Flask-WTF 和 Flask-Bootstrap 来实现登录页面的表单验证和美化，需要安装相关依赖：

class LoginForm(FlaskForm):
    email = StringField('Email', validators=[DataRequired(), Email()])
    password = PasswordField('Password', validators=[DataRequired()])
    remember = BooleanField('Remember Me')
    submit = SubmitField('Login')

@app.route('/login', methods=['GET', 'POST'])
def login():
    form = LoginForm()
    if form.validate_on_submit():
        user = User.query.filter_by(email=form.email.data).first()
        if user and bcrypt.check_password_hash(user.password, form.password.data):
            login_user(user, remember=form.remember.data)
            next_page = request.args.get('next')
            return redirect(next_page) if next_page else redirect(url_for('home'))
        else:
            flash('Login Unsuccessful. Please check email and password', 'danger')
    return render_template('login.html', title='Login', form=form)

6. 创建登出视图函数

使用 Flask-Login 提供的 logout_user() 函数即可实现登出功能：

@app.route('/logout')
@login_required
def logout():
    logout_user()
    return redirect(url_for('home'))

以上是基于 Flask 框架的注册登录功能的基本实现步骤，需要注意的是代码中使用了一些第三方扩展库，需要提前安装。此外，为了保证代码的安全性，密码需要进行哈希处理。