Flask-login 簡單 Demo, 最簡單的實現登錄方法
* 當使用MethodView的時候,login_required 使用方法
class XxxxView(MethodView):
decorators = [login_required] # 相當於 @login_required 裝飾圈
def get(self):
主文件
from flask_sqlalchemy import SQLAlchemy
from flask import Flask, redirect, url_for, request
from flask_login import LoginManager, current_user, login_user, login_required
app = Flask(__name__)
app.secret_key = 'Sqsdsffqrhgh.,/1#$%^&'
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///./db/blog.db'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = True
app.debug = True
db = SQLAlchemy(app)
login_manager = LoginManager()
login_manager.init_app(app)
password = '123' #只要用戶輸入的密碼是 123 就可以登錄
@login_manager.user_loader #使用user_loader裝飾器的回調函數非常重要,他將決定 user 對象是否在登錄狀態
def user_loader(id): #這個id參數的值是在 login_user(user)中傳入的 user 的 id 屬性
from model.model import User
user = User.query.filter_by(id=id).first()
return user
# 添加登錄視圖,如果是GET方法,返回一個簡單的表單
@app.route('/login/', methods=['GET', 'POST'])
def login():
from model.model import User
if request.method == 'GET':
return '''
<form action="#" method="POST">
<span>請輸入賬號</span>
<input type="text" name="name" id="name" placeholder="name">
<span>請輸入密碼</span>
<input type="password" name="pw" id="pw" placeholder="password">
<input type="submit" name="submit">
</form>
'''
name = request.form.get('name')
if request.form.get('pw') == password:
user = User.query.filter_by(name=name).first()
if not user:
user = User(name=name)
db.session.add(user)
db.session.commit()
login_user(user)
return redirect(url_for('index'))
return 'Bad login'
#如果密碼是 123 就會跳轉到視圖函數 index 上
@app.route('/index/')
@login_required
def index():
user = current_user
return 'HelloWorld'
if __name__ == '__main__':
app.run()
model.py
from model import *
'''
創建類的時候繼承UserMixin ,有一些用戶相關屬性
* is_authenticated :是否被驗證
* is_active : 是否被激活
* is_anonymous : 是否是匿名用戶
* get_id() : 獲得用戶的id,並轉換爲 Unicode 類型
'''
class User(db.Model,UserMixin):
__tablename__ = 'login_users'
id = db.Column(db.Integer,primary_key=True)
name = db.Column(db.String(50),unique=True)
login_count = db.Column(db.Integer,default=0)
last_login_ip = db.Column(db.String(128),default='unknown')
在來一個功能全點的Demo
from flask_sqlalchemy import SQLAlchemy
from flask import Flask, redirect, url_for, request, flash, render_template
from flask_login import LoginManager, login_user, login_required, logout_user
app = Flask(__name__)
app.secret_key = 'Sqsdsffqrhgh.,/1#$%^&'
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///./db/blog.db'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = True
app.debug = True
db = SQLAlchemy(app)
login_manager = LoginManager()
login_manager.init_app(app)
login_manager.login_view = "login" # 定義登錄的 視圖
login_manager.login_message = '請登錄以訪問此頁面' # 定義需要登錄訪問頁面的提示消息
@login_manager.user_loader # 使用user_loader裝飾器的回調函數非常重要,他將決定 user 對象是否在登錄狀態
def user_loader(id): # 這個id參數的值是在 login_user(user)中傳入的 user 的 id 屬性
from model.model import User
user = User.query.filter_by(id=id).first()
return user
# 添加登錄視圖,如果是GET方法,返回一個簡單的表單
@app.route('/login/', method=['GET', 'POST'])
def login():
from model.model import User
if request.method == 'POST':
name = request.form.get('name')
user = User.query.filter_by(name=name).first()
if not user:
flash('該用戶不存在')
elif request.form.get('pwd') != user.pwd:
flash('密碼錯誤')
else:
login_user(user, remember=True)
next_url = request.args.get('next')
return redirect(next_url or url_for('login_success'))
return render_template('login.html') # 如果密碼是 123 就會跳轉到視圖函數 index 上
@app.route('/')
@login_required
def index():
return 'Hello Tank'
@app.route('/succees/')
@login_required
def login_success():
return render_template('base.html')
@app.route('/logout/')
@login_required
def logout():
logout_user() # 登出用戶
return '已經退出登錄'
@app.errorhandler(404)
def page_not_found(error):
return render_template('page_not_found.html'), 404
if __name__ == '__main__':
app.run()