Uni-24798-maya-menuitem-icons

Author: inwoods

Assets/Integrations/Autodesk/maya/icons/unity_100.png

@@ -38,6 +38,8 @@ class BaseCommand(OpenMayaMPx.MPxCommand, LoggerMixin):
     """
     Base class for UnityOneClick Plugin Commands.
     """
+    kDefaultIcon = 'unity_100.png'
+    
     def __init__(self):
         OpenMayaMPx.MPxCommand.__init__(self)
         LoggerMixin.__init__(self)
@@ -86,6 +88,14 @@ def loadUnityFbxExportSettings(self):
 
         maya.mel.eval(contents)
         return True
+    
+    @classmethod
+    def iconPath(cls):
+        return ""
+    
+    @classmethod
+    def familyIconPath(cls):
+        return maya.mel.eval('$tempPath = languageResourcePath("{0}");'.format(cls.kDefaultIcon))
 
     def storeAttribute(self, node, attr, attrValue, attrType="string"):
         if not maya.mel.eval('attributeExists "{0}" "{1}"'.format(attr, node)):

Assets/Integrations/Autodesk/maya/icons/unity_150.png

@@ -56,10 +56,30 @@ def installMenu():
     install menu into main window 
     @ingroup UnityUI
     """
-    maya.cmds.menu (kMainWndMenuName, parent='MayaWindow', label=kMainWndMenuLabel, tearOff=True, version=kMayaVersionAdded) # @UndefinedVariable
-    maya.cmds.menuItem(parent=kMainWndMenuName, label=commands.importCmd.kShortLabel, command=commands.importCmd.kScriptCommand, version=kMayaVersionAdded)        # @UndefinedVariable
-    maya.cmds.menuItem(parent=kMainWndMenuName, label=commands.reviewCmd.kShortLabel, command=commands.reviewCmd.kScriptCommand, version=kMayaVersionAdded)    # @UndefinedVariable
-    maya.cmds.menuItem(parent=kMainWndMenuName, label=commands.publishCmd.kShortLabel, command=commands.publishCmd.kScriptCommand, version=kMayaVersionAdded)    # @UndefinedVariable
+    maya.cmds.menu(kMainWndMenuName, 
+                   parent='MayaWindow', 
+                   label=kMainWndMenuLabel, tearOff=True, 
+                   version=kMayaVersionAdded, 
+                   familyImage=commands.importCmd.familyIconPath)
+    
+    maya.cmds.menuItem(parent=kMainWndMenuName, 
+                       label=commands.importCmd.kShortLabel, 
+                       annotation=commands.importCmd.kLabel, 
+                       command=commands.importCmd.kScriptCommand,
+                       image=commands.importCmd.iconPath,
+                       version=kMayaVersionAdded)
+    maya.cmds.menuItem(parent=kMainWndMenuName, 
+                       label=commands.reviewCmd.kShortLabel, 
+                       annotation=commands.reviewCmd.kLabel, 
+                       command=commands.reviewCmd.kScriptCommand, 
+                       image=commands.importCmd.iconPath,
+                       version=kMayaVersionAdded)
+    maya.cmds.menuItem(parent=kMainWndMenuName, 
+                       label=commands.publishCmd.kShortLabel, 
+                       annotation=commands.publishCmd.kLabel, 
+                       command=commands.publishCmd.kScriptCommand, 
+                       image=commands.importCmd.iconPath,
+                       version=kMayaVersionAdded)
 
 def uninstallMenu():
     """