From fdb141f6225ddf7209dc6abff0c1691bbb675465 Mon Sep 17 00:00:00 2001
From: Simon Willison <swillison@gmail.com>
Date: Mon, 8 May 2023 17:51:29 -0700
Subject: [PATCH] shape_arrayfirst for query view

---
 datasette/views/database.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/datasette/views/database.py b/datasette/views/database.py
index 30016188..eac57cd3 100644
--- a/datasette/views/database.py
+++ b/datasette/views/database.py
@@ -941,11 +941,17 @@ async def query_view(
             return {"ok": False, "error": str(error)}
         return [dict(r) for r in results.rows]
 
+    async def shape_arrayfirst(_results):
+        results, error = _results
+        if error:
+            return {"ok": False, "error": str(error)}
+        return [r[0] for r in results.rows]
+
     shape_fn = {
         "arrays": shape_arrays,
         "objects": shape_objects,
         "array": shape_array,
-        # "arrayfirst": shape_arrayfirst,
+        "arrayfirst": shape_arrayfirst,
         # "object": shape_object,
     }[_shape or "objects"]