qbug
621b889e38
- Added new nodes for JSON manipulation: RandomJSONValueNode, JSONObjectIteratorNode, JSONArrayIteratorNode, JSONMergeNode, JSONModifierNode, JSONGeneratorNode, JSONLengthNode, JSONKeyCheckerNode, and JSONStringifierNode. - Updated __init__.py to include new node classes and their display names. - Enhanced README documentation to reflect new features and usage examples. - Bumped version to 1.1.0 in pyproject.toml to signify the addition of new functionalities. This update significantly expands the capabilities of the JSON handling nodes in ComfyUI, providing users with a robust toolkit for JSON data processing.
56 lines
2.0 KiB
Python
56 lines
2.0 KiB
Python
import json
|
|
from typing import Union
|
|
|
|
class JSONGeneratorNode:
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"key_value_pairs": ("STRING", {"multiline": True}),
|
|
"is_array": ("BOOLEAN", {"default": False}),
|
|
},
|
|
}
|
|
|
|
RETURN_TYPES = ("STRING",)
|
|
RETURN_NAMES = ("generated_json",)
|
|
FUNCTION = "generate_json"
|
|
CATEGORY = "utils"
|
|
|
|
def generate_json(self, key_value_pairs: str, is_array: bool) -> tuple[str]:
|
|
try:
|
|
# Split input into lines and filter out empty lines
|
|
pairs = [line.strip() for line in key_value_pairs.split('\n') if line.strip()]
|
|
|
|
if is_array:
|
|
# For array, each line is treated as an array element
|
|
result = []
|
|
for item in pairs:
|
|
try:
|
|
# Try to parse as JSON first
|
|
result.append(json.loads(item))
|
|
except json.JSONDecodeError:
|
|
# If not valid JSON, add as string
|
|
result.append(item)
|
|
else:
|
|
# For object, each line should be in "key: value" format
|
|
result = {}
|
|
for pair in pairs:
|
|
if ':' not in pair:
|
|
raise ValueError(f"Invalid key-value pair: {pair}")
|
|
|
|
key, value = pair.split(':', 1)
|
|
key = key.strip()
|
|
value = value.strip()
|
|
|
|
try:
|
|
# Try to parse value as JSON
|
|
value = json.loads(value)
|
|
except json.JSONDecodeError:
|
|
# If not valid JSON, keep as string
|
|
pass
|
|
|
|
result[key] = value
|
|
|
|
return (json.dumps(result, indent=2),)
|
|
except Exception as e:
|
|
raise ValueError(f"Error generating JSON: {str(e)}") |