Merge pull request #137 from Unity-Technologies/UNI-24847-restore-selection-after-publish

Uni 24847 restore selection after publish

Author: inwoods

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

@@ -280,6 +280,7 @@ def doIt(self, args):
 
         # select the export set for export, if it exists,
         # otherwise take what is currently selected
+        origSelection = maya.cmds.ls(sl=True)
         if self.setExists(self._exportSet):
             maya.cmds.select(self._exportSet, r=True, ne=True)
 
@@ -311,6 +312,10 @@ def doIt(self, args):
         self.displayDebug('doIt({0})'.format(melCommand))
 
         maya.mel.eval(melCommand)
+        
+        if origSelection:
+            maya.cmds.select(cl=True)
+            maya.cmds.select(origSelection, add=True, ne=True)
 
     @classmethod
     def invoke(cls):
@@ -360,6 +365,7 @@ def doIt(self, args):
 
         # select the export set for export, if it exists,
         # otherwise take what is currently selected
+        origSelection = maya.cmds.ls(sl=True)
         if self.setExists(self._exportSet):
             maya.cmds.select(self._exportSet, r=True, ne=True)
 
@@ -373,6 +379,10 @@ def doIt(self, args):
         self.displayDebug('doIt {0}'.format(strCmd))
         maya.mel.eval(strCmd)
 
+        if origSelection:
+            maya.cmds.select(cl=True)
+            maya.cmds.select(origSelection, add=True, ne=True)
+        
     @classmethod
     def invoke(cls):
         """