Better JSON pretty-printing for Python