diff --git a/import2ledger/__main__.py b/import2ledger/__main__.py index 068460118add55a3249c7ad1b695a7c9504c93ad..6d3cd0e826145b5167960feaf273172b57179357 100644 --- a/import2ledger/__main__.py +++ b/import2ledger/__main__.py @@ -49,13 +49,15 @@ class FileImporter: default_date = self.config.get_default_date() in_file.seek(0) for entry_data in importer(in_file): - entry_data['_hook_cancel'] = False for hook in self.hooks: - hook.run(entry_data) - if entry_data['_hook_cancel']: + hook_retval = hook.run(entry_data) + if hook_retval is None: + pass + elif hook_retval is False: break + else: + entry_data = hook_retval else: - del entry_data['_hook_cancel'] render_vars = collections.ChainMap(entry_data, source_vars) print(template.render(render_vars), file=out_file, end='')