From c3c3582de8e2da41cf721e26d234627c3ca87dbf Mon Sep 17 00:00:00 2001 From: zachliu Date: Mon, 14 Mar 2022 12:57:37 -0400 Subject: [PATCH] use different logger to avoid duplicate log entry https://github.com/apache/airflow/pull/22137#issuecomment-1067060405 --- airflow/providers/amazon/aws/hooks/base_aws.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/airflow/providers/amazon/aws/hooks/base_aws.py b/airflow/providers/amazon/aws/hooks/base_aws.py index 16c7385f9c1ef..bbf0bfff83e91 100644 --- a/airflow/providers/amazon/aws/hooks/base_aws.py +++ b/airflow/providers/amazon/aws/hooks/base_aws.py @@ -584,13 +584,14 @@ def decorator_f(self, *args, **kwargs): min_limit = retry_args.get('min', 1) max_limit = retry_args.get('max', 1) stop_after_delay = retry_args.get('stop_after_delay', 10) - tenacity_logger = tenacity.before_log(self.log, logging.INFO) if self.log else None + tenacity_before_logger = tenacity.before_log(self.log, logging.INFO) if self.log else None + tenacity_after_logger = tenacity.after_log(self.log, logging.INFO) if self.log else None default_kwargs = { 'wait': tenacity.wait_exponential(multiplier=multiplier, max=max_limit, min=min_limit), 'retry': tenacity.retry_if_exception(should_retry), 'stop': tenacity.stop_after_delay(stop_after_delay), - 'before': tenacity_logger, - 'after': tenacity_logger, + 'before': tenacity_before_logger, + 'after': tenacity_after_logger, } return tenacity.retry(**default_kwargs)(fun)(self, *args, **kwargs)