qbug
8a91591a38
- Add SimpleJSONParserNode class with JSON parsing functionality - Implement path-based access for nested JSON structures - Handle various data types and error conditions
48 lines
1.4 KiB
Python
48 lines
1.4 KiB
Python
import json
|
|
from typing import Union
|
|
|
|
class SimpleJSONParserNode:
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"json_string": ("STRING", {"multiline": True}),
|
|
"path": ("STRING", {"default": ""}),
|
|
},
|
|
}
|
|
|
|
RETURN_TYPES = ("STRING",)
|
|
FUNCTION = "parse_json"
|
|
CATEGORY = "utils"
|
|
|
|
def parse_json(self, json_string: str, path: str) -> tuple[str]:
|
|
try:
|
|
data = json.loads(json_string)
|
|
if not path:
|
|
return (json.dumps(data, indent=2),)
|
|
|
|
keys = path.split('.')
|
|
for key in keys:
|
|
if key.isdigit():
|
|
data = data[int(key)]
|
|
elif '[' in key and ']' in key:
|
|
list_key, index = key[:-1].split('[')
|
|
data = data[list_key][int(index)]
|
|
else:
|
|
data = data[key]
|
|
|
|
if isinstance(data, (dict, list)):
|
|
return (json.dumps(data, indent=2),)
|
|
else:
|
|
return (str(data),)
|
|
except json.JSONDecodeError:
|
|
raise ValueError("Invalid JSON string",)
|
|
except (KeyError, IndexError, TypeError):
|
|
raise ValueError("Invalid path or key not found",)
|
|
|
|
@classmethod
|
|
def IS_CHANGED(cls, **kwargs):
|
|
return float("NaN")
|
|
|
|
|