json-repair 0.28.0__py3-none-any.whl → 0.28.1__py3-none-any.whl

json_repair/json_repair.py

@@ -24,7 +24,7 @@ All supported use cases are in the unit tests
 
 import os
 import json
-from typing import Any, Dict, List, Optional, Union, TextIO, Tuple
+from typing import Any, Dict, List, Optional, Union, TextIO, Tuple, overload, Literal
 
 
 class StringFileWrapper:
@@ -177,7 +177,7 @@ class JSONParser:
             # <member> starts with a <string>
             key = ""
             while self.get_char_at():
-                key = self.parse_string()
+                key = str(self.parse_string())
 
                 if key != "" or (key == "" and self.get_char_at() == ":"):
                     # If the string is empty but there is a object divider, we are done here
@@ -255,7 +255,7 @@ class JSONParser:
         self.reset_context()
         return arr
 
-    def parse_string(self) -> Union[str, JSONReturnType]:
+    def parse_string(self) -> Union[str, bool, None]:
         # <string> is a string of valid characters enclosed in quotes
         # i.e. { name: "John" }
         # Somehow all weird cases in an invalid JSON happen to be resolved in this function, so be careful here
@@ -382,7 +382,7 @@ class JSONParser:
             string_acc += char
             self.index += 1
             char = self.get_char_at()
-            if len(string_acc) > 0 and string_acc[-1] == "\\":
+            if char and len(string_acc) > 0 and string_acc[-1] == "\\":
                 # This is a special case, if people use real strings this might happen
                 self.log("Found a stray escape sequence, normalizing it", "info")
                 string_acc = string_acc[:-1]
@@ -473,7 +473,7 @@ class JSONParser:
                                 "While parsing a string, we a misplaced quote that would have closed the string but has a different meaning here since this is the last element of the object, ignoring it",
                                 "info",
                             )
-                            string_acc += char
+                            string_acc += str(char)
                             self.index += 1
                             char = self.get_char_at()
                     elif next_c == rstring_delimiter:
@@ -503,7 +503,7 @@ class JSONParser:
                                     "While parsing a string, we a misplaced quote that would have closed the string but has a different meaning here, ignoring it",
                                     "info",
                                 )
-                                string_acc += char
+                                string_acc += str(char)
                                 self.index += 1
                                 char = self.get_char_at()
 
@@ -521,7 +521,8 @@ class JSONParser:
             if self.get_char_at() not in [":", ","]:
                 return ""
 
-        # A fallout of the previous special case in the while loop, we need to update the index only if we had a closing quote
+        # A fallout of the previous special case in the while loop,
+        # we need to update the index only if we had a closing quote
         if char != rstring_delimiter:
             self.log(
                 "While parsing a string, we missed the closing quote, ignoring",
@@ -563,6 +564,7 @@ class JSONParser:
         # <boolean> is one of the literal strings 'true', 'false', or 'null' (unquoted)
         starting_index = self.index
         char = (self.get_char_at() or "").lower()
+        value = None
         if char == "t":
             value = ("true", True)
         elif char == "f":
@@ -583,7 +585,7 @@ class JSONParser:
         self.index = starting_index
         return ""
 
-    def get_char_at(self, count: int = 0) -> Union[str, bool]:
+    def get_char_at(self, count: int = 0) -> Union[str, Literal[False]]:
         # Why not use something simpler? Because try/except in python is a faster alternative to an "if" statement that is often True
         try:
             return self.json_str[self.index + count]
@@ -630,6 +632,50 @@ class JSONParser:
             )
 
 
+@overload
+def repair_json(
+    json_str: str = "",
+    return_objects: Optional[Literal[False]] = False,
+    skip_json_loads: Optional[bool] = False,
+    logging: Optional[Literal[False]] = False,  # None is treated as False
+    json_fd: Optional[TextIO] = None,
+    ensure_ascii: Optional[bool] = True,
+) -> str: ...
+
+
+@overload
+def repair_json(
+    json_str: str = "",
+    return_objects: Literal[True] = True,
+    skip_json_loads: Optional[bool] = False,
+    logging: Optional[Literal[False]] = False,  # None is treated as False
+    json_fd: Optional[TextIO] = None,
+    ensure_ascii: Optional[bool] = True,
+) -> JSONReturnType: ...
+
+
+@overload
+def repair_json(
+    json_str: str = "",
+    return_objects: Optional[Literal[False]] = False,  # None is treated as False
+    skip_json_loads: Optional[bool] = False,
+    logging: Literal[True] = True,
+    json_fd: Optional[TextIO] = None,
+    ensure_ascii: Optional[bool] = True,
+) -> Tuple[str, List[Dict[str, str]]]: ...
+
+
+@overload
+def repair_json(
+    json_str: str = "",
+    return_objects: Literal[True] = True,
+    skip_json_loads: Optional[bool] = False,
+    logging: Literal[True] = True,
+    json_fd: Optional[TextIO] = None,
+    ensure_ascii: Optional[bool] = True,
+) -> Tuple[JSONReturnType, List[Dict[str, str]]]: ...
+
+
 def repair_json(
     json_str: str = "",
     return_objects: Optional[bool] = False,
@@ -643,7 +689,7 @@ def repair_json(
     It will return the fixed string by default.
     When `return_objects=True` is passed, it will return the decoded data structure instead.
     When `skip_json_loads=True` is passed, it will not call the built-in json.loads() function
-    When `logging=True` is passed, it will return an tuple with the repaired json and a log of all repair actions
+    When `logging=True` is passed, it will return a tuple with the repaired json and a log of all repair actions
     """
     parser = JSONParser(json_str, json_fd, logging)
     if skip_json_loads:
@@ -656,12 +702,29 @@ def repair_json(
                 parsed_json = json.loads(json_str)
         except json.JSONDecodeError:
             parsed_json = parser.parse()
-    # It's useful to return the actual object instead of the json string, it allows this lib to be a replacement of the json library
+    # It's useful to return the actual object instead of the json string,
+    # it allows this lib to be a replacement of the json library
     if return_objects or logging:
         return parsed_json
     return json.dumps(parsed_json, ensure_ascii=ensure_ascii)
 
 
+@overload
+def loads(
+    json_str: str,
+    skip_json_loads: Optional[bool] = False,
+    logging: Optional[Literal[False]] = False,  # None is treated as False
+) -> JSONReturnType: ...
+
+
+@overload
+def loads(
+    json_str: str,
+    skip_json_loads: Optional[bool] = False,
+    logging: Literal[True] = True,
+) -> Tuple[JSONReturnType, List[Dict[str, str]]]: ...
+
+
 def loads(
     json_str: str,
     skip_json_loads: Optional[bool] = False,
@@ -679,6 +742,20 @@ def loads(
     )
 
 
+@overload
+def load(
+    fd: TextIO,
+    skip_json_loads: Optional[bool] = False,
+    logging: Optional[Literal[False]] = False,
+) -> JSONReturnType: ...
+
+
+@overload
+def load(
+    fd: TextIO, skip_json_loads: Optional[bool] = False, logging: Literal[True] = True
+) -> Tuple[JSONReturnType, List[Dict[str, str]]]: ...
+
+
 def load(
     fd: TextIO, skip_json_loads: Optional[bool] = False, logging: Optional[bool] = False
 ) -> Union[JSONReturnType, Tuple[JSONReturnType, List[Dict[str, str]]]]:
@@ -694,6 +771,22 @@ def load(
     )
 
 
+@overload
+def from_file(
+    filename: str,
+    skip_json_loads: Optional[bool] = False,
+    logging: Optional[Literal[False]] = False,
+) -> JSONReturnType: ...
+
+
+@overload
+def from_file(
+    filename: str,
+    skip_json_loads: Optional[bool] = False,
+    logging: Literal[True] = True,
+) -> Tuple[JSONReturnType, List[Dict[str, str]]]: ...
+
+
 def from_file(
     filename: str,
     skip_json_loads: Optional[bool] = False,

{json_repair-0.28.0.dist-info → json_repair-0.28.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: json_repair
-Version: 0.28.0
+Version: 0.28.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.28.1.dist-info/RECORD

@@ -0,0 +1,8 @@
+json_repair/__init__.py,sha256=IIzSm1DsCRrr8seF3UeMZXwxcq-tE3j-8d1WBxvEJvE,178
+json_repair/json_repair.py,sha256=_NBAaY6iqIp1cB1W-lnQ3uS6nc5DjZZyf_HeklYmDyY,32502
+json_repair/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+json_repair-0.28.1.dist-info/LICENSE,sha256=wrjQo8MhNrNCicXtMe3MHmS-fx8AmQk1ue8AQwiiFV8,1076
+json_repair-0.28.1.dist-info/METADATA,sha256=i9SaIoWFc7YjuQLLN1Rd8GsmVAy0rWJ9-fNwsVDR_KA,8043
+json_repair-0.28.1.dist-info/WHEEL,sha256=HiCZjzuy6Dw0hdX5R3LCFPDmFS4BWl8H-8W39XfmgX4,91
+json_repair-0.28.1.dist-info/top_level.txt,sha256=7-VZwZN2CgB_n0NlSLk-rEUFh8ug21lESbsblOYuZqw,12
+json_repair-0.28.1.dist-info/RECORD,,

json_repair-0.28.0.dist-info/RECORD

@@ -1,8 +0,0 @@
-json_repair/__init__.py,sha256=IIzSm1DsCRrr8seF3UeMZXwxcq-tE3j-8d1WBxvEJvE,178
-json_repair/json_repair.py,sha256=5WjVoNO7Grdq9wfnbSjggiVlDv2XVcvWbCIQVmREP38,30109
-json_repair/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-json_repair-0.28.0.dist-info/LICENSE,sha256=wrjQo8MhNrNCicXtMe3MHmS-fx8AmQk1ue8AQwiiFV8,1076
-json_repair-0.28.0.dist-info/METADATA,sha256=rPrWno-My7ZQt7EJIR0BodmpLQ84fr0vjJwBM4XYmYU,8043
-json_repair-0.28.0.dist-info/WHEEL,sha256=HiCZjzuy6Dw0hdX5R3LCFPDmFS4BWl8H-8W39XfmgX4,91
-json_repair-0.28.0.dist-info/top_level.txt,sha256=7-VZwZN2CgB_n0NlSLk-rEUFh8ug21lESbsblOYuZqw,12
-json_repair-0.28.0.dist-info/RECORD,,