SQLAlchemy_PostgreSQL_flask_login简单实现登录逻辑

• reference
  https://blog.csdn.net/mouday/article/details/115948156
  https://flask-login.readthedocs.io/en/latest/
  http://www.pythondoc.com/flask-login/
  https://www.jianshu.com/p/5a25413b4828

• code

from flask import Flask, request, redirect, url_for, flash, render_template
from flask_login import LoginManager, UserMixin, login_user, login_required, logout_user
from flask_sqlalchemy import SQLAlchemy


app = Flask(__name__)

# 配置数据库信息
app.config['SQLALCHEMY_DATABASE_URI'] = 'postgresql://dbuser:password@host:port/test_db'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False

app.secret_key = '123456'  # session设置用

login_manager = LoginManager(app)
login_manager.session_protection = 'strong'
login_manager.login_view = 'login'  # 设置用户登录视图函数 endpoint
login_manager.login_message = u"请登录！"
login_manager.init_app(app)  # 初始化应用

db = SQLAlchemy(app)
db.init_app(app)


@app.route('/')
def hello_world():
    return 'Hello World!'


@login_manager.user_loader
def load_user(user_id: str):
    return Admin.query.filter_by(id=user_id).first()


@app.route('/test')
@login_required
def test():
    return 'TEST!'


@app.route('/login', methods=['GET', 'POST'])
def login():
    user_id = request.args.get('userid', '')
    password = request.args.get('password', '')
    if user_id == '':
        return "Please input correct"
    user = Admin.query.filter_by(id=user_id).first()
    if user and user.password_hash == password:
        login_user(user)
        return render_template('login.html')

    return "Please Login"


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


class Admin(db.Model, UserMixin):

    __tablename__ = 'admin'
    id = db.Column(db.Integer, autoincrement=True, primary_key=True)
    username = db.Column(db.String(20))
    password_hash = db.Column(db.String(120))
    blog_title = db.Column(db.String(60))
    blog_sub_title = db.Column(db.String(100))
    name = db.Column(db.String(30))
    about = db.Column(db.TEXT)

    def save(self):
        db.session.add(self)
        db.session.commit()

    def __str__(self):
        return self.name


if __name__ == '__main__':
    # user1 = Admin(id=5, username='小王', password_hash="test", name="wyh")
    # user1.save()
    # print(user1)
    app.run()

• Flask jinja2给模板传递变量的两种方式

1. 直接在 render_template中包含参数，传递到模板中去。
2. 自定义变量，然后在模板中可以自由使用。用@app.context_processor装饰一个自定义函数，然后返回一个dict字典变量。