from flask_basicauth import BasicAuth
from flask import jsonify, request

class CustomBasicAuth(BasicAuth):
    """
        Description: 
        Override BasicAuth to permit anonymous healthcheck at /_status?simple=true
    """
    def challenge(self):
        if "/_status" in request.url:
            if request.args.get('elb', None) or request.args.get('simple', None):
                return jsonify(status="ok"), 200
        return super(CustomBasicAuth, self).challenge()

custom_basic_auth = CustomBasicAuth()