mirror of
https://github.com/astral-sh/ruff.git
synced 2025-10-01 06:11:43 +00:00
c87e3ccb2f
<!-- Thank you for contributing to Ruff! To help us out with reviewing, please consider the following: - Does this pull request include a summary of the change? (See below.) - Does this pull request include a descriptive title? - Does this pull request include references to any relevant issues? --> ## Summary <!-- What's the purpose of the change? What does it do, and why? --> attribute check was missing in the previous implementation e.g. ```python from airflow.api.auth.backend import basic_auth basic_auth.auth_current_user ``` This PR adds this kind of check. ## Test Plan <!-- How was it tested? --> The test case has been added to the button of the existing test fixtures, confirmed to be correct and later reorgnaized
481 lines
14 KiB
Python
481 lines
14 KiB
Python
from __future__ import annotations
|
|
|
|
from airflow.api.auth.backend import basic_auth, kerberos_auth
|
|
from airflow.api.auth.backend.basic_auth import auth_current_user
|
|
from airflow.auth.managers.fab.api.auth.backend import (
|
|
kerberos_auth as backend_kerberos_auth,
|
|
)
|
|
from airflow.auth.managers.fab.fab_auth_manager import FabAuthManager
|
|
from airflow.auth.managers.fab.security_manager import override as fab_override
|
|
from airflow.config_templates.default_celery import DEFAULT_CELERY_CONFIG
|
|
from airflow.executors.celery_executor import CeleryExecutor, app
|
|
from airflow.executors.celery_kubernetes_executor import CeleryKubernetesExecutor
|
|
from airflow.executors.dask_executor import DaskExecutor
|
|
from airflow.executors.kubernetes_executor_types import (
|
|
ALL_NAMESPACES,
|
|
POD_EXECUTOR_DONE_KEY,
|
|
)
|
|
from airflow.hooks.dbapi import ConnectorProtocol, DbApiHook
|
|
from airflow.hooks.dbapi_hook import DbApiHook as DbApiHook2
|
|
from airflow.hooks.docker_hook import DockerHook
|
|
from airflow.hooks.druid_hook import DruidDbApiHook, DruidHook
|
|
from airflow.hooks.filesystem import FSHook
|
|
from airflow.hooks.hive_hooks import (
|
|
HIVE_QUEUE_PRIORITIES,
|
|
HiveCliHook,
|
|
HiveMetastoreHook,
|
|
HiveServer2Hook,
|
|
)
|
|
from airflow.hooks.http_hook import HttpHook
|
|
from airflow.hooks.jdbc_hook import JdbcHook, jaydebeapi
|
|
from airflow.hooks.mssql_hook import MsSqlHook
|
|
from airflow.hooks.mysql_hook import MySqlHook
|
|
from airflow.hooks.oracle_hook import OracleHook
|
|
from airflow.hooks.package_index import PackageIndexHook
|
|
from airflow.hooks.pig_hook import PigCliHook
|
|
from airflow.hooks.postgres_hook import PostgresHook
|
|
from airflow.hooks.presto_hook import PrestoHook
|
|
from airflow.hooks.S3_hook import S3Hook, provide_bucket_name
|
|
from airflow.hooks.samba_hook import SambaHook
|
|
from airflow.hooks.slack_hook import SlackHook
|
|
from airflow.hooks.sqlite_hook import SqliteHook
|
|
from airflow.hooks.subprocess import SubprocessHook, SubprocessResult, working_directory
|
|
from airflow.hooks.webhdfs_hook import WebHDFSHook
|
|
from airflow.hooks.zendesk_hook import ZendeskHook
|
|
from airflow.kubernetes.k8s_model import K8SModel, append_to_pod
|
|
from airflow.kubernetes.kube_client import (
|
|
_disable_verify_ssl,
|
|
_enable_tcp_keepalive,
|
|
get_kube_client,
|
|
)
|
|
from airflow.kubernetes.kubernetes_helper_functions import (
|
|
add_pod_suffix,
|
|
annotations_for_logging_task_metadata,
|
|
annotations_to_key,
|
|
create_pod_id,
|
|
get_logs_task_metadata,
|
|
rand_str,
|
|
)
|
|
from airflow.kubernetes.pod import Port, Resources
|
|
from airflow.kubernetes.pod_generator import (
|
|
PodDefaults,
|
|
PodGenerator,
|
|
PodGeneratorDeprecated,
|
|
datetime_to_label_safe_datestring,
|
|
extend_object_field,
|
|
label_safe_datestring_to_datetime,
|
|
make_safe_label_value,
|
|
merge_objects,
|
|
)
|
|
from airflow.kubernetes.pod_generator import (
|
|
add_pod_suffix as add_pod_suffix2,
|
|
)
|
|
from airflow.kubernetes.pod_generator import (
|
|
rand_str as rand_str2,
|
|
)
|
|
from airflow.kubernetes.pod_generator_deprecated import (
|
|
PodDefaults as PodDefaults3,
|
|
)
|
|
from airflow.kubernetes.pod_generator_deprecated import (
|
|
PodGenerator as PodGenerator2,
|
|
)
|
|
from airflow.kubernetes.pod_generator_deprecated import (
|
|
make_safe_label_value as make_safe_label_value2,
|
|
)
|
|
from airflow.kubernetes.pod_launcher import PodLauncher, PodStatus
|
|
from airflow.kubernetes.pod_launcher_deprecated import (
|
|
PodDefaults as PodDefaults2,
|
|
)
|
|
from airflow.kubernetes.pod_launcher_deprecated import (
|
|
PodLauncher as PodLauncher2,
|
|
)
|
|
from airflow.kubernetes.pod_launcher_deprecated import (
|
|
PodStatus as PodStatus2,
|
|
)
|
|
from airflow.kubernetes.pod_launcher_deprecated import (
|
|
get_kube_client as get_kube_client2,
|
|
)
|
|
from airflow.kubernetes.pod_runtime_info_env import PodRuntimeInfoEnv
|
|
from airflow.kubernetes.secret import K8SModel2, Secret
|
|
from airflow.kubernetes.volume import Volume
|
|
from airflow.kubernetes.volume_mount import VolumeMount
|
|
from airflow.macros.hive import closest_ds_partition, max_partition
|
|
from airflow.operators.bash import BashOperator
|
|
from airflow.operators.bash_operator import BashOperator as LegacyBashOperator
|
|
from airflow.operators.check_operator import (
|
|
CheckOperator,
|
|
IntervalCheckOperator,
|
|
SQLCheckOperator,
|
|
SQLIntervalCheckOperator,
|
|
SQLThresholdCheckOperator,
|
|
SQLValueCheckOperator,
|
|
ThresholdCheckOperator,
|
|
ValueCheckOperator,
|
|
)
|
|
from airflow.operators.datetime import BranchDateTimeOperator, target_times_as_dates
|
|
from airflow.operators.docker_operator import DockerOperator
|
|
from airflow.operators.druid_check_operator import DruidCheckOperator
|
|
from airflow.operators.dummy import DummyOperator, EmptyOperator
|
|
from airflow.operators.email import EmailOperator
|
|
from airflow.operators.email_operator import EmailOperator
|
|
from airflow.operators.gcs_to_s3 import GCSToS3Operator
|
|
from airflow.operators.google_api_to_s3_transfer import (
|
|
GoogleApiToS3Operator,
|
|
GoogleApiToS3Transfer,
|
|
)
|
|
from airflow.operators.hive_operator import HiveOperator
|
|
from airflow.operators.hive_stats_operator import HiveStatsCollectionOperator
|
|
from airflow.operators.hive_to_druid import HiveToDruidOperator, HiveToDruidTransfer
|
|
from airflow.operators.hive_to_mysql import HiveToMySqlOperator, HiveToMySqlTransfer
|
|
from airflow.operators.hive_to_samba_operator import HiveToSambaOperator
|
|
from airflow.operators.http_operator import SimpleHttpOperator
|
|
from airflow.operators.jdbc_operator import JdbcOperator
|
|
from airflow.operators.mssql_operator import MsSqlOperator
|
|
from airflow.operators.mssql_to_hive import MsSqlToHiveOperator, MsSqlToHiveTransfer
|
|
from airflow.operators.mysql_operator import MySqlOperator
|
|
from airflow.operators.mysql_to_hive import MySqlToHiveOperator, MySqlToHiveTransfer
|
|
from airflow.operators.oracle_operator import OracleOperator
|
|
from airflow.operators.papermill_operator import PapermillOperator
|
|
from airflow.operators.pig_operator import PigOperator
|
|
from airflow.operators.postgres_operator import Mapping, PostgresOperator
|
|
from airflow.operators.presto_check_operator import (
|
|
PrestoCheckOperator,
|
|
PrestoIntervalCheckOperator,
|
|
PrestoValueCheckOperator,
|
|
)
|
|
from airflow.operators.presto_check_operator import (
|
|
SQLCheckOperator as SQLCheckOperator2,
|
|
)
|
|
from airflow.operators.presto_check_operator import (
|
|
SQLIntervalCheckOperator as SQLIntervalCheckOperator2,
|
|
)
|
|
from airflow.operators.presto_check_operator import (
|
|
SQLValueCheckOperator as SQLValueCheckOperator2,
|
|
)
|
|
from airflow.operators.presto_to_mysql import (
|
|
PrestoToMySqlOperator,
|
|
PrestoToMySqlTransfer,
|
|
)
|
|
from airflow.operators.python import (
|
|
BranchPythonOperator,
|
|
PythonOperator,
|
|
PythonVirtualenvOperator,
|
|
ShortCircuitOperator,
|
|
)
|
|
from airflow.operators.redshift_to_s3_operator import (
|
|
RedshiftToS3Operator,
|
|
RedshiftToS3Transfer,
|
|
)
|
|
from airflow.operators.s3_file_transform_operator import S3FileTransformOperator
|
|
from airflow.operators.s3_to_hive_operator import S3ToHiveOperator, S3ToHiveTransfer
|
|
from airflow.operators.s3_to_redshift_operator import (
|
|
S3ToRedshiftOperator,
|
|
S3ToRedshiftTransfer,
|
|
)
|
|
from airflow.operators.slack_operator import SlackAPIOperator, SlackAPIPostOperator
|
|
from airflow.operators.sql import (
|
|
BaseSQLOperator,
|
|
BranchSQLOperator,
|
|
SQLTableCheckOperator,
|
|
_convert_to_float_if_possible,
|
|
parse_boolean,
|
|
)
|
|
from airflow.operators.sql import (
|
|
SQLCheckOperator as SQLCheckOperator3,
|
|
)
|
|
from airflow.operators.sql import (
|
|
SQLColumnCheckOperator as SQLColumnCheckOperator2,
|
|
)
|
|
from airflow.operators.sql import (
|
|
SQLIntervalCheckOperator as SQLIntervalCheckOperator3,
|
|
)
|
|
from airflow.operators.sql import (
|
|
SQLThresholdCheckOperator as SQLThresholdCheckOperator2,
|
|
)
|
|
from airflow.operators.sql import (
|
|
SQLValueCheckOperator as SQLValueCheckOperator3,
|
|
)
|
|
from airflow.operators.sqlite_operator import SqliteOperator
|
|
from airflow.operators.trigger_dagrun import TriggerDagRunOperator
|
|
from airflow.operators.weekday import BranchDayOfWeekOperator
|
|
from airflow.sensors import filesystem
|
|
from airflow.sensors.date_time import DateTimeSensor, DateTimeSensorAsync
|
|
from airflow.sensors.date_time_sensor import DateTimeSensor
|
|
from airflow.sensors.external_task import (
|
|
ExternalTaskMarker,
|
|
ExternalTaskSensor,
|
|
ExternalTaskSensorLink,
|
|
)
|
|
from airflow.sensors.filesystem import FileSensor
|
|
from airflow.sensors.hive_partition_sensor import HivePartitionSensor
|
|
from airflow.sensors.http_sensor import HttpSensor
|
|
from airflow.sensors.metastore_partition_sensor import MetastorePartitionSensor
|
|
from airflow.sensors.named_hive_partition_sensor import NamedHivePartitionSensor
|
|
from airflow.sensors.sql import SqlSensor
|
|
from airflow.sensors.sql_sensor import SqlSensor2
|
|
from airflow.sensors.time_delta import TimeDeltaSensor, TimeDeltaSensorAsync, WaitSensor
|
|
from airflow.sensors.time_sensor import TimeSensor, TimeSensorAsync
|
|
from airflow.sensors.web_hdfs_sensor import WebHdfsSensor
|
|
from airflow.sensors.weekday import DayOfWeekSensor
|
|
from airflow.triggers.external_task import DagStateTrigger, WorkflowTrigger
|
|
from airflow.triggers.file import FileTrigger
|
|
from airflow.triggers.temporal import DateTimeTrigger, TimeDeltaTrigger
|
|
from airflow.www.security import FabAirflowSecurityManagerOverride
|
|
|
|
# apache-airflow-providers-amazon
|
|
provide_bucket_name()
|
|
GCSToS3Operator()
|
|
GoogleApiToS3Operator()
|
|
GoogleApiToS3Transfer()
|
|
RedshiftToS3Operator()
|
|
RedshiftToS3Transfer()
|
|
S3FileTransformOperator()
|
|
S3Hook()
|
|
SSQLTableCheckOperator3KeySensor()
|
|
S3ToRedshiftOperator()
|
|
S3ToRedshiftTransfer()
|
|
|
|
# apache-airflow-providers-celery
|
|
DEFAULT_CELERY_CONFIG
|
|
app
|
|
CeleryExecutor()
|
|
CeleryKubernetesExecutor()
|
|
|
|
# apache-airflow-providers-common-sql
|
|
_convert_to_float_if_possible()
|
|
parse_boolean()
|
|
BaseSQLOperator()
|
|
BashOperator()
|
|
LegacyBashOperator()
|
|
BranchSQLOperator()
|
|
CheckOperator()
|
|
ConnectorProtocol()
|
|
DbApiHook()
|
|
DbApiHook2()
|
|
IntervalCheckOperator()
|
|
PrestoCheckOperator()
|
|
PrestoIntervalCheckOperator()
|
|
PrestoValueCheckOperator()
|
|
SQLCheckOperator()
|
|
SQLCheckOperator2()
|
|
SQLCheckOperator3()
|
|
SQLColumnCheckOperator2()
|
|
SQLIntervalCheckOperator()
|
|
SQLIntervalCheckOperator2()
|
|
SQLIntervalCheckOperator3()
|
|
SQLTableCheckOperator()
|
|
SQLThresholdCheckOperator()
|
|
SQLThresholdCheckOperator2()
|
|
SQLValueCheckOperator()
|
|
SQLValueCheckOperator2()
|
|
SQLValueCheckOperator3()
|
|
SqlSensor()
|
|
SqlSensor2()
|
|
ThresholdCheckOperator()
|
|
ValueCheckOperator()
|
|
|
|
# apache-airflow-providers-daskexecutor
|
|
DaskExecutor()
|
|
|
|
# apache-airflow-providers-docker
|
|
DockerHook()
|
|
DockerOperator()
|
|
|
|
# apache-airflow-providers-apache-druid
|
|
DruidDbApiHook()
|
|
DruidHook()
|
|
DruidCheckOperator()
|
|
|
|
# apache-airflow-providers-apache-hdfs
|
|
WebHDFSHook()
|
|
WebHdfsSensor()
|
|
|
|
# apache-airflow-providers-apache-hive
|
|
HIVE_QUEUE_PRIORITIES
|
|
closest_ds_partition()
|
|
max_partition()
|
|
HiveCliHook()
|
|
HiveMetastoreHook()
|
|
HiveOperator()
|
|
HivePartitionSensor()
|
|
HiveServer2Hook()
|
|
HiveStatsCollectionOperator()
|
|
HiveToDruidOperator()
|
|
HiveToDruidTransfer()
|
|
HiveToSambaOperator()
|
|
S3ToHiveOperator()
|
|
S3ToHiveTransfer()
|
|
MetastorePartitionSensor()
|
|
NamedHivePartitionSensor()
|
|
|
|
# apache-airflow-providers-http
|
|
HttpHook()
|
|
HttpSensor()
|
|
SimpleHttpOperator()
|
|
|
|
# apache-airflow-providers-jdbc
|
|
jaydebeapi
|
|
JdbcHook()
|
|
JdbcOperator()
|
|
|
|
# apache-airflow-providers-fab
|
|
basic_auth.CLIENT_AUTH
|
|
basic_auth.init_app
|
|
basic_auth.auth_current_user
|
|
basic_auth.requires_authentication
|
|
|
|
kerberos_auth.log
|
|
kerberos_auth.CLIENT_AUTH
|
|
kerberos_auth.find_user
|
|
kerberos_auth.init_app
|
|
kerberos_auth.requires_authentication
|
|
auth_current_user
|
|
backend_kerberos_auth
|
|
fab_override
|
|
FabAuthManager()
|
|
FabAirflowSecurityManagerOverride()
|
|
|
|
# check whether attribute access
|
|
basic_auth.auth_current_user
|
|
|
|
# apache-airflow-providers-cncf-kubernetes
|
|
ALL_NAMESPACES
|
|
POD_EXECUTOR_DONE_KEY
|
|
_disable_verify_ssl()
|
|
_enable_tcp_keepalive()
|
|
append_to_pod()
|
|
annotations_for_logging_task_metadata()
|
|
annotations_to_key()
|
|
create_pod_id()
|
|
datetime_to_label_safe_datestring()
|
|
extend_object_field()
|
|
get_logs_task_metadata()
|
|
label_safe_datestring_to_datetime()
|
|
merge_objects()
|
|
Port()
|
|
Resources()
|
|
PodRuntimeInfoEnv()
|
|
PodGeneratorDeprecated()
|
|
Volume()
|
|
VolumeMount()
|
|
Secret()
|
|
|
|
add_pod_suffix()
|
|
add_pod_suffix2()
|
|
get_kube_client()
|
|
get_kube_client2()
|
|
make_safe_label_value()
|
|
make_safe_label_value2()
|
|
rand_str()
|
|
rand_str2()
|
|
K8SModel()
|
|
K8SModel2()
|
|
PodLauncher()
|
|
PodLauncher2()
|
|
PodStatus()
|
|
PodStatus2()
|
|
PodDefaults()
|
|
PodDefaults2()
|
|
PodDefaults3()
|
|
PodGenerator()
|
|
PodGenerator2()
|
|
|
|
|
|
# apache-airflow-providers-microsoft-mssql
|
|
MsSqlHook()
|
|
MsSqlOperator()
|
|
MsSqlToHiveOperator()
|
|
MsSqlToHiveTransfer()
|
|
|
|
# apache-airflow-providers-mysql
|
|
HiveToMySqlOperator()
|
|
HiveToMySqlTransfer()
|
|
MySqlHook()
|
|
MySqlOperator()
|
|
MySqlToHiveOperator()
|
|
MySqlToHiveTransfer()
|
|
PrestoToMySqlOperator()
|
|
PrestoToMySqlTransfer()
|
|
|
|
# apache-airflow-providers-oracle
|
|
OracleHook()
|
|
OracleOperator()
|
|
|
|
# apache-airflow-providers-papermill
|
|
PapermillOperator()
|
|
|
|
# apache-airflow-providers-apache-pig
|
|
PigCliHook()
|
|
PigOperator()
|
|
|
|
# apache-airflow-providers-postgres
|
|
Mapping
|
|
PostgresHook()
|
|
PostgresOperator()
|
|
|
|
# apache-airflow-providers-presto
|
|
PrestoHook()
|
|
|
|
# apache-airflow-providers-samba
|
|
SambaHook()
|
|
|
|
# apache-airflow-providers-slack
|
|
SlackHook()
|
|
SlackAPIOperator()
|
|
SlackAPIPostOperator()
|
|
|
|
# apache-airflow-providers-sqlite
|
|
SqliteHook()
|
|
SqliteOperator()
|
|
|
|
# apache-airflow-providers-zendesk
|
|
ZendeskHook()
|
|
|
|
# apache-airflow-providers-smtp
|
|
EmailOperator()
|
|
|
|
# apache-airflow-providers-standard
|
|
filesystem.FileSensor()
|
|
FileSensor()
|
|
TriggerDagRunOperator()
|
|
ExternalTaskMarker()
|
|
ExternalTaskSensor()
|
|
BranchDateTimeOperator()
|
|
BranchDayOfWeekOperator()
|
|
BranchPythonOperator()
|
|
DateTimeSensor()
|
|
DateTimeSensorAsync()
|
|
TimeSensor()
|
|
TimeDeltaSensor()
|
|
DayOfWeekSensor()
|
|
DummyOperator()
|
|
EmptyOperator()
|
|
ExternalTaskMarker()
|
|
ExternalTaskSensor()
|
|
ExternalTaskSensorLink()
|
|
FileSensor()
|
|
FileTrigger()
|
|
FSHook()
|
|
PackageIndexHook()
|
|
SubprocessHook()
|
|
ShortCircuitOperator()
|
|
TimeDeltaSensor()
|
|
TimeSensor()
|
|
TriggerDagRunOperator()
|
|
WorkflowTrigger()
|
|
PythonOperator()
|
|
PythonVirtualenvOperator()
|
|
DagStateTrigger()
|
|
FileTrigger()
|
|
DateTimeTrigger()
|
|
TimeDeltaTrigger()
|
|
SubprocessResult()
|
|
SubprocessHook()
|
|
TimeDeltaSensor()
|
|
TimeDeltaSensorAsync()
|
|
WaitSensor()
|
|
TimeSensor()
|
|
TimeSensorAsync()
|
|
BranchDateTimeOperator()
|
|
working_directory()
|
|
target_times_as_dates()
|