WxPython跨平台开发框架之动态菜单的管理和功能权限的控制

class ToolbarUtil:
    """工具栏菜单的创建类"""

    @staticmethod
    def create_tools():
        """创建工具栏菜单嵌套集合"""
        menus = [
            MenuInfo(
                id="01",
                label="用户管理",
                icon="user",
                path="views.frm_user.FrmUser",
            ),
            MenuInfo(
                id="02",
                label="组织机构管理",
                icon="organ",
                path="views.frm_ou.FrmOU",
            ),
            ...............
        ]
        return menus

    @staticmethod
    def create_tools_dynamic():
        """动态创建工具栏菜单"""

        # 同步获取菜单信息
        menus = api_menu.get_all_nodes_by_user_sync(
            settings.CurrentUser.id, settings.SystemType
        )
        # 定义字段映射
        field_mapping = {"name": "label", "winformtype": "path", "icon": "icon"}
        newmenus = map_with_dynamic_alias(menus, MenuInfo, field_mapping)

        return newmenus

    def _create_toolbar(self, d="H"):
        """创建工具栏"""
        agwStyle = aui.AUI_TB_TEXT | aui.AUI_TB_OVERFLOW

        if d.upper() in ["V", "VERTICAL"]:
            agwStyle = aui.AUI_TB_TEXT | aui.AUI_TB_VERTICAL

        tb = aui.AuiToolBar(
            self, -1, wx.DefaultPosition, wx.DefaultSize, agwStyle=agwStyle
        )
        tb.SetToolBitmapSize(wx.Size(16, 16))

        # 动态创建工具栏
        toolbars = ToolbarUtil.create_tools_dynamic()

        for item in toolbars:
            tool_id = wx.NewIdRef()
            bitmap = get_bitmap(item.icon)
            tb.AddSimpleTool(
                tool_id=tool_id,
                label=item.label,
                bitmap=bitmap,
                short_help_string=item.tips if item.tips else item.label,
            )

            # 绑定事件
            self.Bind(
                wx.EVT_TOOL,
                partial(self.on_tool_event, item),  # 这里传递菜单信息
                id=tool_id,
            )

        # 增加系统常用按钮
        tb.AddSeparator()
        tb.AddSimpleTool(
            self.id_close_all, "关闭所有", images.delete_all.Bitmap, "关闭所有页面"
        )
        tb.AddSimpleTool(self.id_quit, "退出", images.close.Bitmap, "退出程序")

        self.Bind(
            wx.EVT_TOOL, lambda event: EventPub.show_about_dialog(), id=self.id_about
        )
        self.Bind(
            wx.EVT_TOOL, lambda event: EventPub.close_all_page(), id=self.id_close_all
        )
        tb.Realize()
        return tb

import importlib
import wx

"""要自动导入一个给定的类（比如 "views.testaui_panel.DocumentPanel"）并在 Python 中使用它，
可以利用 importlib 库来动态地导入模块和类。importlib 允许你在运行时加载模块，
而不需要在代码中显式地使用 import 语句。"""

def dynamic_import(class_path: str):
    # 拆分字符串为模块路径和类名
    # print(f"Dynamic import: {class_path}")
    module_path, class_name = class_path.rsplit(".", 1)

    # 动态导入模块
    module = importlib.import_module(module_path)

    # 获取类对象
    class_obj = getattr(module, class_name)

    # 返回类对象
    return class_obj

def show_window(class_path: str, parent=None, **kwargs):
    """根据路径和基类对象，显示窗口"""
    window_class = dynamic_import(class_path)
    window = window_class(parent, **kwargs)
    # 判断传入的窗口是 wx.Dialog 还是 wx.Panel
    if isinstance(window, wx.Dialog):
        # 如果是 wx.Dialog，调用 ShowModal
        result = window.ShowModal()
        print(f"Dialog closed with result: {result}")
        window.Destroy()  # 在对话框关闭后销毁
    elif isinstance(window, wx.Panel):
        # 如果是 wx.Panel，调用 Show
        window.Show()
        print("Panel shown")
    else:
        print("Unknown window type")

    async def SetLoginInfo(self):
        """设置登录信息"""
        userid = settings.AccessTokenResult.userid
        res = await api_user.Get(userid)
        if res.success:
            settings.CurrentUser = res.result
            await self.GetSystemType()
            await self.GetFunctionsByUser(userid)
            await self.GetRolesByUser(userid)

            EventPub.user_info_loaded()  # 发布用户信息加载完成事件

    def HasAction(self, action_id: str, entity_name: str = None) -> bool:
        """判断权限缓存是否存在指定操作功能id

        :param action_id: 操作功能id, 如Add,Edit,Delete, Import,Export等
        :param entity_name: 实体名称, 默认根据self.model的名称获取并替换后缀Dto，如Customer,Product等
        :return: 存在返回True，否则返回False
        """
        if not action_id:
            return False

        if not entity_name:
            # 根据self.model的名称获取并替换后缀Dto
            entity_name = self.model.__qualname__.replace("Dto", "")

        # 判断权限缓存是否存在类似"Customer:Add"格式的字符串
        result = f"{entity_name}:{action_id}" in settings.FunctionDict
        return result

    def _CreateCommonButtons(self, pane: wx.Window):
        """父类窗口统一创建通用按钮"""
        # 增加按钮
        btns_sizer = wx.BoxSizer(wx.HORIZONTAL)

        button_list = []
        # 设置图标和位置
        if EventFlags.EVT_Search & self.EVT_FLAGS:
            btn_search = ControlUtil.create_button(
                pane,
                "查询",
                "search",
                handler=self._on_first_page,
                is_async=True,
                id=wx.ID_FIND,
            )
            button_list.append(btn_search)
        if self.has_add:
            btn_add = ControlUtil.create_button(
                pane, "新增", "add", handler=self.OnAdd, is_async=True
            )
            button_list.append(btn_add)
        if self.has_edit:
            btn_edit = ControlUtil.create_button(
                pane, "编辑", "edit", handler=self.OnEdit, is_async=True
            )
            button_list.append(btn_edit)
        if self.has_delete:
            btn_delete = ControlUtil.create_button(
                pane, "批量删除", "delete", handler=self.OnDelete, is_async=True
            )
            button_list.append(btn_delete)
        if self.has_export:
            btn_export = ControlUtil.create_button(
                pane, "导出Excel", "xls", handler=self.OnExport, is_async=True
            )
            button_list.append(btn_export)

        # 将按钮添加到按钮组中
        for btn in button_list:
            btns_sizer.Add(btn, 0, wx.ALL, 3)

        return btns_sizer

    def _on_showmenu(self, event: wx.grid.GridEvent) -> None:
        """父类窗体的右键菜单处理"""

        # 创建右键菜单对象
        menu: wx.Menu = wx.Menu()

        if self.has_add:
            ControlUtil.create_menu(
                self, menu, "新增", icon_name="add", handler=self._OnMenuAdd
            )
        if self.has_edit:
            ControlUtil.create_menu(
                self, menu, "编辑", icon_name="edit", handler=self._OnMenuEdit
            )
        if self.has_delete:
            ControlUtil.create_menu(
                self, menu, "删除选中行", icon_name="delete", handler=self._OnMenuDelete
            )

        if self.has_export:
            ControlUtil.create_menu(
                self, menu, "导出Excel", icon_name="xls", handler=self._OnMenuExport
            )

          ............

    def CreateCustomMenus(self, parent_menu: wx.Menu) -> None:
        """子类重写该方法，创建自定义菜单"""
        # 父类已创建默认菜单，这里添加自定义菜单
        if self.has_batch_add:
            ControlUtil.create_menu(
                self,
                parent_menu,
                "批量添加字典",
                "batch_add",
                handler=self.OnMenuBatchAdd,
            )