Merge pull request #1 from KNMI/support_indent

Support indentation of JSON

README.md

@@ -53,7 +53,7 @@ c = Coverage(
         "temperature": NdArray(axisNames=["x", "y", "t"], shape=[1, 1, 1], values=[42.0])
     }
 )
-print(c.json(exclude_none=True))
+print(c.json(exclude_none=True,indent=True))
 ```
 Will print
 ```json

example.py

@@ -1,15 +1,9 @@
 import datetime
-import json
 
 from covjson_pydantic.coverage import Coverage
 from covjson_pydantic.domain import Domain
 from covjson_pydantic.ndarray import NdArray
 
-
-def pretty_print_json(json_str):
-    print(json.dumps(json.loads(json_str), indent=4))
-
-
 c = Coverage(
     domain=Domain(
         domainType="PointSeries",
@@ -22,4 +16,4 @@ def pretty_print_json(json_str):
     ranges={"temperature": NdArray(axisNames=["x", "y", "t"], shape=[1, 1, 1], values=[42.0])},
 )
 
-pretty_print_json(c.json(exclude_none=True))
+print(c.json(exclude_none=True, indent=True))

src/covjson_pydantic/base_models.py

@@ -3,11 +3,16 @@
 from pydantic import Extra
 
 
-def orjson_dumps(v, *, default):
+def orjson_dumps(v, *, default, indent=None, sort_keys=False):
+    options = orjson.OPT_NON_STR_KEYS | orjson.OPT_UTC_Z | orjson.OPT_NAIVE_UTC
+    if indent:
+        options |= orjson.OPT_INDENT_2
+
+    if sort_keys:
+        options |= orjson.OPT_SORT_KEYS
+
     # orjson.dumps returns bytes, to match standard json.dumps we need to decode
-    return orjson.dumps(
-        v, default=default, option=orjson.OPT_NON_STR_KEYS | orjson.OPT_UTC_Z | orjson.OPT_NAIVE_UTC
-    ).decode()
+    return orjson.dumps(v, default=default, option=options).decode()
 
 
 class BaseModel(PydanticBaseModel):