From 0b2c6a7ebd4fd540d9bdfb169c621452d280e608 Mon Sep 17 00:00:00 2001
From: Simon Willison <swillison@gmail.com>
Date: Mon, 8 Jan 2024 13:12:57 -0800
Subject: [PATCH] Fix for ?_extra=columns bug, closes #2230

Also refs #262 - started a test suite for extras.
---
 datasette/renderer.py   |  2 +-
 tests/test_table_api.py | 24 ++++++++++++++++++++++++
 2 files changed, 25 insertions(+), 1 deletion(-)

diff --git a/datasette/renderer.py b/datasette/renderer.py
index 224031a7..a446e69d 100644
--- a/datasette/renderer.py
+++ b/datasette/renderer.py
@@ -68,7 +68,7 @@ def json_renderer(request, args, data, error, truncated=None):
     elif shape in ("objects", "object", "array"):
         columns = data.get("columns")
         rows = data.get("rows")
-        if rows and columns:
+        if rows and columns and not isinstance(rows[0], dict):
             data["rows"] = [dict(zip(columns, row)) for row in rows]
         if shape == "object":
             shape_error = None
diff --git a/tests/test_table_api.py b/tests/test_table_api.py
index 5dbb8b8f..ae4fdb17 100644
--- a/tests/test_table_api.py
+++ b/tests/test_table_api.py
@@ -1362,3 +1362,27 @@ async def test_col_nocol_errors(ds_client, path, expected_error):
     response = await ds_client.get(path)
     assert response.status_code == 400
     assert response.json()["error"] == expected_error
+
+
+@pytest.mark.asyncio
+@pytest.mark.parametrize(
+    "extra,expected_json",
+    (
+        (
+            "columns",
+            {
+                "ok": True,
+                "next": None,
+                "columns": ["id", "content", "content2"],
+                "rows": [{"id": "1", "content": "hey", "content2": "world"}],
+                "truncated": False,
+            },
+        ),
+    ),
+)
+async def test_table_extras(ds_client, extra, expected_json):
+    response = await ds_client.get(
+        "/fixtures/primary_key_multiple_columns.json?_extra=" + extra
+    )
+    assert response.status_code == 200
+    assert response.json() == expected_json