Python implementation of json_reformat.

Details:

An example of how to use yajl-py and how concise your code can be. It
is more or less the same as using the C version except for not having
to allocate and de-allocate (non-C) items.

Author: pykler

examples/README.rst

@@ -5,3 +5,9 @@ To run the sample parser, from within this dir, run::
 the example can also be run from ../ as::
     
     python example/yajl_py_example.py <doc.json
+
+In this directory you will also find a python implementation of
+json_reformat, named json_reformat.py, it takes the same parameters as
+the C version, run it as follows for a thorough usage statement:
+
+    python json_reformat.py -h

examples/json_reformat.py

@@ -0,0 +1,68 @@
+'''
+Python implementation of the Yajl C json_reformat application
+'''
+
+import os
+import sys
+BASEPATH = os.path.dirname(os.path.realpath(__file__))
+sys.path = [BASEPATH, '%s/..' %BASEPATH] + sys.path
+from yajl import __version__ as yajl_version
+from yajl import *
+
+import optparse
+
+class ReformatContentHandler(YajlContentHandler):
+    '''
+    Content handler to reformat a json file using yajl_gen
+    '''
+    def __init__(self, beautify=True, indent="  "):
+        self.out = sys.stdout
+        self.conf = yajl_gen_config(beautify, indent)
+    def parse_start(self):
+        self.g = yajl.yajl_gen_alloc(byref(self.conf), None)
+    def parse_buf(self):
+        l = c_uint()
+        buf = POINTER(c_ubyte)()
+        yajl.yajl_gen_get_buf(self.g, byref(buf), byref(l))
+        self.out.write(string_at(buf, l.value))
+        yajl.yajl_gen_clear(self.g)
+    def parse_complete(self):
+        yajl.yajl_gen_free(self.g)
+    def yajl_null(self, ctx):
+        yajl.yajl_gen_null(self.g)
+    def yajl_boolean(self, ctx, boolVal):
+        yajl.yajl_gen_bool(self.g, boolVal)
+    def yajl_number(self, ctx, stringNum):
+        yajl.yajl_gen_number(self.g, c_char_p(stringNum), len(stringNum))
+    def yajl_string(self, ctx, stringVal):
+        yajl.yajl_gen_string(self.g, c_char_p(stringVal), len(stringVal))
+    def yajl_start_map(self, ctx):
+        yajl.yajl_gen_map_open(self.g)
+    def yajl_map_key(self, ctx, stringVal):
+        yajl.yajl_gen_string(self.g, c_char_p(stringVal), len(stringVal))
+    def yajl_end_map(self, ctx):
+        yajl.yajl_gen_map_close(self.g)
+    def yajl_start_array(self, ctx):
+        yajl.yajl_gen_array_open(self.g)
+    def yajl_end_array(self, ctx):
+        yajl.yajl_gen_array_close(self.g)
+
+
+def main():
+    opt_parser = optparse.OptionParser(
+        description='reformat json from stdin',
+        version='Yajl-Py for Yajl %s' %yajl_version)
+    opt_parser.add_option("-m",
+        dest="beautify", action="store_false", default=True,
+        help="minimize json rather than beautify (default)")
+    opt_parser.add_option("-u",
+        dest="check_utf8", action='store_false', default=True,
+        help="allow invalid UTF8 inside strings during parsing")
+    (options, args) = opt_parser.parse_args()
+    yajl_parser = YajlParser(
+        ReformatContentHandler(beautify=options.beautify),
+        check_utf8=options.check_utf8)
+    yajl_parser.parse()
+
+if __name__ == "__main__":
+    main()