Update

2023/scripts/animation_tools/studiolibrary/studiolibrarymaya/setsitem.py

@@ -0,0 +1,65 @@
+# Copyright 2020 by Kurt Rathjen. All Rights Reserved.
+#
+# This library is free software: you can redistribute it and/or modify it 
+# under the terms of the GNU Lesser General Public License as published by 
+# the Free Software Foundation, either version 3 of the License, or 
+# (at your option) any later version. This library 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 Lesser General Public License for more details.
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library. If not, see <http://www.gnu.org/licenses/>.
+
+import os
+
+try:
+    import mutils
+except ImportError as error:
+    print(error)
+
+from studiolibrarymaya import baseitem
+
+
+def save(path, *args, **kwargs):
+    """Convenience function for saving a SetsItem."""
+    SetsItem(path).safeSave(*args, **kwargs)
+
+
+def load(path, *args, **kwargs):
+    """Convenience function for loading a SetsItem."""
+    SetsItem(path).load(*args, **kwargs)
+
+
+class SetsItem(baseitem.BaseItem):
+
+    NAME = "Selection Set"
+    EXTENSION = ".set"
+    ICON_PATH = os.path.join(os.path.dirname(__file__), "icons", "selectionSet.png")
+    TRANSFER_CLASS = mutils.SelectionSet
+    TRANSFER_BASENAME = "set.json"
+
+    def loadFromCurrentValues(self):
+        """Load the selection set using the settings for this item."""
+        self.load(namespaces=self.namespaces())
+
+    def load(self, namespaces=None):
+        """
+        :type namespaces: list[str] | None
+        """
+        self.selectContent(namespaces=namespaces)
+
+    def save(self, objects, **kwargs):
+        """
+        Save all the given object data to the item path on disc.
+
+        :type objects: list[str]
+        :type kwargs: dict
+        """
+        super(SetsItem, self).save(**kwargs)
+
+        # Save the selection set to the given path
+        mutils.saveSelectionSet(
+            self.path() + "/set.json",
+            objects,
+            metadata={"description": kwargs.get("comment", "")}
+        )