From e2750c7cc0585adaa8c866be611089e62961ee35 Mon Sep 17 00:00:00 2001
From: Simon Willison <swillison@gmail.com>
Date: Thu, 19 Apr 2018 08:36:14 -0700
Subject: [PATCH] Add col-X classes to HTML table on custom query page

---
 datasette/app.py               | 1 +
 datasette/templates/query.html | 6 +++---
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/datasette/app.py b/datasette/app.py
index 79ce012e..b6a3498a 100644
--- a/datasette/app.py
+++ b/datasette/app.py
@@ -79,6 +79,7 @@ class RenderMixin(HTTPMethodView):
                 **context, **{
                     'app_css_hash': self.ds.app_css_hash(),
                     'select_templates': select_templates,
+                    'zip': zip,
                 }
             })
         )
diff --git a/datasette/templates/query.html b/datasette/templates/query.html
index 3887fadd..3daf5c7f 100644
--- a/datasette/templates/query.html
+++ b/datasette/templates/query.html
@@ -44,14 +44,14 @@
 <table>
     <thead>
         <tr>
-            {% for column in columns %}<th scope="col">{{ column }}</th>{% endfor %}
+            {% for column in columns %}<th class="col-{{ column|to_css_class }}" scope="col">{{ column }}</th>{% endfor %}
         </tr>
     </thead>
     <tbody>
     {% for row in rows %}
         <tr>
-            {% for td in row %}
-                <td>{% if td == None %}{{ "&nbsp;"|safe }}{% else %}{{ td }}{% endif %}</td>
+            {% for column, td in zip(columns, row) %}
+                <td class="col-{{ column|to_css_class }}">{% if td == None %}{{ "&nbsp;"|safe }}{% else %}{{ td }}{% endif %}</td>
             {% endfor %}
         </tr>
     {% endfor %}