Refactored the code since the functions return None. Replaced 'get' with 'set'

Author: Satoshi-Sh

airflow/auth/managers/base_auth_manager.py

@@ -34,7 +34,6 @@
 
 if TYPE_CHECKING:
     import connexion
-    from flask import Blueprint
     from flask_appbuilder.menu import MenuItem
     from sqlalchemy.orm import Session
 
@@ -82,8 +81,8 @@ def get_cli_commands() -> list[CLICommand]:
         """
         return []
 
-    def set_api_endpoints(self, connexion_app: connexion.FlaskApp) -> None | Blueprint:
-        """Return API endpoint(s) definition for the auth manager."""
+    def set_api_endpoints(self, connexion_app: connexion.FlaskApp) -> None:
+        """Set API endpoint(s) definition for the auth manager."""
         return None
 
     def get_user_name(self) -> str:

airflow/providers/fab/auth_manager/fab_auth_manager.py

@@ -23,7 +23,7 @@
 from typing import TYPE_CHECKING, Container
 
 from connexion.options import SwaggerUIOptions
-from flask import Blueprint, url_for
+from flask import url_for
 from sqlalchemy import select
 from sqlalchemy.orm import Session, joinedload
 
@@ -150,7 +150,7 @@ def get_cli_commands() -> list[CLICommand]:
             SYNC_PERM_COMMAND,  # not in a command group
         ]
 
-    def get_api_endpoints(self, connexion_app: connexion.FlaskApp) -> None | Blueprint:
+    def set_api_endpoints(self, connexion_app: connexion.FlaskApp) -> None:
         folder = Path(__file__).parents[0].resolve()  # this is airflow/auth/managers/fab/
         with folder.joinpath("openapi", "v1.yaml").open() as f:
             specification = safe_load(f)
@@ -159,15 +159,15 @@ def get_api_endpoints(self, connexion_app: connexion.FlaskApp) -> None | Bluepri
             swagger_ui=conf.getboolean("webserver", "enable_swagger_ui", fallback=True),
         )
 
-        api = connexion_app.add_api(
+        connexion_app.add_api(
             specification=specification,
             resolver=_LazyResolver(),
             base_path="/auth/fab/v1",
             swagger_ui_options=swagger_ui_options,
             strict_validation=True,
             validate_responses=True,
         )
-        return api.blueprint if api else None
+        return None
 
     def get_user_display_name(self) -> str:
         """Return the user's display name associated to the user in session."""

airflow/www/extensions/init_views.py

@@ -304,8 +304,4 @@ def init_api_experimental(app):
 def init_api_auth_provider(connexion_app: connexion.FlaskApp):
     """Initialize the API offered by the auth manager."""
     auth_mgr = get_auth_manager()
-    blueprint = auth_mgr.get_api_endpoints(connexion_app)
-    if blueprint:
-        base_paths.append(blueprint.url_prefix if blueprint.url_prefix else "")
-        flask_app = connexion_app.app
-        flask_app.extensions["csrf"].exempt(blueprint)
+    auth_mgr.set_api_endpoints(connexion_app)

docs/apache-airflow/core-concepts/auth-manager.rst

@@ -161,7 +161,7 @@ Auth managers may vend CLI commands which will be included in the ``airflow`` co
 Rest API
 ^^^^^^^^
 
-Auth managers may vend Rest API endpoints which will be included in the :doc:`/stable-rest-api-ref` by implementing the ``get_api_endpoints`` method. The endpoints can be used to manage resources such as users, groups, roles (if any) handled by your auth manager. Endpoints are only vended for the currently configured auth manager.
+Auth managers may vend Rest API endpoints which will be included in the :doc:`/stable-rest-api-ref` by implementing the ``set_api_endpoints`` method. The endpoints can be used to manage resources such as users, groups, roles (if any) handled by your auth manager. Endpoints are only vended for the currently configured auth manager.
 
 Next Steps
 ^^^^^^^^^^

tests/auth/managers/test_base_auth_manager.py

@@ -121,8 +121,8 @@ class TestBaseAuthManager:
     def test_get_cli_commands_return_empty_list(self, auth_manager):
         assert auth_manager.get_cli_commands() == []
 
-    def test_get_api_endpoints_return_none(self, auth_manager):
-        assert auth_manager.get_api_endpoints() is None
+    def test_set_api_endpoints_return_none(self, auth_manager):
+        assert auth_manager.set_api_endpoints() is None
 
     def test_get_user_name(self, auth_manager):
         user = Mock()