Updated

Scripts/Animation/studiolibrary/src/mutils/namespace.py

@@ -0,0 +1,88 @@
+# 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