def execute(self, context):
hook = BiqQueryDataTransferServiceHook(gcp_conn_id=self.gcp_conn_id)
self.log.info("Creating DTS transfer config")
response = hook.create_transfer_config(
project_id=self.project_id,
transfer_config=self.transfer_config,
authorization_code=self.authorization_code,
retry=self.retry,
timeout=self.timeout,
metadata=self.metadata,
)
result = MessageToDict(response)
self.log.info("Created DTS transfer config %s", get_object_id(result))
self.xcom_push(context,
key="transfer_config_id",
value=get_object_id(result))
return result
def execute(self, context):
hook = BiqQueryDataTransferServiceHook(gcp_conn_id=self.gcp_conn_id)
self.log.info('Submitting manual transfer for %s',
self.transfer_config_id)
response = hook.start_manual_transfer_runs(
transfer_config_id=self.transfer_config_id,
requested_time_range=self.requested_time_range,
requested_run_time=self.requested_run_time,
project_id=self.project_id,
retry=self.retry,
timeout=self.timeout,
metadata=self.metadata,
)
result = MessageToDict(response)
run_id = None
if 'runs' in result:
run_id = get_object_id(result['runs'][0])
self.xcom_push(context, key="run_id", value=run_id)
self.log.info('Transfer run %s submitted successfully.', run_id)
return result