I think kedro hooks are the right call here https://kedro.readthedocs.io/en/latest/07_extend_kedro/02_hooks.html