2025-07-01

2026-03-17 14:30:01 -06:00
parent f9a22056dd
commit 62b5978595
4579 changed files with 1257472 additions and 0 deletions
@@ -0,0 +1,86 @@
+"""
+Copyright (C) 2023 Adobe.
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program.  If not, see <http://www.gnu.org/licenses/>.
+"""
+
+# file: presets/manager.py
+# brief: Manager for preset related operations
+# author Adobe - 3D & Immersive
+# copyright 2023 Adobe Inc. All rights reserved.
+
+
+import xml.etree.ElementTree as ET
+
+import traceback
+
+from ..utils import SUBSTANCE_Utils
+from ..common import Code_Response, PRESET_EXTENSION
+
+
+class SUBSTANCE_PresetManager():
+    def __init__(self):
+        pass
+
+    def preset_write_file(self, filepath, preset):
+        try:
+            _dir_path = filepath + preset.name + PRESET_EXTENSION
+            _root = ET.Element('sbspresets')
+            _root.set('formatversion', '1.1')
+            _root.set('count', '1')
+
+            _preset_xml = ET.fromstring(preset.value)
+
+            _root.append(_preset_xml)
+
+            _tree = ET.ElementTree(_root)
+            ET.indent(_tree, space=" ", level=0)
+            _tree.write(_dir_path)
+            return Code_Response.success
+
+        except Exception:
+            SUBSTANCE_Utils.log_data("ERROR", "Exception - Writing preset error:")
+            SUBSTANCE_Utils.log_traceback(traceback.format_exc())
+
+            return Code_Response.preset_export_error
+
+    def preset_read_file(self, filepath):
+        try:
+            with open(filepath) as _f:
+                _preset_file = _f.read()
+            _root = ET.fromstring(_preset_file)
+
+            _obj = {}
+
+            for _child in _root.findall('sbspreset'):
+                _preset_value = ET.tostring(_child, encoding='unicode')
+
+                _graph = _child.attrib['pkgurl'].split('?')[0]
+                _graph = _graph.replace("pkg:///", "")
+                _graph = _graph.replace("pkg://", "")
+
+                if _graph not in _obj:
+                    _obj[_graph] = []
+
+                _obj[_graph].append({
+                    "label": _child.attrib['label'],
+                    "value": _preset_value
+                })
+
+            return (Code_Response.success, _obj)
+
+        except Exception:
+            SUBSTANCE_Utils.log_data("ERROR", "Exception - Reading preset error:")
+            SUBSTANCE_Utils.log_traceback(traceback.format_exc())
+
+            return (Code_Response.preset_import_error, None)