kerberos_command.py 1.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243
  1. # Licensed to the Apache Software Foundation (ASF) under one
  2. # or more contributor license agreements. See the NOTICE file
  3. # distributed with this work for additional information
  4. # regarding copyright ownership. The ASF licenses this file
  5. # to you under the Apache License, Version 2.0 (the
  6. # "License"); you may not use this file except in compliance
  7. # with the License. You may obtain a copy of the License at
  8. #
  9. # http://www.apache.org/licenses/LICENSE-2.0
  10. #
  11. # Unless required by applicable law or agreed to in writing,
  12. # software distributed under the License is distributed on an
  13. # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  14. # KIND, either express or implied. See the License for the
  15. # specific language governing permissions and limitations
  16. # under the License.
  17. """Kerberos command."""
  18. from __future__ import annotations
  19. from airflow import settings
  20. from airflow.cli.commands.daemon_utils import run_command_with_daemon_option
  21. from airflow.security import kerberos as krb
  22. from airflow.security.kerberos import KerberosMode
  23. from airflow.utils import cli as cli_utils
  24. from airflow.utils.providers_configuration_loader import providers_configuration_loaded
  25. @cli_utils.action_cli
  26. @providers_configuration_loaded
  27. def kerberos(args):
  28. """Start a kerberos ticket renewer."""
  29. print(settings.HEADER)
  30. mode = KerberosMode.STANDARD
  31. if args.one_time:
  32. mode = KerberosMode.ONE_TIME
  33. run_command_with_daemon_option(
  34. args=args,
  35. process_name="kerberos",
  36. callback=lambda: krb.run(principal=args.principal, keytab=args.keytab, mode=mode),
  37. )