This is an automated email from the ASF dual-hosted git repository.
potiuk pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new 8c05e59922 fix bq_to_mysql init checks (#37653)
8c05e59922 is described below
commit 8c05e59922946ee06bc014d578676581b33dba2c
Author: Albert Olweny <[email protected]>
AuthorDate: Fri Feb 23 21:14:11 2024 +0300
fix bq_to_mysql init checks (#37653)
---
.pre-commit-config.yaml | 1 -
airflow/providers/google/cloud/transfers/bigquery_to_mysql.py | 6 +++++-
2 files changed, 5 insertions(+), 2 deletions(-)
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index 8fcfd4c431..b0e1a27700 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -327,7 +327,6 @@ repos:
^.*__init__\.py$|
^airflow\/providers\/amazon\/aws\/transfers\/gcs_to_s3\.py$|
^airflow\/providers\/databricks\/operators\/databricks\.py$|
-
^airflow\/providers\/google\/cloud\/transfers\/bigquery_to_mysql\.py$|
^airflow\/providers\/amazon\/aws\/transfers\/redshift_to_s3\.py$|
^airflow\/providers\/google\/cloud\/operators\/compute\.py$|
^airflow\/providers\/amazon\/aws\/operators\/emr\.py$|
diff --git a/airflow/providers/google/cloud/transfers/bigquery_to_mysql.py
b/airflow/providers/google/cloud/transfers/bigquery_to_mysql.py
index 166305f766..66084a6fb3 100644
--- a/airflow/providers/google/cloud/transfers/bigquery_to_mysql.py
+++ b/airflow/providers/google/cloud/transfers/bigquery_to_mysql.py
@@ -48,6 +48,8 @@ class BigQueryToMySqlOperator(BigQueryToSqlBaseOperator):
mysql_table: str | None = None,
target_table_name: str | None = None,
mysql_conn_id: str = "mysql_default",
+ dataset_id: str | None = None,
+ table_id: str | None = None,
**kwargs,
) -> None:
if mysql_table is not None:
@@ -65,7 +67,9 @@ class BigQueryToMySqlOperator(BigQueryToSqlBaseOperator):
target_table_name = mysql_table
- super().__init__(target_table_name=target_table_name, **kwargs)
+ super().__init__(
+ target_table_name=target_table_name, dataset_id=dataset_id,
table_id=table_id, **kwargs
+ )
self.mysql_conn_id = mysql_conn_id
def get_sql_hook(self) -> MySqlHook:
