move CustomBasicAuth override to import

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

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)

View File

@@ -3,15 +3,13 @@ from flask import jsonify, request
class CustomBasicAuth(BasicAuth):
"""
Description:
Usage:
Description:
Override BasicAuth to permit anonymous healthcheck at /_status?simple=true
"""
def check_credentials(self, username, password):
# here, for example, you can search user in the database by passed `username` and `password`, etc.
return username == 'user' and password == 'password'
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()