Pass indent to ext module

Author: WillAyd

pandas/_libs/src/ujson/python/objToJSON.c

@@ -2397,6 +2397,8 @@ PyObject *objToJSON(PyObject *self, PyObject *args, PyObject *kwargs) {
         pyEncoder.defaultHandler = odefHandler;
     }
 
+    encoder->indent = indent;
+
     pyEncoder.originalOutputFormat = pyEncoder.outputFormat;
     PRINTMARK();
     ret = JSON_EncodeObject(oinput, encoder, buffer, sizeof(buffer));

pandas/core/generic.py

@@ -2178,7 +2178,7 @@ def to_excel(self, excel_writer, sheet_name="Sheet1", na_rep="",
     def to_json(self, path_or_buf=None, orient=None, date_format=None,
                 double_precision=10, force_ascii=True, date_unit='ms',
                 default_handler=None, lines=False, compression='infer',
-                index=True):
+                index=True, indent=0):
         """
         Convert the object to a JSON string.
 
@@ -2260,6 +2260,11 @@ def to_json(self, path_or_buf=None, orient=None, date_format=None,
 
             .. versionadded:: 0.23.0
 
+        indent : integer, default 0
+           Length of whitespace used to indent each record.
+
+           .. versionadded:: 0.25.0
+
         Returns
         -------
         None or str
@@ -2325,7 +2330,7 @@ def to_json(self, path_or_buf=None, orient=None, date_format=None,
                             force_ascii=force_ascii, date_unit=date_unit,
                             default_handler=default_handler,
                             lines=lines, compression=compression,
-                            index=index)
+                            index=index, indent=indent)
 
     def to_hdf(self, path_or_buf, key, **kwargs):
         """

pandas/io/json/json.py

@@ -33,7 +33,7 @@
 def to_json(path_or_buf, obj, orient=None, date_format='epoch',
             double_precision=10, force_ascii=True, date_unit='ms',
             default_handler=None, lines=False, compression='infer',
-            index=True):
+            index=True, indent=0):
 
     if not index and orient not in ['split', 'table']:
         raise ValueError("'index=False' is only valid when 'orient' is "
@@ -59,7 +59,7 @@ def to_json(path_or_buf, obj, orient=None, date_format='epoch',
         obj, orient=orient, date_format=date_format,
         double_precision=double_precision, ensure_ascii=force_ascii,
         date_unit=date_unit, default_handler=default_handler,
-        index=index).write()
+        index=index, indent=indent).write()
 
     if lines:
         s = _convert_to_line_delimits(s)
@@ -78,7 +78,8 @@ def to_json(path_or_buf, obj, orient=None, date_format='epoch',
 
 class Writer:
     def __init__(self, obj, orient, date_format, double_precision,
-                 ensure_ascii, date_unit, index, default_handler=None):
+                 ensure_ascii, date_unit, index, default_handler=None,
+                 indent=0):
         self.obj = obj
 
         if orient is None:
@@ -91,6 +92,7 @@ def __init__(self, obj, orient, date_format, double_precision,
         self.date_unit = date_unit
         self.default_handler = default_handler
         self.index = index
+        self.indent = indent
 
         self.is_copy = None
         self._format_axes()
@@ -101,18 +103,20 @@ def _format_axes(self):
     def write(self):
         return self._write(self.obj, self.orient, self.double_precision,
                            self.ensure_ascii, self.date_unit,
-                           self.date_format == 'iso', self.default_handler)
+                           self.date_format == 'iso', self.default_handler,
+                           self.indent)
 
     def _write(self, obj, orient, double_precision, ensure_ascii,
-               date_unit, iso_dates, default_handler):
+               date_unit, iso_dates, default_handler, indent):
         return dumps(
             obj,
             orient=orient,
             double_precision=double_precision,
             ensure_ascii=ensure_ascii,
             date_unit=date_unit,
             iso_dates=iso_dates,
-            default_handler=default_handler
+            default_handler=default_handler,
+            indent=indent
         )
 
 
@@ -125,11 +129,11 @@ def _format_axes(self):
                              "'{orient}'".format(orient=self.orient))
 
     def _write(self, obj, orient, double_precision, ensure_ascii,
-               date_unit, iso_dates, default_handler):
+               date_unit, iso_dates, default_handler, indent):
         if not self.index and orient == 'split':
             obj = {"name": obj.name, "data": obj.values}
         return super()._write(obj, orient, double_precision, ensure_ascii,
-                              date_unit, iso_dates, default_handler)
+                              date_unit, iso_dates, default_handler, indent)
 
 
 class FrameWriter(Writer):
@@ -149,19 +153,20 @@ def _format_axes(self):
                              "'{orient}'.".format(orient=self.orient))
 
     def _write(self, obj, orient, double_precision, ensure_ascii,
-               date_unit, iso_dates, default_handler):
+               date_unit, iso_dates, default_handler, indent):
         if not self.index and orient == 'split':
             obj = obj.to_dict(orient='split')
             del obj["index"]
         return super()._write(obj, orient, double_precision, ensure_ascii,
-                              date_unit, iso_dates, default_handler)
+                              date_unit, iso_dates, default_handler, indent)
 
 
 class JSONTableWriter(FrameWriter):
     _default_orient = 'records'
 
     def __init__(self, obj, orient, date_format, double_precision,
-                 ensure_ascii, date_unit, index, default_handler=None):
+                 ensure_ascii, date_unit, index, default_handler=None,
+                 indent=0):
         """
         Adds a `schema` attribute with the Table Schema, resets
         the index (can't do in caller, because the schema inference needs
@@ -170,7 +175,7 @@ def __init__(self, obj, orient, date_format, double_precision,
         """
         super().__init__(obj, orient, date_format, double_precision,
                          ensure_ascii, date_unit, index,
-                         default_handler=default_handler)
+                         default_handler=default_handler, indent=indent)
 
         if date_format != 'iso':
             msg = ("Trying to write with `orient='table'` and "
@@ -211,9 +216,9 @@ def __init__(self, obj, orient, date_format, double_precision,
         self.index = index
 
     def _write(self, obj, orient, double_precision, ensure_ascii,
-               date_unit, iso_dates, default_handler):
+               date_unit, iso_dates, default_handler, indent):
         data = super()._write(obj, orient, double_precision, ensure_ascii,
-                              date_unit, iso_dates, default_handler)
+                              date_unit, iso_dates, default_handler, indent)
         serialized = '{{"schema": {schema}, "data": {data}}}'.format(
                      schema=dumps(self.schema), data=data)
         return serialized