Uni-24797-file-unity

Author: inwoods

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

@@ -29,8 +29,10 @@
 # User Interface
 # ======================================================================'
 
-kMainWndMenuName = 'UnityOneClick'
-kMainWndMenuLabel = 'Unity'
+kMenuName = 'UnityOneClick'
+kMenuDivider = 'UnityOneClickDivider'
+kMenuLabel = 'UNITY'
+kMenuInsertAfter = 'exportActiveFileOptions'
 
 def register(pluginFn):
     """
@@ -50,20 +52,31 @@ def unregister(pluginFn):
 
     return
 
+def getParentMenu():
+    result = maya.mel.eval('$tempVar = $gMainFileMenu;')
+    maya.mel.eval("buildFileMenu")
+    return result
+
 def installMenu():
     """
     install menu into main window 
     @ingroup UnityUI
     """
-    maya.cmds.menu (kMainWndMenuName, parent='MayaWindow', label=kMainWndMenuLabel, tearOff=True) # @UndefinedVariable
-    maya.cmds.menuItem(parent=kMainWndMenuName, label=commands.importCmd.kShortLabel, command=commands.importCmd.kScriptCommand)        # @UndefinedVariable
-    maya.cmds.menuItem(parent=kMainWndMenuName, label=commands.reviewCmd.kShortLabel, command=commands.reviewCmd.kScriptCommand)    # @UndefinedVariable
-    maya.cmds.menuItem(parent=kMainWndMenuName, label=commands.publishCmd.kShortLabel, command=commands.publishCmd.kScriptCommand)    # @UndefinedVariable
+    parentMenu = getParentMenu()
+
+    maya.cmds.menuItem(kMenuDivider, divider=True, longDivider=False, insertAfter=kMenuInsertAfter, parent=parentMenu)
+    maya.cmds.menuItem(kMenuName, parent=parentMenu, insertAfter=kMenuDivider, subMenu=True, label=kMenuLabel, tearOff=True)
+
+    maya.cmds.menuItem(parent=kMenuName, label=commands.importCmd.kShortLabel, command=commands.importCmd.kScriptCommand)        # @UndefinedVariable
+    maya.cmds.menuItem(parent=kMenuName, label=commands.reviewCmd.kShortLabel, command=commands.reviewCmd.kScriptCommand)    # @UndefinedVariable
+    maya.cmds.menuItem(parent=kMenuName, label=commands.publishCmd.kShortLabel, command=commands.publishCmd.kScriptCommand)    # @UndefinedVariable
 
 def uninstallMenu():
     """
     uninstall the unityOneClick menu from main window
     @ingroup UnityUI
     """
-    if maya.cmds.menu(kMainWndMenuName, exists=True):     # @UndefinedVariable
-        maya.cmds.deleteUI(kMainWndMenuName, menu=True)   # @UndefinedVariable
+    if maya.cmds.menu(kMenuName, exists=True):     # @UndefinedVariable
+        maya.cmds.deleteUI(kMenuDivider, menuItem=True)
+        maya.cmds.deleteUI(kMenuName, menuItem=True)
+        maya.cmds.deleteUI(kMenuName, menu=True)   # @UndefinedVariable