溫馨提示×

Flask中怎么實現用戶授權

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

在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
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女