Add formats.

aminalaee · Oct 9, 2024 · 0da8dc0 · 0da8dc0
1 parent 99cbe63
commit 0da8dc0
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/sqladmin/models.py b/sqladmin/models.py
@@ -1183,23 +1183,24 @@ async def generate(writer: Writer) -> AsyncGenerator[Any, None]:
         )
 
     async def _export_json(
-            self,
-            data: List[Any],
+        self,
+        data: List[Any],
     ) -> StreamingResponse:
         async def generate() -> AsyncGenerator[str, None]:
-            yield '['
+            yield "["
             first_row = True
             for row in data:
                 if not first_row:
-                    yield ','
+                    yield ","
                 else:
                     first_row = False
                 row_dict = {
                     name: await self.get_prop_value(row, name)
                     for name in self._export_prop_names
                 }
                 yield json.dumps(row_dict)
-            yield ']'
+            yield "]"
+
         filename = secure_filename(self.get_export_name(export_type="json"))
         return StreamingResponse(
             content=generate(),

diff --git a/tests/test_views/test_view_sync.py b/tests/test_views/test_view_sync.py
@@ -740,6 +740,7 @@ def test_export_json(client: TestClient) -> None:
     response = client.get("/admin/user/export/json")
     assert response.text == '[{"name": "Daniel", "status": "ACTIVE"}]'
 
+
 def test_export_csv_row_count(client: TestClient) -> None:
     def row_count(resp) -> int:
         return resp.text.count("\r\n") - 1