From 497290beaf32e6b779f9683ef15f1c5bc142a41a Mon Sep 17 00:00:00 2001
From: Simon Willison <swillison@gmail.com>
Date: Tue, 1 Nov 2022 12:59:17 -0700
Subject: [PATCH] Handle database errors in /-/insert, refs #1866, #1873

Also improved API explorer to show HTTP status of response, refs #1871
---
 datasette/templates/api_explorer.html | 14 +++++++++-----
 datasette/views/table.py              |  5 ++++-
 tests/test_api_write.py               | 11 +++++++++++
 3 files changed, 24 insertions(+), 6 deletions(-)

diff --git a/datasette/templates/api_explorer.html b/datasette/templates/api_explorer.html
index 38fdb7bc1b..93bacde399 100644
--- a/datasette/templates/api_explorer.html
+++ b/datasette/templates/api_explorer.html
@@ -27,7 +27,8 @@ <h1>API Explorer</h1>
 </form>
 
 <div id="output" style="display: none">
-  <h2>API response</h2>
+  <h2>API response: HTTP <span id="response-status"></span></h2>
+  </h2>
   <ul class="errors message-error"></ul>
   <pre></pre>
 </div>
@@ -64,12 +65,15 @@ <h2>API response</h2>
     headers: {
       'Content-Type': 'application/json',
     }
-  }).then(r => r.json()).then(r => {
+  }).then(r => {
+    document.getElementById('response-status').textContent = r.status;
+    return r.json();
+  }).then(data => {
     var errorList = output.querySelector('.errors');
-    if (r.errors) {
+    if (data.errors) {
       errorList.style.display = 'block';
       errorList.innerHTML = '';
-      r.errors.forEach(error => {
+      data.errors.forEach(error => {
         var li = document.createElement('li');
         li.textContent = error;
         errorList.appendChild(li);
@@ -77,7 +81,7 @@ <h2>API response</h2>
     } else {
       errorList.style.display = 'none';
     }
-    output.querySelector('pre').innerText = JSON.stringify(r, null, 2);
+    output.querySelector('pre').innerText = JSON.stringify(data, null, 2);
     output.style.display = 'block';
   }).catch(err => {
     alert("Error: " + err);
diff --git a/datasette/views/table.py b/datasette/views/table.py
index 7692a4e3bb..6122720695 100644
--- a/datasette/views/table.py
+++ b/datasette/views/table.py
@@ -1206,7 +1206,10 @@ def insert_rows(conn):
             else:
                 table.insert_all(rows, ignore=ignore, replace=replace)
 
-        rows = await db.execute_write_fn(insert_rows)
+        try:
+            rows = await db.execute_write_fn(insert_rows)
+        except Exception as e:
+            return _error([str(e)])
         result = {"ok": True}
         if should_return:
             result["rows"] = rows
diff --git a/tests/test_api_write.py b/tests/test_api_write.py
index d0b0f324ce..0b567f4813 100644
--- a/tests/test_api_write.py
+++ b/tests/test_api_write.py
@@ -156,6 +156,13 @@ async def test_write_rows(ds_write, return_rows):
             400,
             ["Too many rows, maximum allowed is 100"],
         ),
+        (
+            "/data/docs/-/insert",
+            {"rows": [{"id": 1, "title": "Test"}]},
+            "duplicate_id",
+            400,
+            ["UNIQUE constraint failed: docs.id"],
+        ),
         (
             "/data/docs/-/insert",
             {"rows": [{"title": "Test"}], "ignore": True, "replace": True},
@@ -194,6 +201,10 @@ async def test_write_row_errors(
     ds_write, path, input, special_case, expected_status, expected_errors
 ):
     token = write_token(ds_write)
+    if special_case == "duplicate_id":
+        await ds_write.get_database("data").execute_write(
+            "insert into docs (id) values (1)"
+        )
     if special_case == "bad_token":
         token += "bad"
     kwargs = dict(