import wx
import wx.aui

class MyFrame(wx.Frame):
    def __init__(self, *args, **kw):
        super().__init__(*args, **kw)

        # 创建 AUI 管理器
        self.mgr = wx.aui.AuiManager(self)

        # 创建 AuiToolBar
        toolbar = wx.aui.AuiToolBar(self, style=wx.aui.AUI_TB_DEFAULT_STYLE | wx.aui.AUI_TB_OVERFLOW)

        # 添加工具按钮
        open_bmp = wx.ArtProvider.GetBitmap(wx.ART_FILE_OPEN, wx.ART_TOOLBAR)
        save_bmp = wx.ArtProvider.GetBitmap(wx.ART_FILE_SAVE, wx.ART_TOOLBAR)
        exit_bmp = wx.ArtProvider.GetBitmap(wx.ART_QUIT, wx.ART_TOOLBAR)

        toolbar.AddTool(wx.ID_OPEN, "Open", open_bmp, short_help="Open File")
        toolbar.AddTool(wx.ID_SAVE, "Save", save_bmp, short_help="Save File")
        toolbar.AddSeparator()  # 添加分隔符
        toolbar.AddTool(wx.ID_EXIT, "Exit", exit_bmp, short_help="Exit Application")

        # 完成工具栏布局
        toolbar.Realize()

        # 将工具栏添加到 AUI 管理器
        self.mgr.AddPane(toolbar, wx.aui.AuiPaneInfo().Name("Toolbar").Top().Dockable(True))

        # 绑定事件
        self.Bind(wx.EVT_TOOL, self.on_open, id=wx.ID_OPEN)
        self.Bind(wx.EVT_TOOL, self.on_save, id=wx.ID_SAVE)
        self.Bind(wx.EVT_TOOL, self.on_exit, id=wx.ID_EXIT)

        # 设置窗口
        self.SetSize((600, 400))
        self.Centre()
        self.mgr.Update()  # 更新 AUI 管理器

    def on_open(self, event):
        wx.MessageBox("Open clicked", "Info", wx.OK | wx.ICON_INFORMATION)

    def on_save(self, event):
        wx.MessageBox("Save clicked", "Info", wx.OK | wx.ICON_INFORMATION)

    def on_exit(self, event):
        self.Close()

    def OnClose(self, event):
        self.mgr.UnInit()  # 清理 AUI 管理器
        self.Destroy()

class MyApp(wx.App):
    def OnInit(self):
        frame = MyFrame(None, title="wxPython AuiToolBar Example")
        frame.Show()
        return True

if __name__ == "__main__":
    app = MyApp()
    app.MainLoop()

pane_info = wx.aui.AuiPaneInfo().Name("Toolbar").Top().Caption("Main Toolbar").Floatable(True)
self.mgr.AddPane(toolbar, pane_info)

@dataclass
class MenuInfo:
    id: str  # 菜单ID
    pid: str  # 父菜单ID
    label: str  # 菜单名称
    icon: str = None  # 菜单图标
    path: str = None  # 菜单路径，用来定位视图
    tips: str = None  # 菜单提示
    children: list["MenuInfo"] = None

MenuInfo(
    id="01",
    label="用户管理",
    icon="user",
    path="views.testaui_panel.DocumentPanel",
),

MenuInfo(
    id="11",
    label="权限管理系统",
    icon="computer_key",
    children=[
        MenuInfo(
            id="11-1",
            label="用户管理",
            icon="user",
            path="views.testaui_panel.DocumentPanel",
        ),
        MenuInfo(
            id="11-2",
            label="组织机构管理",
            icon="organ",
            path="views.my_banner.BannerDialog2",
        ),
        ...
    ]
}

def create_toolbars(self):
    """创建工具栏"""
    self.tb1 = self._create_toolbar()  # 工具栏

    self.mgr.AddPane(
        self.tb1,
        aui.AuiPaneInfo()
        .Name("ToolBar1")
        .Caption("工具条")
        .ToolbarPane()
        .Top()
        .Row(0)
        .Position(0)
        .Floatable(False),
    )

# 初始化AuiManager，用来管理工具栏以及AuiNotebook等
self.mgr = aui.AuiManager(self)

toolbars = ToolbarUtil.create_tools() #遍历工具栏进行处理
for item in toolbars:
    tool_id = wx.NewIdRef()
    help_string = item.tips if item.tips else item.label
    bitmap = get_bitmap(item.icon)
 tb.AddSimpleTool( tool_id=tool_id, label=item.label, bitmap=bitmap, short_help_string=help_string, ) # 绑定事件
    self.Bind(
        wx.EVT_TOOL,
        partial(self.on_tool_event, item),  # 这里传递菜单信息
        id=tool_id,
    )

self.Bind( wx.EVT_TOOL, lambda event: EventPub.toggle_left_panel(),id=self.id_show_hide_left)
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

from pubsub import pub
from entity.common import MenuInfo

# 定义事件处理类，统一处理事件发布接口
class EventPub:
    """事件发布者"""

    @staticmethod
    def send_event(event_name: str, data=None):
        """发送事件"""
        pub.sendMessage(event_name, data=data)

    @staticmethod
    def show_window(data: MenuInfo, hide_toolbox=True):
        """打开窗口"""
        pub.sendMessage("show_window", info=data, hide_toolbox=hide_toolbox)

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