from flask import Blueprint from flask import current_app from flask import request from werkzeug.exceptions import abort from capsulflask.db import get_model, my_exec_info_message bp = Blueprint("hosts", __name__, url_prefix="/hosts") def authorized_for_host(id): auth_header_value = request.headers.get('Authorization').replace("Bearer ", "") return get_model().authorized_for_host(id, auth_header_value) @bp.route("/heartbeat/", methods=("POST")) def heartbeat(id): if authorized_for_host(id): get_model().host_heartbeat(id) else: current_app.logger.info(f"/hosts/heartbeat/{id} returned 401: invalid token") return abort(401, "invalid host id or token")