apache-airflow-providers-fab 2.4.3__py3-none-any.whl → 2.4.4rc1__py3-none-any.whl

airflow/providers/fab/__init__.py

@@ -29,7 +29,7 @@ from airflow import __version__ as airflow_version
 
 __all__ = ["__version__"]
 
-__version__ = "2.4.3"
+__version__ = "2.4.4"
 
 if packaging.version.parse(packaging.version.parse(airflow_version).base_version) < packaging.version.parse(
     "3.0.2"

airflow/providers/fab/auth_manager/fab_auth_manager.py

@@ -57,7 +57,7 @@ from airflow.cli.cli_config import (
 )
 from airflow.configuration import conf
 from airflow.exceptions import AirflowConfigException, AirflowException
-from airflow.models import DagModel
+from airflow.models import Connection, DagModel, Pool, Variable
 from airflow.providers.fab.auth_manager.cli_commands.definition import (
     DB_COMMANDS,
     PERMISSIONS_CLEANUP_COMMAND,
@@ -437,6 +437,26 @@ class FabAuthManager(BaseAuthManager[User]):
             )
         ]
 
+    @provide_session
+    def get_authorized_connections(
+        self,
+        *,
+        user: User,
+        method: ResourceMethod = "GET",
+        session: Session = NEW_SESSION,
+    ) -> set[str]:
+        """
+        Get connection ids (``conn_id``) the user has access to.
+
+        Fab auth manager does not allow fine-grained access with connections. Thus, return all the connection ids.
+
+        :param user: the user
+        :param method: the method to filter on
+        :param session: the session
+        """
+        rows = session.execute(select(Connection.conn_id)).scalars().all()
+        return set(rows)
+
     @provide_session
     def get_authorized_dag_ids(
         self,
@@ -477,6 +497,46 @@ class FabAuthManager(BaseAuthManager[User]):
                         resources.add(resource[len(permissions.RESOURCE_DAG_PREFIX) :])
         return set(session.scalars(select(DagModel.dag_id).where(DagModel.dag_id.in_(resources))))
 
+    @provide_session
+    def get_authorized_pools(
+        self,
+        *,
+        user: User,
+        method: ResourceMethod = "GET",
+        session: Session = NEW_SESSION,
+    ) -> set[str]:
+        """
+        Get pools the user has access to.
+
+        Fab auth manager does not allow fine-grained access with pools. Thus, return all the pool names.
+
+        :param user: the user
+        :param method: the method to filter on
+        :param session: the session
+        """
+        rows = session.execute(select(Pool.pool)).scalars().all()
+        return set(rows)
+
+    @provide_session
+    def get_authorized_variables(
+        self,
+        *,
+        user: User,
+        method: ResourceMethod = "GET",
+        session: Session = NEW_SESSION,
+    ) -> set[str]:
+        """
+        Get variable keys the user has access to.
+
+        Fab auth manager does not allow fine-grained access with variables. Thus, return all the variable keys.
+
+        :param user: the user
+        :param method: the method to filter on
+        :param session: the session
+        """
+        rows = session.execute(select(Variable.key)).scalars().all()
+        return set(rows)
+
     @cached_property
     def security_manager(self) -> FabAirflowSecurityManagerOverride:
         """Return the security manager specific to FAB."""

airflow/providers/fab/auth_manager/security_manager/override.py

@@ -127,7 +127,7 @@ if AIRFLOW_V_3_1_PLUS:
         with create_session() as session:
             yield from DBDagBag().iter_all_latest_version_dags(session=session)
 else:
-    from airflow.models.dagbag import DagBag
+    from airflow.models.dagbag import DagBag  # type: ignore[attr-defined, no-redef]
 
     def _iter_dags() -> Iterable[DAG | SerializedDAG]:
         dagbag = DagBag(read_dags_from_db=True)  # type: ignore[call-arg]

airflow/providers/fab/www/airflow_flask_app.py

@@ -21,7 +21,12 @@ from typing import TYPE_CHECKING, Any
 from flask import Flask
 
 if TYPE_CHECKING:
-    from airflow.models.dagbag import DagBag
+    from airflow.providers.fab.version_compat import AIRFLOW_V_3_1_PLUS
+
+    if AIRFLOW_V_3_1_PLUS:
+        from airflow.models.dagbag import DBDagBag as DagBag
+    else:
+        from airflow.models.dagbag import DagBag  # type: ignore[no-redef]
 
 
 class AirflowApp(Flask):

airflow/providers/fab/www/package-lock.json

@@ -22,7 +22,7 @@
         "copy-webpack-plugin": "^13.0.1",
         "css-loader": "7.1.2",
         "css-minimizer-webpack-plugin": "^7.0.2",
-        "eslint": "^9.35.0",
+        "eslint": "^9.36.0",
         "eslint-config-prettier": "^10.1.8",
         "eslint-plugin-html": "^8.1.3",
         "eslint-plugin-import": "^2.32.0",
@@ -1807,9 +1807,9 @@
       }
     },
     "node_modules/@eslint/js": {
-      "version": "9.35.0",
-      "resolved": "https://registry.npmjs.org/@eslint/js/-/js-9.35.0.tgz",
-      "integrity": "sha512-30iXE9whjlILfWobBkNerJo+TXYsgVM5ERQwMcMKCHckHflCmf7wXDAHlARoWnh0s1U72WqlbeyE7iAcCzuCPw==",
+      "version": "9.36.0",
+      "resolved": "https://registry.npmjs.org/@eslint/js/-/js-9.36.0.tgz",
+      "integrity": "sha512-uhCbYtYynH30iZErszX78U+nR3pJU3RHGQ57NXy5QupD4SBVwDeU8TNBy+MjMngc1UyIW9noKqsRqfjQTBU2dw==",
       "dev": true,
       "license": "MIT",
       "engines": {
@@ -3998,9 +3998,9 @@
       }
     },
     "node_modules/eslint": {
-      "version": "9.35.0",
-      "resolved": "https://registry.npmjs.org/eslint/-/eslint-9.35.0.tgz",
-      "integrity": "sha512-QePbBFMJFjgmlE+cXAlbHZbHpdFVS2E/6vzCy7aKlebddvl1vadiC4JFV5u/wqTkNUwEV8WrQi257jf5f06hrg==",
+      "version": "9.36.0",
+      "resolved": "https://registry.npmjs.org/eslint/-/eslint-9.36.0.tgz",
+      "integrity": "sha512-hB4FIzXovouYzwzECDcUkJ4OcfOEkXTv2zRY6B9bkwjx/cprAq0uvm1nl7zvQ0/TsUk0zQiN4uPfJpB9m+rPMQ==",
       "dev": true,
       "license": "MIT",
       "dependencies": {
@@ -4010,7 +4010,7 @@
         "@eslint/config-helpers": "^0.3.1",
         "@eslint/core": "^0.15.2",
         "@eslint/eslintrc": "^3.3.1",
-        "@eslint/js": "9.35.0",
+        "@eslint/js": "9.36.0",
         "@eslint/plugin-kit": "^0.3.5",
         "@humanfs/node": "^0.16.6",
         "@humanwhocodes/module-importer": "^1.0.1",

airflow/providers/fab/www/package.json

@@ -48,7 +48,7 @@
     "copy-webpack-plugin": "^13.0.1",
     "css-loader": "7.1.2",
     "css-minimizer-webpack-plugin": "^7.0.2",
-    "eslint": "^9.35.0",
+    "eslint": "^9.36.0",
     "eslint-config-prettier": "^10.1.8",
     "eslint-plugin-html": "^8.1.3",
     "eslint-plugin-import": "^2.32.0",