Source code for airflow.providers.amazon.aws.sensors.redshift
## Licensed to the Apache Software Foundation (ASF) under one# or more contributor license agreements. See the NOTICE file# distributed with this work for additional information# regarding copyright ownership. The ASF licenses this file# to you under the Apache License, Version 2.0 (the# "License"); you may not use this file except in compliance# with the License. You may obtain a copy of the License at## http://www.apache.org/licenses/LICENSE-2.0## Unless required by applicable law or agreed to in writing,# software distributed under the License is distributed on an# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY# KIND, either express or implied. See the License for the# specific language governing permissions and limitations# under the License.fromtypingimportOptionalfromairflow.providers.amazon.aws.hooks.redshiftimportRedshiftHookfromairflow.sensors.baseimportBaseSensorOperator
[docs]classAwsRedshiftClusterSensor(BaseSensorOperator):""" Waits for a Redshift cluster to reach a specific status. :param cluster_identifier: The identifier for the cluster being pinged. :type cluster_identifier: str :param target_status: The cluster status desired. :type target_status: str """
[docs]defpoke(self,context):self.log.info('Poking for status : %s\nfor cluster %s',self.target_status,self.cluster_identifier)returnself.get_hook().cluster_status(self.cluster_identifier)==self.target_status
[docs]defget_hook(self)->RedshiftHook:"""Create and return a RedshiftHook"""ifself.hook:returnself.hookself.hook=RedshiftHook(aws_conn_id=self.aws_conn_id)returnself.hook