Added indent to json.dumps.

vivekkeshore · Aug 11, 2022 · b3b55e0 · b3b55e0
1 parent 0e34ba1
commit b3b55e0
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 7 deletions.
diff --git a/flask_dantic/VERSION b/flask_dantic/VERSION
@@ -1 +1 @@
-0.0.5
+0.0.6
diff --git a/flask_dantic/pydantic_serializer/serializer.py b/flask_dantic/pydantic_serializer/serializer.py
@@ -52,6 +52,7 @@ def serialize(
         exclude_none: bool = False,
         many=False,
         json_dump: bool = True,
+        json_indent: int = 2,
 ):
     """Create python dict from an object using a Pydantic Type"""
     if many:
@@ -63,7 +64,7 @@ def serialize(
     if not errors:
         encoded_obj = jsonable_encoder(response_dict, include=include, exclude=exclude, exclude_none=exclude_none)
         if json_dump:
-            return json.dumps(encoded_obj)
+            return json.dumps(encoded_obj, indent=json_indent)
         return encoded_obj
 
     else:

diff --git a/tests/test_serializer.py b/tests/test_serializer.py
@@ -94,19 +94,19 @@ def test_serialize_json_dumped_objects(db):
 
     # Returns dumped object.
     res = serialize(single_user, UserModel, json_dump=True)
-    assert res == json.dumps({"username": USERNAMES[0], "age": AGE, "phone": None})
+    assert res == json.dumps({"username": USERNAMES[0], "age": AGE, "phone": None}, indent=2)
 
     res = serialize(users, UserModel, many=True, json_dump=True)
-    assert res == json.dumps([{"username": username, "age": AGE, "phone": None} for username in USERNAMES])
+    assert res == json.dumps([{"username": username, "age": AGE, "phone": None} for username in USERNAMES], indent=2)
 
     res = serialize(users, UserModel, many=True, include=["username"], json_dump=True)
-    assert res == json.dumps([{"username": username} for username in USERNAMES])
+    assert res == json.dumps([{"username": username} for username in USERNAMES], indent=2)
 
     res = serialize(users, UserModel, many=True, exclude=["age"], json_dump=True)
-    assert res == json.dumps([{"username": username, "phone": None} for username in USERNAMES])
+    assert res == json.dumps([{"username": username, "phone": None} for username in USERNAMES], indent=2)
 
     res = serialize(users, UserModel, many=True, exclude_none=True, json_dump=True)
-    assert res == json.dumps([{"username": username, "age": AGE} for username in USERNAMES])
+    assert res == json.dumps([{"username": username, "age": AGE} for username in USERNAMES], indent=2)
 
 
 def test_serialize_negative(db):