Flask中怎么实现用户授权 - 问答

在Flask中实现用户授权通常使用Flask提供的扩展包，比如Flask-Login和Flask-Security。下面是一个简单的示例来实现用户授权：

首先，安装Flask-Login和Flask-Security：

pip install Flask-Login Flask-Security

然后，创建一个Flask应用并配置用户授权：

from flask import Flask, render_template
from flask_login import LoginManager, login_user, logout_user, login_required
from flask_security import Security, UserMixin, RoleMixin, SQLAlchemyUserDatastore

from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)
app.config['SECRET_KEY'] = 'your_secret_key'
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///users.db'

db = SQLAlchemy(app)

# 定义用户模型
class User(db.Model, UserMixin):
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(50), unique=True)
    password = db.Column(db.String(255))

# 初始化Flask-Security
user_datastore = SQLAlchemyUserDatastore(db, User, None)
security = Security(app, user_datastore)

# 初始化Flask-Login
login_manager = LoginManager()
login_manager.init_app(app)

@login_manager.user_loader
def load_user(user_id):
    return User.query.get(user_id)

# 路由定义
@app.route('/')
def home():
    return render_template('home.html')

@app.route('/login')
def login():
    # 登录逻辑
    user = User.query.filter_by(username='admin').first()
    login_user(user)
    return 'Logged in successfully!'

@app.route('/logout')
@login_required
def logout():
    logout_user()
    return 'Logged out successfully!'

@app.route('/profile')
@login_required
def profile():
    return 'User profile page'

if __name__ == '__main__':
    db.create_all()
    app.run(debug=True)

以上示例中，我们创建了一个简单的Flask应用并配置了用户授权功能。用户模型使用了Flask-Security提供的UserMixin，用户数据存储使用了Flask-Security提供的SQLAlchemyUserDatastore。我们还定义了几个路由来模拟用户登录、注销和访问需要授权的页面。通过Flask-Login的login_required装饰器，我们可以保护需要授权的页面只能被登录用户访问。

请注意，以上示例仅供参考，实际项目中需要根据需求进行适当的修改和扩展。

0 赞

0 踩