Exemplos de JWT.jwt_headers_callback em Python

Exemplo n.º 1

from flask import jsonify
from sqlalchemy.exc import SQLAlchemyError
from flask_jwt import JWT
from user_api.resources.user import UserResource
from user_api import (user_api_app, api, db)
from user_api.auth import (generate_jwt_token, generate_jwt_payload,
                           generate_jwt_headers, generate_auth_response,
                           generate_error_response, authenticate,
                           load_identity)

api.add_resource(UserResource, "/users", "/users/<int:id>")

jwt = JWT(user_api_app, authenticate, load_identity)
jwt.jwt_encode_callback = generate_jwt_token
jwt.jwt_payload_callback = generate_jwt_payload
jwt.jwt_headers_callback = generate_jwt_headers
jwt.auth_response_callback = generate_auth_response


@jwt.jwt_error_handler
def error_handler(e):
    return generate_error_response(e)


@user_api_app.route("/api/healthcheck")
def healthcheck():
    try:
        db.engine.execute("SELECT 1;").fetchone()
        return jsonify({"status": "OK"})
    except SQLAlchemyError:
        return jsonify({"status": "DOWN"})