From ac9882307cedb1a20957a1117783483cbe99b940 Mon Sep 17 00:00:00 2001 From: romsharon98 Date: Fri, 19 Jan 2024 16:41:19 +0200 Subject: [PATCH 1/2] add templated fields to constructor --- airflow/providers/apache/hive/operators/hive_stats.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/airflow/providers/apache/hive/operators/hive_stats.py b/airflow/providers/apache/hive/operators/hive_stats.py index d816541ffa62f..86e8a599b9caf 100644 --- a/airflow/providers/apache/hive/operators/hive_stats.py +++ b/airflow/providers/apache/hive/operators/hive_stats.py @@ -76,6 +76,8 @@ def __init__( metastore_conn_id: str = "metastore_default", presto_conn_id: str = "presto_default", mysql_conn_id: str = "airflow_db", + ds: str = "{{ ds }}", + dttm: str = "{{ execution_date.isoformat() }}", **kwargs: Any, ) -> None: if "col_blacklist" in kwargs: @@ -96,8 +98,8 @@ def __init__( self.presto_conn_id = presto_conn_id self.mysql_conn_id = mysql_conn_id self.assignment_func = assignment_func - self.ds = "{{ ds }}" - self.dttm = "{{ execution_date.isoformat() }}" + self.ds = ds + self.dttm = dttm def get_default_exprs(self, col: str, col_type: str) -> dict[Any, Any]: """Get default expressions.""" From 5a8c62fe8f14a951344c38a18341c6993b553e2e Mon Sep 17 00:00:00 2001 From: rom sharon <33751805+romsharon98@users.noreply.github.com> Date: Fri, 19 Jan 2024 17:37:22 +0200 Subject: [PATCH 2/2] Fix from execution_date to logical_date Co-authored-by: Andrey Anshin --- airflow/providers/apache/hive/operators/hive_stats.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/airflow/providers/apache/hive/operators/hive_stats.py b/airflow/providers/apache/hive/operators/hive_stats.py index 86e8a599b9caf..327e8a676fc5b 100644 --- a/airflow/providers/apache/hive/operators/hive_stats.py +++ b/airflow/providers/apache/hive/operators/hive_stats.py @@ -77,7 +77,7 @@ def __init__( presto_conn_id: str = "presto_default", mysql_conn_id: str = "airflow_db", ds: str = "{{ ds }}", - dttm: str = "{{ execution_date.isoformat() }}", + dttm: str = "{{ logical_date.isoformat() }}", **kwargs: Any, ) -> None: if "col_blacklist" in kwargs: