def __init__(self,
instance_id,
main_cluster_id,
main_cluster_zone,
project_id=None,
replica_cluster_id=None,
replica_cluster_zone=None,
instance_display_name=None,
instance_type=None,
instance_labels=None,
cluster_nodes=None,
cluster_storage_type=None,
timeout=None,
*args,
**kwargs):
self.project_id = project_id
self.instance_id = instance_id
self.main_cluster_id = main_cluster_id
self.main_cluster_zone = main_cluster_zone
self.replica_cluster_id = replica_cluster_id
self.replica_cluster_zone = replica_cluster_zone
self.instance_display_name = instance_display_name
self.instance_type = instance_type
self.instance_labels = instance_labels
self.cluster_nodes = cluster_nodes
self.cluster_storage_type = cluster_storage_type
self.timeout = timeout
self._validate_inputs()
self.hook = BigtableHook()
super(BigtableInstanceCreateOperator, self).__init__(*args, **kwargs)
def __init__(self, # pylint: disable=too-many-arguments
instance_id,
main_cluster_id,
main_cluster_zone,
project_id=None,
replica_cluster_id=None,
replica_cluster_zone=None,
instance_display_name=None,
instance_type=None,
instance_labels=None,
cluster_nodes=None,
cluster_storage_type=None,
timeout=None,
*args, **kwargs):
self.project_id = project_id
self.instance_id = instance_id
self.main_cluster_id = main_cluster_id
self.main_cluster_zone = main_cluster_zone
self.replica_cluster_id = replica_cluster_id
self.replica_cluster_zone = replica_cluster_zone
self.instance_display_name = instance_display_name
self.instance_type = instance_type
self.instance_labels = instance_labels
self.cluster_nodes = cluster_nodes
self.cluster_storage_type = cluster_storage_type
self.timeout = timeout
self._validate_inputs()
self.hook = BigtableHook()
super().__init__(*args, **kwargs)
def __init__(self, project_id, instance_id, table_id, *args, **kwargs):
self.project_id = project_id
self.instance_id = instance_id
self.table_id = table_id
self._validate_inputs()
self.hook = BigtableHook()
super(BigtableTableWaitForReplicationSensor,
self).__init__(*args, **kwargs)
def __init__(self, project_id, instance_id, cluster_id, nodes, *args,
**kwargs):
self.project_id = project_id
self.instance_id = instance_id
self.cluster_id = cluster_id
self.nodes = nodes
self._validate_inputs()
self.hook = BigtableHook()
super(BigtableClusterUpdateOperator, self).__init__(*args, **kwargs)
def __init__(self,
instance_id,
project_id=None,
*args, **kwargs):
self.project_id = project_id
self.instance_id = instance_id
self._validate_inputs()
self.hook = BigtableHook()
super().__init__(*args, **kwargs)
def __init__(self,
instance_id,
project_id=None,
gcp_conn_id='google_cloud_default',
*args,
**kwargs):
self.project_id = project_id
self.instance_id = instance_id
self._validate_inputs()
self.hook = BigtableHook(gcp_conn_id=gcp_conn_id)
super().__init__(*args, **kwargs)
def __init__(self,
instance_id,
table_id,
project_id=None,
app_profile_id=None,
*args, **kwargs):
self.project_id = project_id
self.instance_id = instance_id
self.table_id = table_id
self.app_profile_id = app_profile_id
self._validate_inputs()
self.hook = BigtableHook()
super(BigtableTableDeleteOperator, self).__init__(*args, **kwargs)
def __init__(self,
instance_id,
table_id,
project_id=None,
gcp_conn_id='google_cloud_default',
*args,
**kwargs):
self.project_id = project_id
self.instance_id = instance_id
self.table_id = table_id
self._validate_inputs()
self.hook = BigtableHook(gcp_conn_id=gcp_conn_id)
super(BigtableTableWaitForReplicationSensor,
self).__init__(*args, **kwargs)
def __init__(self,
instance_id,
cluster_id,
nodes,
project_id=None,
gcp_conn_id='google_cloud_default',
*args,
**kwargs):
self.project_id = project_id
self.instance_id = instance_id
self.cluster_id = cluster_id
self.nodes = nodes
self._validate_inputs()
self.hook = BigtableHook(gcp_conn_id=gcp_conn_id)
super(BigtableClusterUpdateOperator, self).__init__(*args, **kwargs)
def __init__(self,
instance_id,
table_id,
project_id=None,
app_profile_id=None,
gcp_conn_id='google_cloud_default',
*args,
**kwargs):
self.project_id = project_id
self.instance_id = instance_id
self.table_id = table_id
self.app_profile_id = app_profile_id
self._validate_inputs()
self.hook = BigtableHook(gcp_conn_id=gcp_conn_id)
super(BigtableTableDeleteOperator, self).__init__(*args, **kwargs)
def __init__(self,
instance_id,
table_id,
project_id=None,
initial_split_keys=None,
column_families=None,
*args, **kwargs):
self.project_id = project_id
self.instance_id = instance_id
self.table_id = table_id
self.initial_split_keys = initial_split_keys or list()
self.column_families = column_families or dict()
self._validate_inputs()
self.hook = BigtableHook()
self.instance = None
super().__init__(*args, **kwargs)
def __init__(self,
instance_id,
table_id,
project_id=None,
initial_split_keys=None,
column_families=None,
gcp_conn_id='google_cloud_default',
*args,
**kwargs):
self.project_id = project_id
self.instance_id = instance_id
self.table_id = table_id
self.initial_split_keys = initial_split_keys or list()
self.column_families = column_families or dict()
self._validate_inputs()
self.hook = BigtableHook(gcp_conn_id=gcp_conn_id)
self.instance = None
super(BigtableTableCreateOperator, self).__init__(*args, **kwargs)
def setUp(self):
with mock.patch(
'airflow.contrib.hooks.gcp_api_base_hook.GoogleCloudBaseHook.__init__',
new=mock_base_gcp_hook_no_default_project_id):
self.bigtable_hook_no_default_project_id = BigtableHook(
gcp_conn_id='test')
def __init__(self, project_id, instance_id, *args, **kwargs):
self.project_id = project_id
self.instance_id = instance_id
self._validate_inputs()
self.hook = BigtableHook()
super(BigtableInstanceDeleteOperator, self).__init__(*args, **kwargs)
