rewriting .write and ._write

Author: reidy-p

pandas/io/json/json.py

@@ -55,7 +55,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).write()
 
  if lines:
  s = _convert_to_line_delimits(s)
@@ -95,15 +95,21 @@ def __init__(self, obj, orient, date_format, double_precision,
  def _format_axes(self):
  raise AbstractMethodError(self)
 
- def _write(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)
+
+ def _write(self, obj, orient, double_precision, ensure_ascii,
+ date_unit, iso_dates, default_handler):
  return dumps(
- self.obj,
- orient=self.orient,
- double_precision=self.double_precision,
- ensure_ascii=self.ensure_ascii,
- date_unit=self.date_unit,
- iso_dates=self.date_format == 'iso',
- default_handler=self.default_handler
+ obj,
+ orient=orient,
+ double_precision=double_precision,
+ ensure_ascii=ensure_ascii,
+ date_unit=date_unit,
+ iso_dates=iso_dates,
+ default_handler=default_handler
  )
 
 
@@ -115,10 +121,14 @@ def _format_axes(self):
  raise ValueError("Series index must be unique for orient="
  "'{orient}'".format(orient=self.orient))
 
- def _write(self):
- if not self.index and self.orient == 'split':
- self.obj = {"name": self.obj.name, "data": self.obj.values}
- return super(SeriesWriter, self)._write()
+ def _write(self, obj, orient, double_precision, ensure_ascii,
+ date_unit, iso_dates, default_handler):
+ if not self.index and orient == 'split':
+ obj = {"name": obj.name, "data": obj.values}
+ return super(SeriesWriter, self)._write(obj, orient,
+ double_precision,
+ ensure_ascii, date_unit,
+ iso_dates, default_handler)
 
 
 class FrameWriter(Writer):
@@ -135,11 +145,15 @@ def _format_axes(self):
  raise ValueError("DataFrame columns must be unique for orient="
  "'{orient}'.".format(orient=self.orient))
 
- def _write(self):
- if not self.index and self.orient == 'split':
- self.obj = self.obj.to_dict(orient='split')
- del self.obj["index"]
- return super(FrameWriter, self)._write()
+ def _write(self, obj, orient, double_precision, ensure_ascii,
+ date_unit, iso_dates, default_handler):
+ if not self.index and orient == 'split':
+ obj = obj.to_dict(orient='split')
+ del obj["index"]
+ return super(FrameWriter, self)._write(obj, orient,
+ double_precision,
+ ensure_ascii, date_unit,
+ iso_dates, default_handler)
 
 
 class JSONTableWriter(FrameWriter):
@@ -191,10 +205,15 @@ def __init__(self, obj, orient, date_format, double_precision,
  self.orient = 'records'
  self.index = index
 
- def _write(self):
+ def _write(self, obj, orient, double_precision, ensure_ascii,
+ date_unit, iso_dates, default_handler):
  if not self.index:
- self.obj = self.obj.drop('index', axis=1)
- data = super(JSONTableWriter, self)._write()
+ obj = obj.drop('index', axis=1)
+ data = super(JSONTableWriter, self)._write(obj, orient,
+ double_precision,
+ ensure_ascii, date_unit,
+ iso_dates,
+ default_handler)
  serialized = '{{"schema": {schema}, "data": {data}}}'.format(
  schema=dumps(self.schema), data=data)
  return serialized