在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裝飾器,我們可以保護需要授權的頁面只能被登錄用戶訪問。
請注意,以上示例僅供參考,實際項目中需要根據需求進行適當的修改和擴展。