Ejemplo n.º 1
0
 def execute(self, context: 'Context'):
     hook = CloudTasksHook(
         gcp_conn_id=self.gcp_conn_id,
         impersonation_chain=self.impersonation_chain,
     )
     task = hook.get_task(
         location=self.location,
         queue_name=self.queue_name,
         task_name=self.task_name,
         project_id=self.project_id,
         response_view=self.response_view,
         retry=self.retry,
         timeout=self.timeout,
         metadata=self.metadata,
     )
     return Task.to_dict(task)
Ejemplo n.º 2
0
 def execute(self, context: 'Context'):
     hook = CloudTasksHook(
         gcp_conn_id=self.gcp_conn_id,
         impersonation_chain=self.impersonation_chain,
     )
     tasks = hook.list_tasks(
         location=self.location,
         queue_name=self.queue_name,
         project_id=self.project_id,
         response_view=self.response_view,
         page_size=self.page_size,
         retry=self.retry,
         timeout=self.timeout,
         metadata=self.metadata,
     )
     return [Task.to_dict(t) for t in tasks]
Ejemplo n.º 3
0
 def execute(self, context: 'Context'):
     hook = CloudTasksHook(
         gcp_conn_id=self.gcp_conn_id,
         impersonation_chain=self.impersonation_chain,
     )
     task = hook.run_task(
         location=self.location,
         queue_name=self.queue_name,
         task_name=self.task_name,
         project_id=self.project_id,
         response_view=self.response_view,
         retry=self.retry,
         timeout=self.timeout,
         metadata=self.metadata,
     )
     CloudTasksQueueLink.persist(
         operator_instance=self,
         context=context,
         queue_name=task.name,
     )
     return Task.to_dict(task)
Ejemplo n.º 4
0
 def execute(self, context: 'Context'):
     hook = CloudTasksHook(
         gcp_conn_id=self.gcp_conn_id,
         impersonation_chain=self.impersonation_chain,
     )
     tasks = hook.list_tasks(
         location=self.location,
         queue_name=self.queue_name,
         project_id=self.project_id,
         response_view=self.response_view,
         page_size=self.page_size,
         retry=self.retry,
         timeout=self.timeout,
         metadata=self.metadata,
     )
     CloudTasksQueueLink.persist(
         operator_instance=self,
         context=context,
         queue_name=f"projects/{self.project_id or hook.project_id}/"
         f"locations/{self.location}/queues/{self.queue_name}",
     )
     return [Task.to_dict(t) for t in tasks]