on import create a set containing what was imported

Author: vkovec

Assets/Integrations/Autodesk/maya2017/scripts/unityOneClick/commands.py

@@ -39,6 +39,7 @@ class BaseCommand(OpenMayaMPx.MPxCommand, LoggerMixin):
     def __init__(self):
         OpenMayaMPx.MPxCommand.__init__(self)
         LoggerMixin.__init__(self)
+        self._exportSet = "UnityFbxExportSet"
 
     def __del__(self):
         LoggerMixin.__del__(self)
@@ -84,9 +85,28 @@ def scriptCmd(cls):
         return
 
     def doIt(self, args):
+        # Gather everything that is in the scene
+        origItemsInScene = maya.cmds.ls(tr=True, o=True, r=True)
+        
         strCmd = 'Import'
         self.displayDebug('doIt {0}'.format(strCmd))
-        maya.mel.eval(strCmd)
+        result = maya.cmds.Import()
+        
+        # figure out what has been added after import
+        itemsInScene = maya.cmds.ls(tr=True, o=True, r=True)
+        newItems = list(set(itemsInScene) - set(origItemsInScene))
+        
+        # Get or create the Unity Fbx Export Set
+        allSets = maya.cmds.listSets(allSets=True)
+        if self._exportSet in allSets:
+            if maya.cmds.sets(self._exportSet, size=True, q=True) > 0:
+                self.displayDebug('Set {0} is not empty, cannot update'.format(self._exportSet))
+                return
+        else:
+            # couldn't find export set so create it
+            maya.cmds.sets(name=self._exportSet)
+        
+        maya.cmds.sets(newItems, add=self._exportSet)    
 
     @classmethod
     def invoke(cls):