Пример #1
0
class ExplorerFrame(xrcExplorerFrame):
    """A frame to explorer all objects of a project, both data project and type project.

    At most 2 instances of ExplorerFrame might exists, while only one is enabled  at the same time.  
    """
    
    def __init__(self, parent, project=None):
        self._project = None            
        # [EditorFrame, ...]
        self._editor_frames = []   
        
        # {EditorFrame: wxid}  
        self._editor_to_menuid = {}
        
        # [fs_nodes, ...]
        self._path_history_pos = 0      
        self._path_history = []    
        
        # tool ids
        self._create_obj_tool_ids = [] 
             
        super(ExplorerFrame, self).__init__(parent)

        # icon
        icon = wx.EmptyIcon()
        icon_name = 'icons/setting.png' if (project and project.is_type_editor) else 'icons/table.png'
        icon.CopyFromBitmap(get_bitmap(icon_name, (64, 64), project))
        self.SetIcon(icon)
        
        # Replace tool icons
        self._initialized_toolbar()

        # XRCed can't assign variable for wxMenu correctly        
        self.menu_windows = self.menu_close_all.GetMenu()
        self.menu_new = self.menu_new_dummy.GetMenu()
        self.menu_new.Remove(self.menu_new.FindItemByPosition(0).GetId())
        
        self.menu_plugins = self.menu_plugin_dummy.GetMenu()
        self.menu_plugins.Remove(self.menu_plugins.FindItemByPosition(0).GetId())
        
        self.SetMinSize((640, 480))        
        self._create_tree()
        self._create_list()
        # self._create_status_view()
        # self._create_log_view()
        self._layout()
        
        self.Bind(wx.EVT_CHAR_HOOK, self._on_char_hook)
          
        if project:
            self.update_project(project)

    # def ProcessEvent(self, event):
    #     a = xrcExplorerFrame.ProcessEvent(self, event)
    #     b = event.GetSkipped()
    #     print 'ProcessEvent:', event, a, b
    #     return a

    @property
    def list(self):
        return self._list

    def _initialized_toolbar(self):
        project = self.project
        tbsize = self.tool_bar.GetToolBitmapSize()
        size = (tbsize.width, tbsize.height)
        self.tool_bar.SetToolNormalBitmap(self.tool_open.GetId(), get_bitmap('icons/open.png', size, project))
        self.tool_bar.SetToolNormalBitmap(self.tool_setting.GetId(), get_bitmap('icons/setting.png', size, project))
        # self.tool_bar.SetToolNormalBitmap(self.tool_repair.GetId(), get_bitmap('icons/repair.png', size, project))
        self.tool_bar.SetToolNormalBitmap(self.tool_export.GetId(), get_bitmap('icons/export.png', size, project))

        # self.tool_bar.SetToolNormalBitmap(self.tool_create_folder.GetId(), get_bitmap(ICON_FOLDER, size, project))
        # self.tool_bar.SetToolNormalBitmap(self.tool_create_filter.GetId(), get_bitmap(ICON_FILTER, size, project))

        self.Bind(wx.EVT_SIZE, self._on_size_changed)

    def _on_size_changed(self, evt):

        # adjust the width of address_box
        tb_size = self.tool_bar.GetSize()
        ab_size = self.address_box.GetSize()
        ab_pos = self.address_box.GetPosition()
        self.address_box.SetSize((tb_size[0] - ab_pos[0], ab_size[1]))
        # evt.Veto()

    def _create_tree(self):
        self._tree_panel = scrolled.ScrolledPanel(self.main_panel, -1,
                                                  style=wx.TAB_TRAVERSAL | wx.SUNKEN_BORDER, name="TreePanel")
        
        self._tree = ExplorerTree(self._tree_panel, self)  
              
        sizer = wx.BoxSizer(wx.VERTICAL)
        sizer.Add(self._tree, 1, wx.EXPAND)
        self._tree_panel.SetSizer(sizer)
        
        self._tree_panel.SetAutoLayout(1)
        self._tree_panel.SetupScrolling()
        self._tree_panel.SetInitialSize()        
    
    def _create_list(self):        
        self._list_panel = scrolled.ScrolledPanel(self.main_panel, -1,
                                                  style=wx.TAB_TRAVERSAL | wx.SUNKEN_BORDER, name="ListPanel")
        
        self._list = ExplorerList(self._list_panel, self)

        tb = xrcListToolbar(self._list_panel)
        tb.list = self._list
        tbsize = tb.GetToolBitmapSize()
        tb.SetToolNormalBitmap(tb.sort_by_type.GetId(), get_bitmap('icons/class.png', tbsize, self.project))
        tb.SetToolNormalBitmap(tb.sort_by_time.GetId(), get_bitmap('icons/time.png', tbsize, self.project))
        tb.SetToolNormalBitmap(tb.sort_by_name.GetId(), get_bitmap('icons/name.png', tbsize, self.project))
        tb.SetToolNormalBitmap(tb.list_icon.GetId(), get_bitmap('icons/icon_view.png', tbsize, self.project))

        if self._list.list_mode == LIST_MODE_ICON:
            tb.ToggleTool(tb.list_icon.GetId(), True)

        sizer = wx.BoxSizer(wx.VERTICAL)
        sizer.Add(tb, 0, wx.EXPAND)
        sizer.Add(self._list, 1, wx.EXPAND)
        self._list_panel.SetSizer(sizer)
        
        self._list_panel.SetAutoLayout(1)
        self._list_panel.SetupScrolling()
        self._list_panel.SetInitialSize()
        
    def _create_status_view(self):
        pass
    
    def _create_log_view(self):
        self._log = wx.TextCtrl(self.main_panel, -1,
                                style=wx.TE_MULTILINE | wx.TE_READONLY | wx.HSCROLL)
        wx.Log_SetActiveTarget(wx.LogTextCtrl(self._log))        
        pass
    
    def _layout(self):
        self._aui_manager = aui.AuiManager()
        self._aui_manager.SetManagedWindow(self.main_panel)
                
        # Use the aui manager to set up everything
        allow_aui_floating = False
        
#         self.SetToolBar(None)        
#         self._aui_manager.AddPane(self.tool_bar, aui.AuiPaneInfo().
#                       Name("Toolbar").Caption("Toolbar").
#                       ToolbarPane().Top().Floatable(allow_aui_floating).
#                       LeftDockable(False).RightDockable(False))        

        self._aui_manager.AddPane(self._tree_panel,
                                  aui.AuiPaneInfo().
                                  Left().Layer(2).BestSize((240, -1)).
                                  MinSize((160, -1)).
                                  Floatable(allow_aui_floating).FloatingSize((240, 700)).
                                  Caption(u"Object Explorer").
                                  CloseButton(False).
                                  Name("ResourceTree"))
        self._aui_manager.AddPane(self._list_panel, aui.AuiPaneInfo().CenterPane().Name("FileList"))
#         self._aui_manager.AddPane(self._log,
#                          aui.AuiPaneInfo().
#                          Bottom().BestSize((-1, 60)).
#                          MinSize((-1, 60)).Name("Log").Caption("Log") )
        # self._aui_cfgs[DEFAULT_PERSPECTIVE] = self._aui_manager.SavePerspective()
        self._aui_manager.Update()
        # self._aui_manager.SetFlags(self._aui_manager.GetFlags() ^ aui.AUI_MGR_TRANSPARENT_DRAG)
        pass
    
    def _on_char_hook(self, evt):
        keystr = hotkey.build_keystr(evt)
        
        if hotkey.check(hotkey.EXPLORER_UP_LEVEL, keystr):
            self.go_up()
            return
        
        if hotkey.check(hotkey.EXPLORER_HISTORY_PREV, keystr):
            self.go_back()
            return

        if hotkey.check(hotkey.EXPLORER_HISTORY_NEXT, keystr):
            self.go_forward()
            return

        evt.Skip()

    @property
    def project(self):
        return self._project
    
    @property
    def list(self):
        return self._list
    
    @property
    def tree(self):
        return self._tree
    
    @property
    def is_type_editor(self):
        return bool(self._project and self._project.is_type_editor)
    
    def update_project(self, project): 
        # Less elegant but more convenience than ExplorerFrame.get_frame(project)        
        project.explorer = self
        
        self._project = project
        self.SetTitle("%s - [%s]" % (project.name, project.path))
                    
        if not project.is_type_editor and project.get_editor_project().has_error():
            wx.MessageBox("Invalid types, please fix!")
            self.set_status_msg('Can not edit because types are invalid. Open Settings to fix!')            
        
        self._tree.update_project()
        self._list.update_project()
        
        # Create buttons for Creating clazzes
        for tid in self._create_obj_tool_ids:
            self.tool_bar.DeleteTool(tid)
            
        # tbsize = self.tool_bar.GetToolBitmapSize()
        # size = (tbsize.width, tbsize.height)
        # clazzes = project.type_manager.get_clazzes()
        # clazzes.sort(key=lambda x: x.name)
        # for clazz in clazzes:
        #     tool = self.tool_bar.AddTool(wx.NewId(), get_clazz_bitmap(clazz, size, project),
        # shortHelpString=clazz.name)
        #     self._bind_tool_create(clazz, tool.GetId())
        #     self._create_obj_tool_ids.append(tool.GetId())
        # self.tool_bar.Realize()
                
        self.set_path(project.fs_manager.root)
        
        # Update frame icon
        icon = get_icon('table.png', FRAME_ICON_SIZE, project)
        if icon:
            self.SetIcon(icon)
            
        self.init_plugins()
    
    def init_plugins(self):
        # Remove current plugins
        while 1:
            # noinspection PyBroadException
            try:
                item = self.menu_plugins.FindItemByPosition(0)                
                self.menu_plugins.Remove(item.GetId())
            except:
                break
            
        from structerui.plugin_manager import PluginManager
        pm = PluginManager()      
        pm.load_plugins()      

        def get_callback(plugin_):
            def callback(evt):
                _ = evt
                try:
                    plugin_.execute(self.project)
                except Exception, e:
                    log.error('PluginError: %s', e)
                    wx.MessageBox('PluginError: %s' % e)
            return callback

        # add all        
        for plugin in pm.iter_plugins():            
            mi = wx.MenuItem(self.menu_plugins, wx.NewId(), plugin.label)
            self.menu_plugins.Bind(wx.EVT_MENU, get_callback(plugin), mi)
            self.menu_plugins.AppendItem(mi)
Пример #2
0
class ExplorerFrame(xrcExplorerFrame):
    """A frame to explorer all objects of a project, both data project and type project.

    At most 2 instances of ExplorerFrame might exists, while only one is enabled  at the same time.  
    """
    def __init__(self, parent, project=None):
        self._project = None
        # [EditorFrame, ...]
        self._editor_frames = []

        # {EditorFrame: wxid}
        self._editor_to_menuid = {}

        # [fs_nodes, ...]
        self._path_history_pos = 0
        self._path_history = []

        # tool ids
        self._create_obj_tool_ids = []

        super(ExplorerFrame, self).__init__(parent)

        # icon
        icon = wx.EmptyIcon()
        icon_name = 'icons/setting.png' if (
            project and project.is_type_editor) else 'icons/table.png'
        icon.CopyFromBitmap(get_bitmap(icon_name, (64, 64), project))
        self.SetIcon(icon)

        # Replace tool icons
        self._initialized_toolbar()

        # XRCed can't assign variable for wxMenu correctly
        self.menu_windows = self.menu_close_all.GetMenu()
        self.menu_new = self.menu_new_dummy.GetMenu()
        self.menu_new.Remove(self.menu_new.FindItemByPosition(0).GetId())

        self.menu_plugins = self.menu_plugin_dummy.GetMenu()
        self.menu_plugins.Remove(
            self.menu_plugins.FindItemByPosition(0).GetId())

        self.SetMinSize((640, 480))
        self._create_tree()
        self._create_list()
        # self._create_status_view()
        # self._create_log_view()
        self._layout()

        self.Bind(wx.EVT_CHAR_HOOK, self._on_char_hook)

        if project:
            self.update_project(project)

    # def ProcessEvent(self, event):
    #     a = xrcExplorerFrame.ProcessEvent(self, event)
    #     b = event.GetSkipped()
    #     print 'ProcessEvent:', event, a, b
    #     return a

    @property
    def list(self):
        return self._list

    def _initialized_toolbar(self):
        project = self.project
        tbsize = self.tool_bar.GetToolBitmapSize()
        size = (tbsize.width, tbsize.height)
        self.tool_bar.SetToolNormalBitmap(
            self.tool_open.GetId(), get_bitmap('icons/open.png', size,
                                               project))
        self.tool_bar.SetToolNormalBitmap(
            self.tool_setting.GetId(),
            get_bitmap('icons/setting.png', size, project))
        # self.tool_bar.SetToolNormalBitmap(self.tool_repair.GetId(), get_bitmap('icons/repair.png', size, project))
        self.tool_bar.SetToolNormalBitmap(
            self.tool_export.GetId(),
            get_bitmap('icons/export.png', size, project))

        # self.tool_bar.SetToolNormalBitmap(self.tool_create_folder.GetId(), get_bitmap(ICON_FOLDER, size, project))
        # self.tool_bar.SetToolNormalBitmap(self.tool_create_filter.GetId(), get_bitmap(ICON_FILTER, size, project))

        self.Bind(wx.EVT_SIZE, self._on_size_changed)

    def _on_size_changed(self, evt):

        # adjust the width of address_box
        tb_size = self.tool_bar.GetSize()
        ab_size = self.address_box.GetSize()
        ab_pos = self.address_box.GetPosition()
        self.address_box.SetSize((tb_size[0] - ab_pos[0], ab_size[1]))
        # evt.Veto()

    def _create_tree(self):
        self._tree_panel = scrolled.ScrolledPanel(self.main_panel,
                                                  -1,
                                                  style=wx.TAB_TRAVERSAL
                                                  | wx.SUNKEN_BORDER,
                                                  name="TreePanel")

        self._tree = ExplorerTree(self._tree_panel, self)

        sizer = wx.BoxSizer(wx.VERTICAL)
        sizer.Add(self._tree, 1, wx.EXPAND)
        self._tree_panel.SetSizer(sizer)

        self._tree_panel.SetAutoLayout(1)
        self._tree_panel.SetupScrolling()
        self._tree_panel.SetInitialSize()

    def _create_list(self):
        self._list_panel = scrolled.ScrolledPanel(self.main_panel,
                                                  -1,
                                                  style=wx.TAB_TRAVERSAL
                                                  | wx.SUNKEN_BORDER,
                                                  name="ListPanel")

        self._list = ExplorerList(self._list_panel, self)

        tb = xrcListToolbar(self._list_panel)
        tb.list = self._list
        tbsize = tb.GetToolBitmapSize()
        tb.SetToolNormalBitmap(
            tb.sort_by_type.GetId(),
            get_bitmap('icons/class.png', tbsize, self.project))
        tb.SetToolNormalBitmap(
            tb.sort_by_time.GetId(),
            get_bitmap('icons/time.png', tbsize, self.project))
        tb.SetToolNormalBitmap(
            tb.sort_by_name.GetId(),
            get_bitmap('icons/name.png', tbsize, self.project))
        tb.SetToolNormalBitmap(
            tb.list_icon.GetId(),
            get_bitmap('icons/icon_view.png', tbsize, self.project))

        if self._list.list_mode == LIST_MODE_ICON:
            tb.ToggleTool(tb.list_icon.GetId(), True)

        sizer = wx.BoxSizer(wx.VERTICAL)
        sizer.Add(tb, 0, wx.EXPAND)
        sizer.Add(self._list, 1, wx.EXPAND)
        self._list_panel.SetSizer(sizer)

        self._list_panel.SetAutoLayout(1)
        self._list_panel.SetupScrolling()
        self._list_panel.SetInitialSize()

    def _create_status_view(self):
        pass

    def _create_log_view(self):
        self._log = wx.TextCtrl(self.main_panel,
                                -1,
                                style=wx.TE_MULTILINE | wx.TE_READONLY
                                | wx.HSCROLL)
        wx.Log_SetActiveTarget(wx.LogTextCtrl(self._log))
        pass

    def _layout(self):
        self._aui_manager = aui.AuiManager()
        self._aui_manager.SetManagedWindow(self.main_panel)

        # Use the aui manager to set up everything
        allow_aui_floating = False

        #         self.SetToolBar(None)
        #         self._aui_manager.AddPane(self.tool_bar, aui.AuiPaneInfo().
        #                       Name("Toolbar").Caption("Toolbar").
        #                       ToolbarPane().Top().Floatable(allow_aui_floating).
        #                       LeftDockable(False).RightDockable(False))

        self._aui_manager.AddPane(
            self._tree_panel,
            aui.AuiPaneInfo().Left().Layer(2).BestSize((240, -1)).MinSize(
                (160, -1)).Floatable(allow_aui_floating).FloatingSize(
                    (240, 700)).Caption(u"Object Explorer").CloseButton(
                        False).Name("ResourceTree"))
        self._aui_manager.AddPane(
            self._list_panel,
            aui.AuiPaneInfo().CenterPane().Name("FileList"))
        #         self._aui_manager.AddPane(self._log,
        #                          aui.AuiPaneInfo().
        #                          Bottom().BestSize((-1, 60)).
        #                          MinSize((-1, 60)).Name("Log").Caption("Log") )
        # self._aui_cfgs[DEFAULT_PERSPECTIVE] = self._aui_manager.SavePerspective()
        self._aui_manager.Update()
        # self._aui_manager.SetFlags(self._aui_manager.GetFlags() ^ aui.AUI_MGR_TRANSPARENT_DRAG)
        pass

    def _on_char_hook(self, evt):
        keystr = hotkey.build_keystr(evt)

        if hotkey.check(hotkey.EXPLORER_UP_LEVEL, keystr):
            self.go_up()
            return

        if hotkey.check(hotkey.EXPLORER_HISTORY_PREV, keystr):
            self.go_back()
            return

        if hotkey.check(hotkey.EXPLORER_HISTORY_NEXT, keystr):
            self.go_forward()
            return

        evt.Skip()

    @property
    def project(self):
        return self._project

    @property
    def list(self):
        return self._list

    @property
    def tree(self):
        return self._tree

    @property
    def is_type_editor(self):
        return bool(self._project and self._project.is_type_editor)

    def update_project(self, project):
        # Less elegant but more convenience than ExplorerFrame.get_frame(project)
        project.explorer = self

        self._project = project
        self.SetTitle("%s - [%s]" % (project.name, project.path))

        if not project.is_type_editor and project.get_editor_project(
        ).has_error():
            wx.MessageBox("Invalid types, please fix!")
            self.set_status_msg(
                'Can not edit because types are invalid. Open Settings to fix!'
            )

        self._tree.update_project()
        self._list.update_project()

        # Create buttons for Creating clazzes
        for tid in self._create_obj_tool_ids:
            self.tool_bar.DeleteTool(tid)

        # tbsize = self.tool_bar.GetToolBitmapSize()
        # size = (tbsize.width, tbsize.height)
        # clazzes = project.type_manager.get_clazzes()
        # clazzes.sort(key=lambda x: x.name)
        # for clazz in clazzes:
        #     tool = self.tool_bar.AddTool(wx.NewId(), get_clazz_bitmap(clazz, size, project),
        # shortHelpString=clazz.name)
        #     self._bind_tool_create(clazz, tool.GetId())
        #     self._create_obj_tool_ids.append(tool.GetId())
        # self.tool_bar.Realize()

        self.set_path(project.fs_manager.root)

        # Update frame icon
        icon = get_icon('table.png', FRAME_ICON_SIZE, project)
        if icon:
            self.SetIcon(icon)

        self.init_plugins()

    def init_plugins(self):
        # Remove current plugins
        while 1:
            # noinspection PyBroadException
            try:
                item = self.menu_plugins.FindItemByPosition(0)
                self.menu_plugins.Remove(item.GetId())
            except:
                break

        from structerui.plugin_manager import PluginManager
        pm = PluginManager()
        pm.load_plugins()

        def get_callback(plugin_):
            def callback(evt):
                _ = evt
                try:
                    plugin_.execute(self.project)
                except Exception, e:
                    log.error('PluginError: %s', e)
                    wx.MessageBox('PluginError: %s' % e)

            return callback

        # add all
        for plugin in pm.iter_plugins():
            mi = wx.MenuItem(self.menu_plugins, wx.NewId(), plugin.label)
            self.menu_plugins.Bind(wx.EVT_MENU, get_callback(plugin), mi)
            self.menu_plugins.AppendItem(mi)