apache-airflow-providers-fab 1.5.3__py3-none-any.whl → 2.0.0b1__py3-none-any.whl

airflow/providers/fab/auth_manager/api_fastapi/services/login.py

@@ -0,0 +1,58 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, cast
+
+from starlette import status
+from starlette.exceptions import HTTPException
+
+from airflow.api_fastapi.app import get_auth_manager
+from airflow.configuration import conf
+from airflow.providers.fab.auth_manager.api_fastapi.datamodels.login import LoginBody, LoginResponse
+from airflow.providers.fab.auth_manager.fab_auth_manager import FabAuthManager
+
+if TYPE_CHECKING:
+    from airflow.providers.fab.auth_manager.models import User
+
+
+class FABAuthManagerLogin:
+    """Login Service for FABAuthManager."""
+
+    @classmethod
+    def create_token(
+        cls, body: LoginBody, expiration_time_in_sec: int = conf.getint("api_auth", "jwt_expiration_time")
+    ) -> LoginResponse:
+        """Create a new token."""
+        if not body.username or not body.password:
+            raise HTTPException(
+                status_code=status.HTTP_400_BAD_REQUEST, detail="Username and password must be provided"
+            )
+
+        auth_manager = cast(FabAuthManager, get_auth_manager())
+        user: User = auth_manager.security_manager.find_user(username=body.username)
+        if not user:
+            raise HTTPException(status_code=status.HTTP_401_UNAUTHORIZED, detail="Invalid username")
+
+        if auth_manager.security_manager.check_password(username=body.username, password=body.password):
+            return LoginResponse(
+                jwt_token=auth_manager.generate_jwt(
+                    user=user, expiration_time_in_seconds=expiration_time_in_sec
+                )
+            )
+        else:
+            raise HTTPException(status_code=status.HTTP_401_UNAUTHORIZED, detail="Invalid password")

airflow/providers/fab/auth_manager/cli_commands/db_command.py

@@ -17,7 +17,7 @@
 from __future__ import annotations
 
 from airflow import settings
-from airflow.cli.commands.db_command import run_db_downgrade_command, run_db_migrate_command
+from airflow.cli.commands.local_commands.db_command import run_db_downgrade_command, run_db_migrate_command
 from airflow.providers.fab.auth_manager.models.db import _REVISION_HEADS_MAP, FABDBManager
 from airflow.utils import cli as cli_utils
 from airflow.utils.providers_configuration_loader import providers_configuration_loaded
@@ -38,9 +38,7 @@ def migratedb(args):
     """Migrates the metadata database."""
     session = settings.Session()
     upgrade_command = FABDBManager(session).upgradedb
-    run_db_migrate_command(
-        args, upgrade_command, revision_heads_map=_REVISION_HEADS_MAP, reserialize_dags=False
-    )
+    run_db_migrate_command(args, upgrade_command, revision_heads_map=_REVISION_HEADS_MAP)
 
 
 @cli_utils.action_cli(check_db=False)

airflow/providers/fab/auth_manager/cli_commands/user_command.py

@@ -23,10 +23,10 @@ import getpass
 import json
 import os
 import random
+import re
 import string
 from typing import Any
 
-import re2
 from marshmallow import Schema, fields, validate
 from marshmallow.exceptions import ValidationError
 
@@ -178,7 +178,7 @@ def users_export(args):
         # In the User model the first and last name fields have underscores,
         # but the corresponding parameters in the CLI don't
         def remove_underscores(s):
-            return re2.sub("_", "", s)
+            return re.sub("_", "", s)
 
         users = [
             {

airflow/providers/fab/auth_manager/cli_commands/utils.py

@@ -18,29 +18,30 @@
 from __future__ import annotations
 
 import os
+from collections.abc import Generator
 from contextlib import contextmanager
-from functools import lru_cache
+from functools import cache
 from os.path import isabs
-from typing import TYPE_CHECKING, Generator
+from typing import TYPE_CHECKING
 
 from flask import Flask
+from sqlalchemy.engine import make_url
 
 import airflow
 from airflow.configuration import conf
 from airflow.exceptions import AirflowConfigException
-from airflow.www.app import make_url
-from airflow.www.extensions.init_appbuilder import init_appbuilder
-from airflow.www.extensions.init_session import init_airflow_session_interface
-from airflow.www.extensions.init_views import init_plugins
+from airflow.providers.fab.www.extensions.init_appbuilder import init_appbuilder
+from airflow.providers.fab.www.extensions.init_session import init_airflow_session_interface
+from airflow.providers.fab.www.extensions.init_views import init_plugins
 
 if TYPE_CHECKING:
-    from airflow.www.extensions.init_appbuilder import AirflowAppBuilder
+    from airflow.providers.fab.www.extensions.init_appbuilder import AirflowAppBuilder
 
 
-@lru_cache(maxsize=None)
+@cache
 def _return_appbuilder(app: Flask) -> AirflowAppBuilder:
     """Return an appbuilder instance for the given app."""
-    init_appbuilder(app)
+    init_appbuilder(app, enable_plugins=False)
     init_plugins(app)
     init_airflow_session_interface(app)
     return app.appbuilder  # type: ignore[attr-defined]