def poke(self, context: Dict):
context.update(self.op_kwargs)
context['templates_dict'] = self.templates_dict
self.op_kwargs = PythonOperator.determine_op_kwargs(self.python_callable, context, len(self.op_args))
self.log.info("Poking callable: %s", str(self.python_callable))
return_value = self.python_callable(*self.op_args, **self.op_kwargs)
return bool(return_value)
def poke(self, context: Dict):
self.log.info('Poking: %s', self.endpoint)
try:
response = self.hook.run(self.endpoint,
data=self.request_params,
headers=self.headers,
extra_options=self.extra_options)
if self.response_check:
op_kwargs = PythonOperator.determine_op_kwargs(self.response_check, context)
return self.response_check(response, **op_kwargs)
except AirflowException as ae:
if str(ae).startswith("404"):
return False
raise ae
return True
