ryanyuan commented on a change in pull request #10340:
URL: https://github.com/apache/airflow/pull/10340#discussion_r471038378
##########
File path: airflow/providers/google/cloud/hooks/bigtable.py
##########
@@ -184,6 +185,51 @@ def create_instance(
operation.result(timeout)
return instance
+ @GoogleBaseHook.fallback_to_default_project_id
+ def update_instance(
+ self,
+ instance_id: str,
+ project_id: str,
+ instance_display_name: Optional[str] = None,
+ instance_type: Optional[Union[enums.Instance.Type, enum.IntEnum]] =
None,
+ instance_labels: Optional[Dict] = None,
+ timeout: Optional[float] = None
+ ) -> Instance:
+ """
+ Update an existing instance.
+
+ :type instance_id: str
+ :param instance_id: The ID for the existing instance.
+ :type project_id: str
+ :param project_id: Optional, Google Cloud Platform project ID where the
+ BigTable exists. If set to None or missing,
+ the default project_id from the GCP connection is used.
+ :type instance_display_name: str
+ :param instance_display_name: (optional) Human-readable name of the
instance.
+ :type instance_type: enums.Instance.Type or enum.IntEnum
+ :param instance_type: (optional) The type of the instance.
+ :type instance_labels: dict
+ :param instance_labels: (optional) Dictionary of labels to associate
with the
+ instance.
+ :type timeout: int
+ :param timeout: (optional) timeout (in seconds) for instance update.
+ If None is not specified, Operator will wait indefinitely.
+ """
+ instance_type = enums.Instance.Type(instance_type)
+
+ instance = Instance(
+ instance_id=instance_id,
+ client=self._get_client(project_id=project_id),
+ display_name=instance_display_name,
+ instance_type=instance_type,
+ labels=instance_labels,
+ )
+
+ operation = instance.update()
+ operation.result(timeout)
Review comment:
Same. Until I saw the timeout usage here in documentation:
https://googleapis.dev/python/bigtable/latest/instance.html#google.cloud.bigtable.instance.Instance.create
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
