forked from 3wordchant/capsul-flask
23 lines
695 B
Python
23 lines
695 B
Python
|
|
||
|
|
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/<string:id>", 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")
|
||
|
|
|