Skip to content

Commit 9449a10

Browse files
authored
DatastoreHook: Remove datastore_conn_id (#23323)
* `DatastoreHook`: Remove `datastore_conn_id`
1 parent 1f32603 commit 9449a10

File tree

4 files changed

+5
-13
lines changed

4 files changed

+5
-13
lines changed

airflow/providers/google/CHANGELOG.rst

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,8 @@ Breaking changes
3434
For more information, see `Deprecation and sunset <https://developers.google.com/google-ads/api/docs/sunset-dates>`_
3535
and `Upgrading to the newest version <https://developers.google.com/google-ads/api/docs/version-migration>`_
3636

37+
* ``DatastoreHook``: Remove ``datastore_conn_id``. Please use ``gcp_conn_id``
38+
3739
* ``CloudBuildCreateBuildOperator``: Remove ``body``. Please use ``build``
3840

3941
* ``BigtableCreateInstanceOperator`` Remove ``replica_cluster_id``, ``replica_cluster_zone``. Please use ``replica_clusters``.

airflow/providers/google/cloud/hooks/datastore.py

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,6 @@
2020

2121

2222
import time
23-
import warnings
2423
from typing import Any, Dict, Optional, Sequence, Union
2524

2625
from googleapiclient.discovery import Resource, build
@@ -43,17 +42,8 @@ def __init__(
4342
gcp_conn_id: str = "google_cloud_default",
4443
delegate_to: Optional[str] = None,
4544
api_version: str = 'v1',
46-
datastore_conn_id: Optional[str] = None,
4745
impersonation_chain: Optional[Union[str, Sequence[str]]] = None,
4846
) -> None:
49-
if datastore_conn_id:
50-
warnings.warn(
51-
"The datastore_conn_id parameter has been deprecated. You should pass "
52-
"the gcp_conn_id parameter.",
53-
DeprecationWarning,
54-
stacklevel=2,
55-
)
56-
gcp_conn_id = datastore_conn_id
5747
super().__init__(
5848
gcp_conn_id=gcp_conn_id,
5949
delegate_to=delegate_to,

airflow/providers/google/cloud/operators/datastore.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -119,8 +119,8 @@ def execute(self, context: 'Context') -> dict:
119119
gcs_hook.delete(self.bucket, obj)
120120

121121
ds_hook = DatastoreHook(
122-
self.datastore_conn_id,
123-
self.delegate_to,
122+
gcp_conn_id=self.datastore_conn_id,
123+
delegate_to=self.delegate_to,
124124
impersonation_chain=self.impersonation_chain,
125125
)
126126
result = ds_hook.export_to_storage_bucket(

tests/providers/google/cloud/operators/test_datastore.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ def test_execute(self, mock_hook):
5959
)
6060
op.execute(context={'ti': mock.MagicMock()})
6161

62-
mock_hook.assert_called_once_with(CONN_ID, None, impersonation_chain=None)
62+
mock_hook.assert_called_once_with(gcp_conn_id=CONN_ID, delegate_to=None, impersonation_chain=None)
6363
mock_hook.return_value.export_to_storage_bucket.assert_called_once_with(
6464
project_id=PROJECT_ID,
6565
bucket=BUCKET,

0 commit comments

Comments
 (0)