diff --git a/dash_auth/basic_auth.py b/dash_auth/basic_auth.py index e8d56f4..68fe30c 100644 --- a/dash_auth/basic_auth.py +++ b/dash_auth/basic_auth.py @@ -30,17 +30,11 @@ def login_request(self): def auth_wrapper(self, f): def wrap(*args, **kwargs): - if not self.is_authorized(): - return flask.Response(status=403) + if self.is_authorized(): + return f(*args, **kwargs) + return self.login_request() - response = f(*args, **kwargs) - return response return wrap def index_auth_wrapper(self, original_index): - def wrap(*args, **kwargs): - if self.is_authorized(): - return original_index(*args, **kwargs) - else: - return self.login_request() - return wrap + return self.auth_wrapper(original_index) diff --git a/tests/test_basic_auth_integration.py b/tests/test_basic_auth_integration.py index 8f7e791..3dc049c 100644 --- a/tests/test_basic_auth_integration.py +++ b/tests/test_basic_auth_integration.py @@ -33,7 +33,7 @@ def update_output(new_value): def test_failed_views(url): assert requests.get(url).status_code == 401 - assert requests.get(url.strip("/") + "/_dash-layout").status_code == 403 + assert requests.get(url.strip("/") + "/_dash-layout").status_code == 401 test_failed_views(base_url)