From 16c52aeb8c5b74ea013bddb1ecceba7aa258ae9e Mon Sep 17 00:00:00 2001 From: jmanteau Date: Sat, 25 Apr 2020 14:24:51 +0200 Subject: [PATCH] feat: implement pre_run features in reports --- netbox/extras/reports.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/netbox/extras/reports.py b/netbox/extras/reports.py index 373acdde7..7913ff9b9 100644 --- a/netbox/extras/reports.py +++ b/netbox/extras/reports.py @@ -183,6 +183,9 @@ class Report(object): """ self.logger.info(f"Running report") + # Perform any pre-run tasks + self.pre_run() + for method_name in self.test_methods: self.active_test = method_name test_method = getattr(self, method_name) @@ -202,6 +205,12 @@ class Report(object): # Perform any post-run tasks self.post_run() + def pre_run(self): + """ + Extend this method to include any tasks which should execute before the report has been run. + """ + pass + def post_run(self): """ Extend this method to include any tasks which should execute after the report has been run.