fix json printer

Author: boolangery

luaparser/ast.py

@@ -82,6 +82,9 @@ def to_xml_str(tree):
 
 class JSONEncoder(json.JSONEncoder):
     def default(self, o):
+        if isinstance(o, bytes):
+            return o.decode()
+
         try:
             to_json = getattr(o, "to_json")
             if callable(to_json):

luaparser/tests/test_ast.py

@@ -99,3 +99,43 @@ def test_cont_int_1(self):
         ]
         for node, exp in zip(nodes, expected_cls):
             self.assertIsInstance(node, exp)
+
+    def test_to_pretty_json(self):
+        src = textwrap.dedent(
+            """\
+            local a = "foo"\
+            """
+        )
+        exp = textwrap.dedent(
+            """\
+            {
+                "comments": [],
+                "body": {
+                    "comments": [],
+                    "body": [
+                        {
+                            "comments": [],
+                            "wrapped": false,
+                            "targets": [
+                                {
+                                    "comments": [],
+                                    "wrapped": false,
+                                    "id": "a",
+                                    "attribute": null
+                                }
+                            ],
+                            "values": [
+                                {
+                                    "comments": [],
+                                    "wrapped": false,
+                                    "s": "foo",
+                                    "raw": "foo",
+                                    "delimiter": {}
+                                }
+                            ]
+                        }
+                    ]
+                }
+            }"""
+        )
+        self.assertEqual(ast.to_pretty_json(ast.parse(src)), exp)