Esempio n. 1
0
 def execute(self, context: dict) -> dict:
     self.log.info("Fetching data for job with id: %d ...", self.job_id)
     hook = GoogleDataprepHook(
         dataprep_conn_id="dataprep_default",
     )
     response = hook.get_jobs_for_job_group(job_id=self.job_id)
     return response
Esempio n. 2
0
 def execute(self, context: dict) -> dict:
     self.log.info("Fetching data for job with id: %d ...", self.job_group_id)
     hook = GoogleDataprepHook(dataprep_conn_id=self.dataprep_conn_id)
     response = hook.get_job_group(
         job_group_id=self.job_group_id,
         embed=self.embed,
         include_deleted=self.include_deleted,
     )
     return response
Esempio n. 3
0
 def execute(self, context: None) -> dict:
     self.log.info("Creating a job...")
     hook = GoogleDataprepHook(dataprep_conn_id=self.dataprep_conn_id)
     response = hook.run_job_group(body_request=self.body_request)
     return response