def handle_callback(self, *args, **kwargs):
self.log.debug(f"handle_callback({args}, {kwargs})")
try:
dagrun = args[0]
self.log.debug(f"handle_callback() dagrun : {dagrun}")
_MARQUEZ.create_namespace()
self._report_task_instances(dagrun,
DagUtils.get_run_args(**kwargs),
kwargs.get('session'))
except Exception as e:
self.log.error(
f'Failed to record dagrun callback: {e} '
f'dag_id={self.dag_id}',
exc_info=True)
return super().handle_callback(*args)
def create_dagrun(self, *args, **kwargs):
# run Airflow's create_dagrun() first
dagrun = super(DAG, self).create_dagrun(*args, **kwargs)
create_dag_start_ms = self._now_ms()
try:
_MARQUEZ.create_namespace()
self._register_dagrun(dagrun,
DagUtils.get_execution_date(**kwargs),
DagUtils.get_run_args(**kwargs))
except Exception as e:
self.log.error(
f'Failed to record metadata: {e} '
f'{self._timed_log_message(create_dag_start_ms)}',
exc_info=True)
return dagrun