Skip to content
Snippets Groups Projects
Commit 8a799b00 authored by Luca Rood's avatar Luca Rood
Browse files

Fix T100423: Addon's custom context menu entries get overridden by other addons

This introduces a new `UI_MT_button_context_menu` class which is
registered at startup. Addons can append/prepend draw functions to this
class, in order to add their custom context menu entries.

The new class replaces the old `WM_MT_button_context` class, thus
requiring a small change in addons using this feature. This is done
because addons were previously required to register the class
themselves, which caused addons to override each other's context menu
entries.

Now the class registration is handled by Blender, and addons need only
append their draw functions. The new class name ensures that addons
using the old method don't override menu entries made using the new
class.

Menu entries added with the legacy `WM_MT_button_context` class are
still drawn for backwards compatibility, but this class must not be used
going forward, as any addon using it still runs the risk of having its
menu entries overridden, and support for the legacy class is subject to
removal in a future version.

Reviewed By: campbellbarton

Maniphest Tasks: T100423

Differential Revision: https://developer.blender.org/D15702
parent c3757504
No related branches found
No related tags found
No related merge requests found
...@@ -3,8 +3,8 @@ Extending the Button Context Menu ...@@ -3,8 +3,8 @@ Extending the Button Context Menu
+++++++++++++++++++++++++++++++++ +++++++++++++++++++++++++++++++++
This example enables you to insert your own menu entry into the common This example enables you to insert your own menu entry into the common
right click menu that you get while hovering over a value field, right click menu that you get while hovering over a UI button (e.g. operator,
color, string, etc. value field, color, string, etc.)
To make the example work, you have to first select an object To make the example work, you have to first select an object
then right click on an user interface element (maybe a color in the then right click on an user interface element (maybe a color in the
...@@ -14,7 +14,6 @@ Executing the operator will then print all values. ...@@ -14,7 +14,6 @@ Executing the operator will then print all values.
""" """
import bpy import bpy
from bpy.types import Menu
def dump(obj, text): def dump(obj, text):
...@@ -47,36 +46,20 @@ class WM_OT_button_context_test(bpy.types.Operator): ...@@ -47,36 +46,20 @@ class WM_OT_button_context_test(bpy.types.Operator):
return {'FINISHED'} return {'FINISHED'}
# This class has to be exactly named like that to insert an entry in the right click menu def draw_menu(self, context):
class WM_MT_button_context(Menu):
bl_label = "Unused"
def draw(self, context):
pass
def menu_func(self, context):
layout = self.layout layout = self.layout
layout.separator() layout.separator()
layout.operator(WM_OT_button_context_test.bl_idname) layout.operator(WM_OT_button_context_test.bl_idname)
classes = (
WM_OT_button_context_test,
WM_MT_button_context,
)
def register(): def register():
for cls in classes: bpy.utils.register_class(WM_OT_button_context_test)
bpy.utils.register_class(cls) bpy.types.UI_MT_button_context_menu.append(draw_menu)
bpy.types.WM_MT_button_context.append(menu_func)
def unregister(): def unregister():
for cls in classes: bpy.types.UI_MT_button_context_menu.remove(draw_menu)
bpy.utils.unregister_class(cls) bpy.utils.unregister_class(WM_OT_button_context_test)
bpy.types.WM_MT_button_context.remove(menu_func)
if __name__ == "__main__": if __name__ == "__main__":
......
Subproject commit a2eb507891449a0b67582be9561840075513661d Subproject commit 4d67fb6e2773619392b3d2099188ae742ef9662a
Subproject commit 7a8502871c34db0343cc7de52d6b49b15a84238a Subproject commit 32baafe44dc56edd1baf6e5a16b4439ded8238f2
Subproject commit 95107484d076bc965239942e857c83433bfa86d7 Subproject commit 42da56aa73726710107031787af5eea186797984
...@@ -239,3 +239,23 @@ class UI_MT_list_item_context_menu(bpy.types.Menu): ...@@ -239,3 +239,23 @@ class UI_MT_list_item_context_menu(bpy.types.Menu):
bpy.utils.register_class(UI_MT_list_item_context_menu) bpy.utils.register_class(UI_MT_list_item_context_menu)
class UI_MT_button_context_menu(bpy.types.Menu):
"""
UI button context menu definition. Scripts can append/prepend this to
add own operators to the context menu. They must check context though, so
their items only draw in a valid context and for the correct buttons.
"""
bl_label = "List Item"
bl_idname = "UI_MT_button_context_menu"
def draw(self, context):
# Draw menu entries created with the legacy `WM_MT_button_context` class.
# This is deprecated, and support will be removed in a future release.
if hasattr(bpy.types, "WM_MT_button_context"):
self.layout.menu_contents("WM_MT_button_context")
bpy.utils.register_class(UI_MT_button_context_menu)
...@@ -1230,7 +1230,7 @@ bool ui_popup_context_menu_for_button(bContext *C, uiBut *but, const wmEvent *ev ...@@ -1230,7 +1230,7 @@ bool ui_popup_context_menu_for_button(bContext *C, uiBut *but, const wmEvent *ev
} }
} }
MenuType *mt = WM_menutype_find("WM_MT_button_context", true); MenuType *mt = WM_menutype_find("UI_MT_button_context_menu", true);
if (mt) { if (mt) {
UI_menutype_draw(C, mt, uiLayoutColumn(layout, false)); UI_menutype_draw(C, mt, uiLayoutColumn(layout, false));
} }
......
...@@ -3028,7 +3028,14 @@ void uiItemMContents(uiLayout *layout, const char *menuname) ...@@ -3028,7 +3028,14 @@ void uiItemMContents(uiLayout *layout, const char *menuname)
if (WM_menutype_poll(C, mt) == false) { if (WM_menutype_poll(C, mt) == false) {
return; return;
} }
bContextStore *previous_ctx = CTX_store_get(C);
UI_menutype_draw(C, mt, layout); UI_menutype_draw(C, mt, layout);
/* Restore context that was cleared by `UI_menutype_draw`. */
if (layout->context) {
CTX_store_set(C, previous_ctx);
}
} }
void uiItemDecoratorR_prop(uiLayout *layout, PointerRNA *ptr, PropertyRNA *prop, int index) void uiItemDecoratorR_prop(uiLayout *layout, PointerRNA *ptr, PropertyRNA *prop, int index)
......
Subproject commit da8bdd7244c7b6c2eadf4c949ff391d0cc430275 Subproject commit 548055f40213c775a6b77025525c91e8466e70d6
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment