From b3bdee60bb104f11dc828af4141d2f0359e9a4e7 Mon Sep 17 00:00:00 2001
From: Lance Edgar <lance@edbob.org>
Date: Mon, 28 Nov 2022 17:55:08 -0600
Subject: [PATCH] Add way to quickly re-run "any" report

---
 .../templates/reports/generated/view.mako     | 22 +++++++++++++++++++
 tailbone/views/reports.py                     | 22 +++++++++++++++++--
 2 files changed, 42 insertions(+), 2 deletions(-)

diff --git a/tailbone/templates/reports/generated/view.mako b/tailbone/templates/reports/generated/view.mako
index 496857c5..6260efba 100644
--- a/tailbone/templates/reports/generated/view.mako
+++ b/tailbone/templates/reports/generated/view.mako
@@ -1,6 +1,28 @@
 ## -*- coding: utf-8; -*-
 <%inherit file="/master/view.mako" />
 
+<%def name="object_helpers()">
+  % if master.has_perm('create'):
+      <nav class="panel">
+        <p class="panel-heading">Tools</p>
+        <div class="panel-block buttons">
+          <div style="display: flex; flex-direction: column;">
+          <once-button type="is-primary"
+                       % if rerun_report_url:
+                       tag="a" href="${rerun_report_url}"
+                       % else:
+                       disabled title="Unknown report type"
+                       % endif
+                       text="Re-run This Report"
+                       icon-pack="fas"
+                       icon-left="arrow-circle-right">
+          </once-button>
+          </div>
+        </div>
+      </nav>
+  % endif
+</%def>
+
 <%def name="modify_this_page_vars()">
   ${parent.modify_this_page_vars()}
   <script type="text/javascript">
diff --git a/tailbone/views/reports.py b/tailbone/views/reports.py
index 7cb5ecfe..640dc6a9 100644
--- a/tailbone/views/reports.py
+++ b/tailbone/views/reports.py
@@ -316,10 +316,18 @@ class ReportOutputView(ExportMasterView):
 
     def template_kwargs_view(self, **kwargs):
         kwargs = super(ReportOutputView, self).template_kwargs_view(**kwargs)
+        output = kwargs['instance']
 
         if self.get_use_buefy():
-            report = kwargs['instance']
-            kwargs['params_data'] = self.get_params_context(report)
+            kwargs['params_data'] = self.get_params_context(output)
+
+        # build custom URL to re-build this report
+        url = None
+        if output.report_type:
+            url = self.request.route_url('generate_specific_report',
+                                         type_key=output.report_type,
+                                         _query=output.params)
+        kwargs['rerun_report_url'] = url
 
         return kwargs
 
@@ -386,6 +394,7 @@ class ReportOutputView(ExportMasterView):
         input parameters specific to the report type, then creates a new report
         and redirects user to view the output.
         """
+        app = self.get_rattail_app()
         use_buefy = self.get_use_buefy()
         type_key = self.request.matchdict['type_key']
         report = self.report_handler.get_report(type_key)
@@ -445,6 +454,15 @@ class ReportOutputView(ExportMasterView):
                 if len(values) == 1:
                     form.set_default(param.name, values[0][0])
 
+        # set default field values according to query string, if applicable
+        if self.request.GET:
+            for param in report_params:
+                if param.name in self.request.GET:
+                    value = self.request.GET[param.name]
+                    if param.type is datetime.date:
+                        value = app.parse_date(value)
+                    form.set_default(param.name, value)
+
         # if form validates, start generating new report output; show progress page
         if form.validate(newstyle=True):
             key = 'report_output.generate'