diff --git a/tts_yaml_unpacker.py b/tts_yaml_unpacker.py index 624bee7..03af549 100755 --- a/tts_yaml_unpacker.py +++ b/tts_yaml_unpacker.py @@ -97,8 +97,12 @@ def unpackJson(json_file, output_name): with open(json_file) as f: data = json.load(f) + if 'ObjectStates' in data: + prop = 'ObjectStates' + else: + prop = 'ContainedObjects' recursivelyUnpackObject( - '', data, subobject_prop='ObjectStates', base_name=output_name) + '', data, subobject_prop=prop, base_name=output_name) def packYaml(yaml_file, output_json_file):