syedahsn commented on code in PR #38287:
URL: https://github.com/apache/airflow/pull/38287#discussion_r1532521802
##########
airflow/providers/amazon/aws/hooks/neptune.py:
##########
@@ -83,3 +89,32 @@ def get_cluster_status(self, cluster_id: str) -> str:
:return: The status of the cluster.
"""
return
self.get_conn().describe_db_clusters(DBClusterIdentifier=cluster_id)["DBClusters"][0]["Status"]
+
+ def get_db_instance_status(self, instance_id: str) -> str:
+ """
+ Get the status of a Neptune instance.
+
+ :param instance_id: The ID of the instance to get the status of.
+ :return: The status of the instance.
+ """
+ return
self.get_conn().describe_db_instances(DBInstanceIdentifier=instance_id)["DBInstances"][0][
+ "DBInstanceStatus"
+ ]
+
+ def wait_for_cluster_instance_availability(
+ self, cluster_id: str, delay: int = 30, max_attempts: int = 60
Review Comment:
Just a small nit here: We've been _trying_ to get standardized names for
these types of parameters, and have mostly settled on `waiter_delay` and
`waiter_max_attempts`.
--
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.
To unsubscribe, e-mail: commits-unsubscr...@airflow.apache.org
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
