66 lines
2.1 KiB
Python
66 lines
2.1 KiB
Python
# 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", "")}
|
|
)
|