set Unity project as workspace on Import

switch back to prev workspace once import complete

Author: vkovec

Assets/Integrations/Autodesk/maya2017/scripts/configureUnityOneClick.mel

@@ -8,9 +8,6 @@ global proc configureUnityOneClick(
     optionVar -stringValue "UnityTempSavePath" $unityTempSavePath;
     optionVar -stringValue "UnityFbxExportSettings" $unityFbxExportSettings;
     optionVar -intValue "UnityOneClick_Headless" $headless;
-    
-    // set project as the current workspace
-    workspace -o $unityProject;
 
     // configure auto load of plugin
     if (`pluginInfo -q -loaded unityOneClickPlugin` == false)

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

@@ -177,6 +177,12 @@ def afterImport(self, *args, **kwargs):
 
     def doIt(self, args):
         self.loadDependencies()
+        
+        # set Unity project as the current workspace
+        currWorkspace = maya.cmds.workspace(o=True, q=True)
+        unityProject = maya.cmds.optionVar(q='UnityProject')
+        if unityProject:
+            maya.cmds.workspace(unityProject, o=True)
 
         self._tempPath = None
         self._tempName = None
@@ -191,6 +197,9 @@ def doIt(self, args):
 
         OpenMaya.MMessage.removeCallback(callbackId)
         OpenMaya.MMessage.removeCallback(callbackId2)
+        
+        if currWorkspace:
+            maya.cmds.workspace(currWorkspace, o=True)
 
     @classmethod
     def invoke(cls):