json-repair 0.13.0__py3-none-any.whl → 0.13.1__py3-none-any.whl

json_repair/json_repair.py

@@ -34,6 +34,7 @@ class JSONParser:
         self.index = 0
         # This is used in the object member parsing to manage the special cases of missing quotes in key or value
         self.context = ""
+        self.context_stack = []
 
     def parse(self) -> Union[Dict[str, Any], List[Any], str, float, int, bool, None]:
         return self.parse_json()
@@ -97,7 +98,7 @@ class JSONParser:
 
             # We are now searching for they string key
             # Context is used in the string parser to manage the lack of quotes
-            self.context = "object_key"
+            self.update_context("object_key")
 
             self.skip_whitespaces_at()
 
@@ -119,12 +120,13 @@ class JSONParser:
             if (self.get_char_at() or "") != ":":
                 self.insert_char_at(":")
             self.index += 1
-            self.context = "object_value"
+            self.update_context("")
+            self.update_context("object_value")
             # The value can be any valid json
             value = self.parse_json()
 
             # Reset context since our job is done
-            self.context = ""
+            self.update_context("")
             obj[key] = value
 
             if (self.get_char_at() or "") in [",", "'", '"']:
@@ -157,6 +159,9 @@ class JSONParser:
             while char and (char.isspace() or char == ","):
                 self.index += 1
                 char = self.get_char_at()
+            # If this is the right value of an object and we are closing the object, it means the array is over
+            if self.context == "object_value" and char == "}":
+                break
 
         # Especially at the end of an LLM generated json you might miss the last "]"
         char = self.get_char_at()
@@ -166,6 +171,7 @@ class JSONParser:
                 # Remove trailing "," before adding the "]"
                 self.remove_char_at()
             self.insert_char_at("]")
+            self.index -= 1
 
         self.index += 1
         return arr
@@ -319,6 +325,19 @@ class JSONParser:
             except IndexError:
                 return
 
+    def update_context(self, value: str) -> None:
+        # If a value is provided update the context variable and save in stack
+        if value:
+            if self.context:
+                self.context_stack.append(self.context)
+            self.context = value
+        # Otherwise pop and update the context, or empty if the stack is empty
+        else:
+            try:
+                self.context = self.context_stack.pop()
+            except Exception:
+                self.context = ""
+
 
 def repair_json(
     json_str: str, return_objects: bool = False, skip_json_loads: bool = False

{json_repair-0.13.0.dist-info → json_repair-0.13.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: json_repair
-Version: 0.13.0
+Version: 0.13.1
 Summary: A package to repair broken json strings
 Author-email: Stefano Baccianella <4247706+mangiucugna@users.noreply.github.com>
 License: MIT License

json_repair-0.13.1.dist-info/RECORD

@@ -0,0 +1,7 @@
+json_repair/__init__.py,sha256=AlNie5y6BZBioGi5fzTAUvum_y0U5aL5aNsuQ_68LQc,175
+json_repair/json_repair.py,sha256=DB220fZ1BCf--9CeP6AzL2FCk9tpE1Eh3WxgFo33P88,15460
+json_repair-0.13.1.dist-info/LICENSE,sha256=wrjQo8MhNrNCicXtMe3MHmS-fx8AmQk1ue8AQwiiFV8,1076
+json_repair-0.13.1.dist-info/METADATA,sha256=gcfdAU5nhvlHlE3wMvGcwHRQUzgPT2p08cgDg7iyLvw,7200
+json_repair-0.13.1.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+json_repair-0.13.1.dist-info/top_level.txt,sha256=7-VZwZN2CgB_n0NlSLk-rEUFh8ug21lESbsblOYuZqw,12
+json_repair-0.13.1.dist-info/RECORD,,

json_repair-0.13.0.dist-info/RECORD

@@ -1,7 +0,0 @@
-json_repair/__init__.py,sha256=AlNie5y6BZBioGi5fzTAUvum_y0U5aL5aNsuQ_68LQc,175
-json_repair/json_repair.py,sha256=BHDcvo8C-vc3QO4dhgD5cP7IZVkGWGDbv8hNmQH1B-I,14653
-json_repair-0.13.0.dist-info/LICENSE,sha256=wrjQo8MhNrNCicXtMe3MHmS-fx8AmQk1ue8AQwiiFV8,1076
-json_repair-0.13.0.dist-info/METADATA,sha256=E6v_JcBwXe4oMJhtxgF-9-24kG2j5LiH-ivqufNHFpA,7200
-json_repair-0.13.0.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-json_repair-0.13.0.dist-info/top_level.txt,sha256=7-VZwZN2CgB_n0NlSLk-rEUFh8ug21lESbsblOYuZqw,12
-json_repair-0.13.0.dist-info/RECORD,,