Add documentation for using with flask-jwt about flask-principal HOT 6 OPEN

def identity_loader(payload):
    """Return user object referred to in payload."""
    username = payload["identity"]["username"] or None
    current_user = User(username=username)
    identity = Identity(username)
    if hasattr(current_user, "roles"):
        for role in current_user.roles:
             identity.provides.add(RoleNeed(role))
    identity_changed.send(current_app._get_current_object(),
                          identity=identity)
    return current_user


curator_permission = Permission(RoleNeed("Curator"))
curator_permission.description = "User must be a Curator"


@app.route("/projectEdit/<projectID>/<tableName>", methods=["POST"])
@jwt_required()
@curator_permission.require(http_exception=403)
def projectEdit(projectID, tableName):
    """ Update specified table for specified project."""

def identity_loader(payload):
    user_id = payload['identity']

    try:
        current_user = User.query.filter_by(id=user_id).first()
        identity_changed.send(current_app._get_current_object(),
                              identity=Identity(user_id))
    except Exception as e:
        db_session.rollback()
        db_session.flush()
        raise e

    return current_user

@principal.identity_loader
def _identity_loader():
    if hasattr(current_user, 'id'):
        return Identity(current_user.id)

@user_loaded_from_request.connect_via(app)
def on_user_loaded_from_request(sender, user):
    identity_changed.send(
        flask.current_app._get_current_object(),
        identity=Identity(user.id)
    )

@identity_loaded.connect
def on_identity_loaded(sender, identity):

    # Set the identity user object
    identity.user = current_user

    if not hasattr(current_user, 'id'):
        return

    # Add the UserNeed to the identity
    identity.provides.add(UserNeed(current_user.id))

    for role in current_user.roles:
        identity.provides.add(RoleNeed(role.name))