darkhelm/notifications-admin
1
0
Fork 0
mirror of https://github.com/GSA/notifications-admin.git synced 2026-08-19 05:59:44 -04:00
Code Issues Packages Projects Releases Wiki Activity

move CustomBasicAuth override to import

Browse Source
This commit is contained in:
Jim Moffet
2022-07-01 11:34:51 -07:00
parent f8f87e0596
commit e9a3b3d5e3
2 changed files with 18 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
app/__init__.py
View File

@@ -39,6 +39,7 @@ from werkzeug.local import LocalProxy
from app import proxy_fix, webauthn_server
from app.asset_fingerprinter import asset_fingerprinter
from app.config import configs
from app.custom_auth import CustomBasicAuth
from app.extensions import antivirus_client, redis_client, zendesk_client
from app.formatters import (
convert_to_boolean,
@@ -593,16 +594,16 @@ def init_jinja(application):
jinja_loader = jinja2.FileSystemLoader(template_folders)
application.jinja_loader = jinja_loader
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()
# 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()
def setup_basic_auth(application):
application.basic_auth = CustomBasicAuth(application)