Merge pull request #95 from Unity-Technologies/UNI-22019-scene-export-set

Uni 22019 working scene remembers what to publish

Author: benoithudson

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,24 @@ 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 not self._exportSet in allSets:
+            # 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):