__init__.py 1.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748
  1. #
  2. # Licensed to the Apache Software Foundation (ASF) under one
  3. # or more contributor license agreements. See the NOTICE file
  4. # distributed with this work for additional information
  5. # regarding copyright ownership. The ASF licenses this file
  6. # to you under the Apache License, Version 2.0 (the
  7. # "License"); you may not use this file except in compliance
  8. # with the License. You may obtain a copy of the License at
  9. #
  10. # http://www.apache.org/licenses/LICENSE-2.0
  11. #
  12. # Unless required by applicable law or agreed to in writing,
  13. # software distributed under the License is distributed on an
  14. # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  15. # KIND, either express or implied. See the License for the
  16. # specific language governing permissions and limitations
  17. # under the License.
  18. """Authentication backend."""
  19. from __future__ import annotations
  20. import logging
  21. from importlib import import_module
  22. from airflow.configuration import conf
  23. from airflow.exceptions import AirflowConfigException, AirflowException
  24. log = logging.getLogger(__name__)
  25. def load_auth():
  26. """Load authentication backends."""
  27. auth_backends = "airflow.api.auth.backend.default"
  28. try:
  29. auth_backends = conf.get("api", "auth_backends")
  30. except AirflowConfigException:
  31. pass
  32. backends = []
  33. try:
  34. for backend in auth_backends.split(","):
  35. auth = import_module(backend.strip())
  36. log.info("Loaded API auth backend: %s", backend)
  37. backends.append(auth)
  38. except ImportError as err:
  39. log.critical("Cannot import %s for API authentication due to: %s", backend, err)
  40. raise AirflowException(err)
  41. return backends