diff --git a/airflow/providers/amazon/aws/log/s3_task_handler.py b/airflow/providers/amazon/aws/log/s3_task_handler.py index 761c4ce463814..f3664f7c415bd 100644 --- a/airflow/providers/amazon/aws/log/s3_task_handler.py +++ b/airflow/providers/amazon/aws/log/s3_task_handler.py @@ -78,7 +78,9 @@ def hook(self): ) def set_context(self, ti: TaskInstance, *, identifier: str | None = None) -> None: - if getattr(self, "supports_task_context_logging", False): + # todo: remove-at-min-airflow-version-2.8 + # after Airflow 2.8 can always pass `identifier` + if getattr(super(), "supports_task_context_logging", False): super().set_context(ti, identifier=identifier) else: super().set_context(ti) diff --git a/airflow/providers/elasticsearch/log/es_task_handler.py b/airflow/providers/elasticsearch/log/es_task_handler.py index 1e8c75b7e3eb7..c9d3a180e1edc 100644 --- a/airflow/providers/elasticsearch/log/es_task_handler.py +++ b/airflow/providers/elasticsearch/log/es_task_handler.py @@ -443,7 +443,9 @@ def set_context(self, ti: TaskInstance, *, identifier: str | None = None) -> Non self.handler.setLevel(self.level) self.handler.setFormatter(self.formatter) else: - if getattr(self, "supports_task_context_logging", False): + # todo: remove-at-min-airflow-version-2.8 + # after Airflow 2.8 can always pass `identifier` + if getattr(super(), "supports_task_context_logging", False): super().set_context(ti, identifier=identifier) else: super().set_context(ti) diff --git a/airflow/providers/google/cloud/log/gcs_task_handler.py b/airflow/providers/google/cloud/log/gcs_task_handler.py index 39d0f072a8acd..9921bb8753f03 100644 --- a/airflow/providers/google/cloud/log/gcs_task_handler.py +++ b/airflow/providers/google/cloud/log/gcs_task_handler.py @@ -142,7 +142,9 @@ def client(self) -> storage.Client: ) def set_context(self, ti: TaskInstance, *, identifier: str | None = None) -> None: - if getattr(self, "supports_task_context_logging", False): + # todo: remove-at-min-airflow-version-2.8 + # after Airflow 2.8 can always pass `identifier` + if getattr(super(), "supports_task_context_logging", False): super().set_context(ti, identifier=identifier) else: super().set_context(ti) diff --git a/airflow/providers/microsoft/azure/log/wasb_task_handler.py b/airflow/providers/microsoft/azure/log/wasb_task_handler.py index f3a00e8432b9d..c57de1acb1d06 100644 --- a/airflow/providers/microsoft/azure/log/wasb_task_handler.py +++ b/airflow/providers/microsoft/azure/log/wasb_task_handler.py @@ -96,7 +96,9 @@ def hook(self): return None def set_context(self, ti: TaskInstance, *, identifier: str | None = None) -> None: - if getattr(self, "supports_task_context_logging", False): + # todo: remove-at-min-airflow-version-2.8 + # after Airflow 2.8 can always pass `identifier` + if getattr(super(), "supports_task_context_logging", False): super().set_context(ti, identifier=identifier) else: super().set_context(ti)