89 lines
2.2 KiB
Python
89 lines
2.2 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 logging
|
|
import traceback
|
|
from collections import OrderedDict
|
|
|
|
try:
|
|
import maya.cmds
|
|
except ImportError:
|
|
traceback.print_exc()
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
__all__ = [
|
|
"getAll",
|
|
"getFromDagPath",
|
|
"getFromDagPaths",
|
|
"getFromSelection",
|
|
]
|
|
|
|
|
|
def getFromDagPaths(dagPaths):
|
|
"""
|
|
:type dagPaths: list[str]
|
|
:rtype: list[str]
|
|
"""
|
|
namespaces = []
|
|
|
|
for dagPath in dagPaths:
|
|
namespace = getFromDagPath(dagPath)
|
|
namespaces.append(namespace)
|
|
|
|
# Remove any duplicates while retaining the order
|
|
return list(OrderedDict.fromkeys(namespaces))
|
|
|
|
|
|
def getFromDagPath(dagPath):
|
|
"""
|
|
:type dagPath: str
|
|
:rtype: str
|
|
"""
|
|
shortName = dagPath.split("|")[-1]
|
|
namespace = ":".join(shortName.split(":")[:-1])
|
|
return namespace
|
|
|
|
|
|
def getFromSelection():
|
|
"""
|
|
Get the current namespaces from the selected objects in Maya.
|
|
|
|
:rtype: list[str]
|
|
"""
|
|
namespaces = [""]
|
|
|
|
try:
|
|
names = maya.cmds.ls(selection=True)
|
|
namespaces = getFromDagPaths(names) or namespaces
|
|
except NameError as error:
|
|
# Catch any errors when running this command outside of Maya
|
|
logger.exception(error)
|
|
|
|
return namespaces
|
|
|
|
|
|
def getAll():
|
|
"""
|
|
Get all the available namespaces in the scene
|
|
|
|
:rtype: list[str]
|
|
"""
|
|
IGNORE_NAMESPACES = ['UI', 'shared']
|
|
|
|
namespaces = maya.cmds.namespaceInfo(listOnlyNamespaces=True, recurse=True)
|
|
namespaces = list(set(namespaces) - set(IGNORE_NAMESPACES))
|
|
namespaces = sorted(namespaces)
|
|
|
|
return namespaces
|