Das deutsche Python-Forum

Code: Alles auswählen

class MyToolBar(wx.ToolBar):
    """ a toolbar with dictionary-like access to its items
    
        Use toolbar["label"] or toolbar[id] to get the item.
        
        To add an item:
        toolbar["label"] = ToolBarToolBase(...) to add an item per label,
        toolbar[id] = ToolBarToolBase(...) to add an item per id...
        You can get an item per ID even if you add it per label and vice versa!
        
        To delete an item simply do
        del toolbar["label"]
        
        You can check the Bar for tools with constructs like
        if "label" in toolbar: ...
        
        Also the normal ToolBar.AddLabelTool(...) way is possible.
        Other add-methods are not implemented, yet!
        
        In methods like "DeleteTool" you can use the label or the id of an item.
    """
    def __init__(self, parent, *args, **kwargs):
        self._itemdict = dict()
        wx.ToolBar.__init__(self, parent, *args, **kwargs)
        
    def __getitem__(self, label_or_id):
        if type(label_or_id) is int:
            for item in self._itemdict.iteritems:
                if item.GetId() == label_or_id: return item
        else:
            return self._itemdict[label_or_id]
        
    def __setitem__(self, tool):
        id = tool.GetId()
        self._itemdict[tool.GetLabel()] = self._itemlist[id]
        self.AddToolItem(tool)
        
    def __delitem__(self, label_or_id): self.DeleteTool(label_or_id)
        
    def __contains__(self, label): return label in self._itemdict
        
    def AddLabelTool(self, id, label, bitmap, bmpDisabled=wx.NullBitmap, kind=0, shortHelp='', longHelp='', clientData=None):
        toolbase = wx.ToolBar.AddLabelTool(self, id, label, bitmap, bmpDisabled, kind, shortHelp, longHelp, clientData)
        self._itemdict[label] = toolbase
        
    def DeleteTool(self, label_or_id):
        if type(label_or_id) is int:
            for item in self._itemdict.itervalues():
                if item.GetId() == label_or_id: tool = item
        else:
            tool = self._itemdict[label_or_id]
        del self._itemdict[tool.GetLabel()]
        wx.ToolBar.DeleteTool(self, tool.GetId())