57 lines
3.1 KiB
Python
57 lines
3.1 KiB
Python
# Copyright 2025 FeiMao-326
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# Copyright 2025 FeiMao-326
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
import json
|
|
|
|
class FeiMao_326_JsonParser:
|
|
MAX_OUTPUTS = 50
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
default_json = {"shot": {"composition": "central low-angle wide"}, "subject": {"description": "CBR 150 assembling"}, "scene": {"location": "desert basin"}, "audio": {"music": "trailer score"}}
|
|
return {"required": {"json_payload": ("STRING", {"multiline": True, "default": json.dumps(default_json, indent=4)}), "keys_to_extract": ("STRING", {"multiline": False, "default": "shot, subject, scene, audio"})}}
|
|
RETURN_TYPES = tuple(["STRING"] * 50); RETURN_NAMES = tuple([f"output_{i}" for i in range(1, 51)]); FUNCTION = "execute"; CATEGORY = "FeiMao-326"
|
|
def _find_and_combine_values(self, data, target_key):
|
|
found_values = []
|
|
if isinstance(data, dict):
|
|
for key, value in data.items():
|
|
if key == target_key:
|
|
if isinstance(value, dict): found_values.append(", ".join(filter(None, [str(v) for v in value.values() if v])))
|
|
elif isinstance(value, list): found_values.append(", ".join(filter(None, [str(v) for v in value if v])))
|
|
elif value: found_values.append(str(value))
|
|
else: found_values.extend(self._find_and_combine_values(value, target_key))
|
|
elif isinstance(data, list):
|
|
for item in data: found_values.extend(self._find_and_combine_values(item, target_key))
|
|
return found_values
|
|
def execute(self, json_payload, keys_to_extract):
|
|
output_keys = [key.strip() for key in keys_to_extract.split(',') if key.strip()]
|
|
try: data = json.loads(json_payload)
|
|
except json.JSONDecodeError as e:
|
|
error_message = f"ERROR: Invalid JSON format. {e}"; print(f"❌ [FeiMao-326 JSON Parser] {error_message}")
|
|
return tuple([error_message] + [""] * (self.MAX_OUTPUTS - 1))
|
|
results = []
|
|
for i in range(self.MAX_OUTPUTS):
|
|
if i < len(output_keys):
|
|
key = output_keys[i]; found = self._find_and_combine_values(data, key); results.append(", ".join(found))
|
|
else: results.append("")
|
|
return tuple(results) |