python用户登陆

要实现用户登录，可以使用 Python 的 Flask 框架和 Flask-Login 扩展。Flask-Login 提供了用户认证和会话管理的功能，可以轻松地实现用户登录和注销等功能。

首先需要安装 Flask 和 Flask-Login：

pip install flask flask-login

然后在 Flask 应用中初始化 Flask-Login：

from flask import Flask
from flask_login import LoginManager

app = Flask(__name__)
login_manager = LoginManager()
login_manager.init_app(app)

接下来需要定义一个 User 类，用于表示用户信息。这个类需要实现 Flask-Login 中的 UserMixin 接口，以便 Flask-Login 可以管理用户会话。

from flask_login import UserMixin

class User(UserMixin):
    def __init__(self, id, username, password):
        self.id = id
        self.username = username
        self.password = password

    def get_id(self):
        return str(self.id)

然后需要实现一个用户认证函数，用于验证用户输入的用户名和密码是否正确。这个函数需要返回一个 User 对象，表示认证成功，或者返回 None，表示认证失败。

from werkzeug.security import check_password_hash

def authenticate(username, password):
    # 在这里查询数据库或其他存储方式，验证用户名和密码是否正确
    if username == 'admin' and check_password_hash('pbkdf2:sha256:150000$zjJZ5v9L$e7f6d8c3a4b1f7d9c8e7a6d5f4b3c2a1d9e8f7c6b5a4d3e2c1b0a9b8c7d6e5f4', password):
        return User(1, username, password)
    else:
        return None

最后需要实现一个回调函数，用于根据用户 ID 获取 User 对象。这个函数需要返回一个 User 对象，表示用户存在，或者返回 None，表示用户不存在。

@login_manager.user_loader
def load_user(user_id):
    # 在这里查询数据库或其他存储方式，根据 user_id 获取 User 对象
    if user_id == '1':
        return User(1, 'admin', 'pbkdf2:sha256:150000$zjJZ5v9L$e7f6d8c3a4b1f7d9c8e7a6d5f4b3c2a1d9e8f7c6b5a4d3e2c1b0a9b8c7d6e5f4')
    else:
        return None

现在可以在 Flask 应用中使用 Flask-Login 提供的 login_required 装饰器来保护需要登录才能访问的页面。

from flask_login import login_required

@app.route('/dashboard')
@login_required
def dashboard():
    return 'Welcome to the dashboard!'