Fixed a test and implemented show_hide_hidden

simonw · Aug 7, 2023 · ea24c9c · ea24c9c
1 parent a791115
commit ea24c9c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/datasette/views/database.py b/datasette/views/database.py
@@ -209,6 +209,9 @@ class QueryContext:
     columns: list = field(metadata={"help": "List of column names"})
     renderers: dict = field(metadata={"help": "Dictionary of renderer name to URL"})
     url_csv: str = field(metadata={"help": "URL for CSV export"})
+    show_hide_hidden: str = field(
+        metadata={"help": "Hidden input field for the _show_sql parameter"}
+    )
     metadata: dict = field(metadata={"help": "Metadata about the query/database"})
     database_color: Callable = field(
         metadata={"help": "Function that returns a color for a given database name"}
@@ -516,6 +519,7 @@ async def fetch_data_for_csv(request, _next=None):
                             request=request, format="csv"
                         )  # , extra_qs=url_csv_args)
                     ),
+                    show_hide_hidden=markupsafe.Markup(show_hide_hidden),
                     metadata=metadata,
                     database_color=lambda _: "#ff0000",
                 ),

diff --git a/tests/test_api.py b/tests/test_api.py
@@ -650,6 +650,7 @@ async def test_custom_sql(ds_client):
             {"content": "RENDER_CELL_ASYNC"},
         ],
         "columns": ["content"],
+        "ok": True,
         "truncated": False,
     }