Untitled
unknown
python
2 years ago
929 B
34
Indexable
def getProperty(self, png_file_path, node_id, input_key):
# Open the PNG file
with Image.open(png_file_path) as img:
# Extract metadata
metadata = img.info
# Look for the custom JSON key in the metadata
json_data = metadata.get('prompt')
if json_data is None:
raise ValueError("No 'prompt' JSON metadata found in the PNG file.")
# Parse the JSON metadata
try:
data = json.loads(json_data)
except json.JSONDecodeError as e:
raise ValueError("Invalid JSON data found in the metadata.") from e
# Search for the node with the given ID
node_info = data.get(str(node_id))
if node_info is None:
raise ValueError(f"Node with ID {node_id} not found.")
# Extract the input value
input_value = node_info.get('inputs', {}).get(input_key)
if input_value is None:
raise ValueError(f"Input key '{input_key}' not found in node ID {node_id}.")
print(input_value)Editor is loading...
Leave a Comment