From 886e91fef8bddb98e43eeb014efa757ef960596c Mon Sep 17 00:00:00 2001
From: "Karl O. Pinc" <kop@karlpinc.com>
Date: Sun, 29 Sep 2024 14:33:41 -0500
Subject: [PATCH] Pass template a "locals" dict so it can modify variables

---
 src/pgwui_sql/templates/sql.mak   | 9 ++++++---
 src/pgwui_sql/views/sql.py        | 1 +
 tests/templates/test_templates.py | 1 +
 3 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/src/pgwui_sql/templates/sql.mak b/src/pgwui_sql/templates/sql.mak
index c3726d5..10584fb 100644
--- a/src/pgwui_sql/templates/sql.mak
+++ b/src/pgwui_sql/templates/sql.mak
@@ -84,11 +84,14 @@
       }<em class="success">!</em>
   </p>
   % if sql:
-       <% if upload_sql:
-              sql = pgwui_sql.lib.normalize_newlines(sql)
+       <%
+       if upload_sql:
+            locals['sql'] = pgwui_sql.lib.normalize_newlines(sql)
+       else:
+            locals['sql'] = sql
        %>
        <ol>
-         % for sql_line in sql.rstrip().split('\n'):
+         % for sql_line in locals['sql'].rstrip().split('\n'):
            <li>${sql_line}</li>
          % endfor
        </ol>
diff --git a/src/pgwui_sql/views/sql.py b/src/pgwui_sql/views/sql.py
index 69eeab1..dd2e229 100644
--- a/src/pgwui_sql/views/sql.py
+++ b/src/pgwui_sql/views/sql.py
@@ -377,6 +377,7 @@ def sql_view(request):
         response['pgwui']['pgwui_sql'] = settings['pgwui']['pgwui_sql']
 
         response['result_rows'] = uh.sql_results
+        response['locals'] = {}   # Template uses this for local vars
 
         log_response(response)
         return response
diff --git a/tests/templates/test_templates.py b/tests/templates/test_templates.py
index dacf952..6bfd916 100644
--- a/tests/templates/test_templates.py
+++ b/tests/templates/test_templates.py
@@ -47,6 +47,7 @@ request = pyramid.testing.DummyRequest()
 request.registry = config.registry
 
 stock_template_args = {
+    'locals': {},
     'pgwui': config.get_settings()['pgwui'],
     'havecreds': False,
     'errors': [],
-- 
2.34.1