🐛 fix for dumping JSON in CLI

Author: ianardee

CHANGELOG.md

@@ -6,6 +6,7 @@
 * :bug: fix for locale constructor
 * :bug: fix custom document in CLI
 * :label: declare type info to mypy
+* :bug: fix for dumping JSON in CLI
 
 ### Changes
 * :label: set stricter pylint and mypy settings

mindee/__main__.py

@@ -4,6 +4,7 @@
 from typing import Any, Dict
 
 from mindee import Client
+from mindee.documents.base import serialize_for_json
 
 DOCUMENTS: Dict[str, Dict[str, Any]] = {
     "invoice": {
@@ -91,7 +92,8 @@ def call_endpoint(args: Namespace):
     if args.output_type == "raw":
         print(json.dumps(parsed_data.http_response, indent=2))
     elif args.output_type == "parsed":
-        print(json.dumps(getattr(parsed_data, doc_type), indent=2, default=vars))
+        doc = getattr(parsed_data, doc_type)
+        print(json.dumps(doc, indent=2, default=serialize_for_json))
     else:
         print(getattr(parsed_data, doc_type))
 

mindee/documents/base.py

@@ -1,9 +1,21 @@
+import datetime
 from typing import List, Optional, Type
 
 # from mindee.inputs import InputDocument
 from mindee.http import Endpoint
 
 
+def serialize_for_json(obj):
+    """
+    Custom serializer for Document objects.
+
+    Use as the `default` argument of the `json.dump` functions.
+    """
+    if isinstance(obj, datetime.date):
+        return obj.__str__()
+    return vars(obj)
+
+
 class Document:
     type: str
     checklist: dict = {}