溫馨提示×

Flask中怎么實現(xiàn)用戶授權(quán)

小億
91
2024-05-10 18:06:03
欄目: 編程語言

在Flask中實現(xiàn)用戶授權(quán)通常使用Flask提供的擴展包,比如Flask-Login和Flask-Security。下面是一個簡單的示例來實現(xiàn)用戶授權(quán):

首先,安裝Flask-Login和Flask-Security:

pip install Flask-Login Flask-Security

然后,創(chuàng)建一個Flask應用并配置用戶授權(quán):

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)

以上示例中,我們創(chuàng)建了一個簡單的Flask應用并配置了用戶授權(quán)功能。用戶模型使用了Flask-Security提供的UserMixin,用戶數(shù)據(jù)存儲使用了Flask-Security提供的SQLAlchemyUserDatastore。我們還定義了幾個路由來模擬用戶登錄、注銷和訪問需要授權(quán)的頁面。通過Flask-Login的login_required裝飾器,我們可以保護需要授權(quán)的頁面只能被登錄用戶訪問。

請注意,以上示例僅供參考,實際項目中需要根據(jù)需求進行適當?shù)男薷暮蛿U展。

0