# Name: xrced.py # Purpose: XRC editor, main module # Author: Roman Rolinsky # Created: 20.08.2001 # RCS-ID: $Id$ from wxPython.wx import * from wxPython.xrc import * from wxPython.html import * import wxPython.lib.wxpTag from xml.dom import minidom import os import tempfile import images # String constants faceColour = wxSystemSettings_GetSystemColour(wxSYS_COLOUR_3DFACE) # Stange wxHtmlWindow behavior: when bgcolor is exact, it turns white bgcolor = (faceColour.Red()-1, faceColour.Green()-1, faceColour.Blue()-1) htmlHeader = '\n' % bgcolor htmlFooter = '\n' progname = 'XRCed' version = '0.0.5' # Local modules from xxx import * # Globals testWin = None testWinPos = wxDefaultPosition # 1 adds CMD command to Help menu debug = 1 if debug: import traceback import time # Set menu to list items. # Each menu command is a tuple (id, label, help) # submenus are lists [id, label, help, submenu] # and separators are any other type def SetMenu(m, list): for l in list: if type(l) == types.TupleType: apply(m.Append, l) elif type(l) == types.ListType: subMenu = wxMenu() SetMenu(subMenu, l[2:]) m.AppendMenu(wxNewId(), l[0], subMenu, l[1]) else: # separator m.AppendSeparator() ################################################################################ # Properties panel containing notebook class Panel(wxNotebook): def __init__(self, parent, id = -1): wxNotebook.__init__(self, parent, id, style=wxNB_BOTTOM) #self.SetBackgroundColour(wxColour(bgcolor)) sys.modules['params'].panel = self self.page1 = HtmlPage(self) self.AddPage(self.page1, 'Properties') self.page2 = None def SetData(self, xxx): self.page1.SetPageData(xxx) # Replace/remove style page if self.page2: self.RemovePage(1) self.page2.Destroy() if xxx and xxx.treeObject().hasStyle: self.page2 = StylePage(self, xxx.treeObject()) self.AddPage(self.page2, 'Style') else: self.page2 = None def Clear(self): self.page1.Clear() if self.page2: self.RemovePage(1) self.page2.Destroy() self.page2 = None # If some parameter on some page has changed def IsModified(self): return self.page1.IsModified() or self.page2 and self.page2.IsModified() def SetModified(self, value): self.page1.SetModified(value) if self.page2: self.page2.SetModified(value) ################################################################################ # General class for notebook pages class ParamPage: def __init__(self): # Register event handlers for id in paramIDs.values(): EVT_CHECKBOX(self, id, self.OnCheckParams) def OnCheckParams(self, evt): selected = tree.GetSelection() xxx = tree.GetPyData(selected) winName = evt.GetEventObject().GetName() sizerParam = false if winName[0] == '_': param = winName[7:] sizerParam = true else: if xxx.hasChild: xxx = xxx.child param = winName[6:] # Set current object if sizerParam: w = self.FindWindowByName('_data_' + param) else: w = self.FindWindowByName('data_' + param) elem = xxx.element if evt.IsChecked(): # Ad new text node in order of allParams w.SetValue('') # set empty (default) value # For font element, we have to create another object if param == 'font': # Make XXX object textElem = tree.dom.createElement('font') font = xxxFont(xxx, textElem) xxx.params['font'] = font else: textElem = tree.dom.createElement(param) textNode = tree.dom.createTextNode('') textElem.appendChild(textNode) xxx.params[param] = textNode # Find place to put new element: first present element after param found = false paramStyles = xxx.allParams + xxx.styles for p in paramStyles[paramStyles.index(param) + 1:]: # Content params don't have same type if xxx.params.has_key(p) and p != 'content': found = true break if found: nextTextElem = xxx.params[p].parentNode elem.insertBefore(textElem, nextTextElem) else: elem.appendChild(textElem) else: # Remove parameter element and following text node textElem = xxx.params[param].parentNode newline = textElem.nextSibling if newline and newline.nodeType == minidom.Node.TEXT_NODE: elem.removeChild(newline) elem.removeChild(textElem) del xxx.params[param] w.SetValue('') w.Enable(evt.IsChecked()) # Set modified flas self.SetModified(true) ################################################################################ # Properties panel notebook page class HtmlPage(wxHtmlWindow, ParamPage): def __init__(self, parent, id = -1): wxHtmlWindow.__init__(self, parent, id) ParamPage.__init__(self) self.SetBorders(5) if wxGetOsVersion()[1] == 1: self.SetFonts('', '', [8, 10, 12, 14, 16, 19, 24]) else: self.SetFonts("", "", [7, 8, 10, 12, 16, 22, 30]) self.modified = false def Clear(self): self.SetPage(htmlHeader + 'select a tree item on the left' + htmlFooter) def SetPageData(self, xxx): if not xxx: self.SetPage(htmlHeader + 'this item has no properties' + htmlFooter) return self.SetPage(htmlHeader + xxx.generateHtml() + htmlFooter) # Set values, checkboxes to false, disable defaults if xxx.hasChild: prefix = '_' else: prefix = '' for param in xxx.allParams: if xxx.params.has_key(param): if param == 'content': value = [] for text in xxx.params[param]: value.append(str(text.data)) # convert from unicode else: value = xxx.params[param].data self.FindWindowByName(prefix + 'data_' + param).SetValue(value) if not param in xxx.required: self.FindWindowByName(prefix + 'check_' + param).SetValue(true) else: self.FindWindowByName(prefix + 'data_' + param).Enable(false) # Same for the child of sizeritem if xxx.hasChild: xxx = xxx.child for param in xxx.allParams: if xxx.params.has_key(param): if param == 'content': value = [] for text in xxx.params[param]: value.append(str(text.data)) # convert from unicode else: value = xxx.params[param].data self.FindWindowByName('data_' + param).SetValue(value) if not param in xxx.required: self.FindWindowByName('check_' + param).SetValue(true) else: self.FindWindowByName('data_' + param).Enable(false) # If some parameter has changed def IsModified(self): return self.modified def SetModified(self, value): self.modified = value ################################################################################ # Style notebook page class StylePage(wxPanel, ParamPage): def __init__(self, parent, xxx): wxPanel.__init__(self, parent, -1) ParamPage.__init__(self) if wxGetOsVersion()[1] == 1: self.SetFont(wxFont(12, wxDEFAULT, wxNORMAL, wxNORMAL)) else: self.SetFont(wxFont(10, wxDEFAULT, wxNORMAL, wxNORMAL)) topSizer = wxBoxSizer(wxVERTICAL) sizer = wxFlexGridSizer(len(xxx.styles), 2, 0, 1) self.controls = {} # save python objects for param in xxx.styles: present = param in xxx.params.keys() check = wxCheckBox(self, paramIDs[param], param + ':', name = 'check_' + param) check.SetValue(present) control = paramDict[param](self, -1, '', (-1, -1), 'data_' + param) if present: control.SetValue(xxx.params[param].data) else: control.SetValue('') control.Enable(present) sizer.AddMany([ (check, 0, 0), (control, 0, 0) ]) self.controls[param] = control topSizer.Add(sizer, 1, wxALL, 5) self.SetAutoLayout(true) self.SetSizer(topSizer) self.modified = false # If some parameter has changed def IsModified(self): return self.modified def SetModified(self, value): self.modified = value ################################################################################ class HightLightBox: def __init__(self, pos, size): w = testWin.panel l1 = wxWindow(w, -1, pos, wxSize(size.x, 2)) l1.SetBackgroundColour(wxRED) l2 = wxWindow(w, -1, pos, wxSize(2, size.y)) l2.SetBackgroundColour(wxRED) l3 = wxWindow(w, -1, wxPoint(pos.x + size.x - 2, pos.y), wxSize(2, size.y)) l3.SetBackgroundColour(wxRED) l4 = wxWindow(w, -1, wxPoint(pos.x, pos.y + size.y - 2), wxSize(size.x, 2)) l4.SetBackgroundColour(wxRED) self.lines = [l1, l2, l3, l4] # Move highlight to a new position def Replace(self, pos, size): self.lines[0].SetDimensions(pos.x, pos.y, size.x, 2, wxSIZE_ALLOW_MINUS_ONE) self.lines[1].SetDimensions(pos.x, pos.y, 2, size.y, wxSIZE_ALLOW_MINUS_ONE) self.lines[2].SetDimensions(pos.x + size.x - 2, pos.y, 2, size.y, wxSIZE_ALLOW_MINUS_ONE) self.lines[3].SetDimensions(pos.x, pos.y + size.y - 2, size.x, 2, wxSIZE_ALLOW_MINUS_ONE) # Remove it def Remove(self): map(wxWindow.Destroy, self.lines) testWin.highLight = None ################################################################################ class MemoryFile: def __init__(self, name): self.name = name self.buffer = '' def write(self, data): self.buffer = self.buffer + data.encode() def close(self): f = open(self.name, 'w') f.write(self.buffer) f.close() # !!! memory FS will work someday #self.file = wxMemoryFSHandler_AddFile(self.name, self.buffer) class XML_Tree(wxTreeCtrl): def __init__(self, parent, id): wxTreeCtrl.__init__(self, parent, id, style=wxTR_HAS_BUTTONS | wxTR_LINES_AT_ROOT) self.SetBackgroundColour(wxColour(224, 248, 224)) EVT_TREE_SEL_CHANGED(self, self.GetId(), self.OnSelChanged) EVT_TREE_ITEM_ACTIVATED(self, self.GetId(), self.OnItemActivated) EVT_RIGHT_DOWN(self, self.OnRightDown) self.needUpdate = false self.pendingHighLight = None self.ctrl = false self.dom = None # Create image list il = wxImageList(16, 16, true) self.rootImage = il.AddIcon(wxIconFromXPMData(images.getTreeRootData())) xxxObject.image = il.AddIcon(wxIconFromXPMData(images.getTreeDefaultData())) xxxPanel.image = il.AddIcon(wxIconFromXPMData(images.getTreePanelData())) xxxDialog.image = il.AddIcon(wxIconFromXPMData(images.getTreeDialogData())) xxxFrame.image = il.AddIcon(wxIconFromXPMData(images.getTreeFrameData())) xxxMenuBar.image = il.AddIcon(wxIconFromXPMData(images.getTreeMenuBarData())) xxxMenu.image = il.AddIcon(wxIconFromXPMData(images.getTreeMenuData())) xxxSizer.imageH = il.AddIcon(wxIconFromXPMData(images.getTreeSizerHData())) xxxSizer.imageV = il.AddIcon(wxIconFromXPMData(images.getTreeSizerVData())) xxxStaticBoxSizer.imageH = il.AddIcon(wxIconFromXPMData(images.getTreeStaticBoxSizerHData())) xxxStaticBoxSizer.imageV = il.AddIcon(wxIconFromXPMData(images.getTreeStaticBoxSizerVData())) xxxGridSizer.image = il.AddIcon(wxIconFromXPMData(images.getTreeSizerGridData())) xxxFlexGridSizer.image = il.AddIcon(wxIconFromXPMData(images.getTreeSizerFlexGridData())) self.il = il self.SetImageList(il) # !!! temporary solution for GetOldItem problem def Unselect(self): self.selection = wxTreeItemId() wxTreeCtrl.Unselect(self) def GetSelection(self): return self.selection def ExpandAll(self, item): if self.ItemHasChildren(item): self.Expand(item) i, cookie = self.GetFirstChild(item, 0) children = [] while i.IsOk(): children.append(i) i, cookie = self.GetNextChild(item, cookie) for i in children: self.ExpandAll(i) def SetData(self, dom): self.dom = dom # Find 'resource' child, add it's children self.mainNode = dom.getElementsByTagName('resource')[0] nodes = self.mainNode.childNodes[:] for node in nodes: if IsObject(node): self.AddNode(self.GetRootItem(), None, node) else: self.mainNode.removeChild(node) node.unlink() self.Unselect() # Add tree item for given parent item if node is DOM element node with # 'object' tag. xxxParent is parent xxx object def AddNode(self, itemParent, xxxParent, node): # Set item data to current node xxx = MakeXXXFromDOM(xxxParent, node) treeObj = xxx.treeObject() # Append tree item item = self.AppendItem(itemParent, treeObj.treeName(), image=treeObj.treeImage(), data=wxTreeItemData(xxx)) # Try to find children objects if treeObj.hasChildren: nodes = treeObj.element.childNodes[:] for n in nodes: if IsObject(n): self.AddNode(item, treeObj, n) elif n.nodeType != minidom.Node.ELEMENT_NODE: treeObj.element.removeChild(n) n.unlink() # Remove leaf of tree, return it's data object def RemoveLeaf(self, leaf): xxx = self.GetPyData(leaf) node = xxx.element parent = node.parentNode parent.removeChild(node) self.Delete(leaf) # Update view? #if testWin and self.GetItemAncestor(leaf) == testWin.item: # if testWin.highLight: # testWin.highLight.Remove() # self.needUpdate = true return node # Find position relative to the top-level window def FindNodePos(self, item): itemParent = self.GetItemParent(item) if itemParent == self.GetRootItem(): return wxPoint(0, 0) obj = self.FindNodeObject(item) # Find first ancestor which is a wxWindow (not a sizer) winParent = itemParent while self.GetPyData(winParent).isSizer: winParent = self.GetItemParent(winParent) parentPos = self.FindNodePos(winParent) return parentPos + obj.GetPosition() # Find window (or sizer) corresponding to a tree item. def FindNodeObject(self, item): itemParent = self.GetItemParent(item) # If top-level, return testWin (or panel if wxFrame) if itemParent == self.GetRootItem(): return testWin.panel xxx = self.GetPyData(item).treeObject() parentWin = self.FindNodeObject(itemParent) # Top-level sizer? return window's sizer if xxx.isSizer and isinstance(parentWin, wxWindowPtr): return parentWin.GetSizer() # Otherwise get parent's object and it's child n = 0 # index of sibling prev = self.GetPrevSibling(item) while prev.IsOk(): prev = self.GetPrevSibling(prev) n += 1 child = parentWin.GetChildren()[n] # Return window or sizer for sizer items if child.GetClassName() == 'wxSizerItem': if child.IsWindow(): child = child.GetWindow() elif child.IsSizer(): child = child.GetSizer() # Test for notebook sizers if isinstance(child, wxNotebookSizerPtr): child = child.GetNotebook() return child def OnSelChanged(self, evt): # Apply changes # !!! problem with wxGTK #oldItem = evt.GetOldItem() oldItem = self.selection if oldItem.IsOk(): xxx = self.GetPyData(oldItem) # If some data was modified, apply changes if xxx: if panel.IsModified(): self.Apply(xxx, oldItem) #if conf.autoRefresh: if testWin and testWin.highLight: testWin.highLight.Remove() self.needUpdate = true # Generate HTML view item = evt.GetItem() self.selection = item # !!! fix xxx = self.GetPyData(item) # Update panel panel.SetData(xxx) # Remove highlight? if not xxx and testWin and testWin.highLight: testWin.highLight.Remove() return # Clear flag panel.SetModified(false) # Hightlighting is done in OnIdle tree.pendingHighLight = item # Find top-level parent def GetItemAncestor(self, item): while self.GetItemParent(item) != self.GetRootItem(): item = self.GetItemParent(item) return item # Highlight selected item def HighLight(self, item): self.pendingHighLight = None if not testWin or self.GetPyData(testWin.item).className \ not in ['wxDialog', 'wxPanel', 'wxFrame']: return # Top-level does not have highlight if item == testWin.item: if testWin.highLight: testWin.highLight.Remove() return # If a control from another window is selected, remove highlight if self.GetItemAncestor(item) != testWin.item and testWin.highLight: testWin.highLight.Remove() return # Get window/sizer object obj, pos = self.FindNodeObject(item), self.FindNodePos(item) size = obj.GetSize() # For notebook, select item's page. # For children of page, nothing happens (too much work) if isinstance(self.GetPyData(item).parent, xxxNotebook): notebook = self.FindNodeObject(self.GetItemParent(item)) # Find position n = 0 prev = self.GetPrevSibling(item) while prev.IsOk(): n += 1 prev = self.GetPrevSibling(prev) notebook.SetSelection(n) # Highlight try: # finally I use exceptions testWin.highLight.Replace(pos, size) except AttributeError: testWin.highLight = HightLightBox(pos, size) testWin.highLight.item = item # Double-click def OnItemActivated(self, evt): item = evt.GetItem() xxx = self.GetPyData(item) if not xxx: return # if root selected, do nothing if panel.IsModified(): self.Apply(xxx, item) # apply changes self.CreateTestWin(item) # (re)create test window def CreateTestWin(self, node): global testWin # Create a window with this resource xxx = self.GetPyData(node).treeObject() if not xxx: return # if root selected, do nothing # If noname element, display error if not xxx.hasName or not xxx.name: wxLogError("Can't display a noname element") return # Close old window, remember where it was highLight = None if testWin: pos = testWin.GetPosition() if node == testWin.item: # Remember highlight if same top-level window if testWin.highLight: highLight = testWin.highLight.item # !!! if 0 is removed, refresh is broken (notebook not deleted?) if 0 and xxx.className == 'wxPanel': if testWin.highLight: testWin.pendingHighLight = highLight testWin.highLight.Remove() testWin.panel.Destroy() testWin.panel = None else: testWin.Destroy() testWin = None else: testWin.Destroy() testWin = None else: pos = testWinPos # Save in temporary file before activating memFile = MemoryFile(tempfile.mktemp('xrc')) #memFile = MemoryFile('core.xrc') # to write debug file # Create partial XML file - faster for big files dom = minidom.Document() mainNode = dom.createElement('resource') dom.appendChild(mainNode) # Remove temporarily from old parent elem = xxx.element parent = elem.parentNode next = elem.nextSibling parent.replaceChild(self.dummyNode, elem) # Append to new DOM, write it mainNode.appendChild(elem) dom.writexml(memFile) # Put back in place mainNode.removeChild(elem) dom.unlink() parent.replaceChild(elem, self.dummyNode) memFile.close() # write to wxMemoryFS res = wxXmlResource('') res.Load(memFile.name) if xxx.className == 'wxFrame': # Create new frame testWin = wxPreFrame() res.LoadFrame(testWin, frame, xxx.name) testWin.panel = testWin testWin.SetPosition(pos) testWin.Show(true) elif xxx.className == 'wxPanel': # Create new frame if not testWin: testWin = wxFrame(frame, -1, 'Panel: ' + xxx.name, pos=pos) testWin.panel = res.LoadPanel(testWin, xxx.name) testWin.SetSize(testWin.panel.GetSize()) testWin.Show(true) elif xxx.className == 'wxDialog': # Create new frame testWin = res.LoadDialog(None, xxx.name) testWin.panel = testWin testWin.SetPosition(pos) testWin.Show(true) elif xxx.className == 'wxMenuBar': testWin = wxFrame(frame, -1, 'MenuBar: ' + xxx.name, pos=pos) # Set status bar to display help testWin.CreateStatusBar() testWin.menuBar = res.LoadMenuBar(xxx.name) testWin.SetMenuBar(testWin.menuBar) testWin.Show(true) else: wxLogMessage('No view for this element yet') return os.unlink(memFile.name) # remove tmp file testWin.item = node testWin.Connect(testWin.GetId(), -1, wxEVT_CLOSE_WINDOW, self.OnCloseTestWin) testWin.highLight = None if highLight and not tree.pendingHighLight: self.HighLight(highLight) def OnCloseTestWin(self, evt): global testWin, testWinPos testWinPos = testWin.GetPosition() testWin.Destroy() testWin = None evt.Skip() # True if next item should be inserted after current (vs. appended to it) def NeedInsert(self, item): xxx = self.GetPyData(item) if not xxx: return false # root item if self.ctrl: return true # if Ctrl pressed, always insert if xxx.hasChildren and not self.ItemHasChildren(item): return false return not (self.IsExpanded(item) and self.ItemHasChildren(item)) # Pull-down def OnRightDown(self, evt): # Setup menu menu = wxMenu() item = self.GetSelection() if not item.IsOk(): menu.Append(pullDownMenu.ID_EXPAND, 'Expand', 'Expand tree') else: self.ctrl = evt.ControlDown() # save Ctrl state m = wxMenu() # create menu if item != self.GetRootItem(): needInsert = self.NeedInsert(item) if item == self.GetRootItem() or \ self.GetItemParent(item) == self.GetRootItem() and needInsert: m.Append(pullDownMenu.ID_NEW_PANEL, 'Panel', 'Create panel') m.Append(pullDownMenu.ID_NEW_DIALOG, 'Dialog', 'Create dialog') m.Append(pullDownMenu.ID_NEW_FRAME, 'Frame', 'Create frame') m.AppendSeparator() m.Append(pullDownMenu.ID_NEW_MENU_BAR, 'MenuBar', 'Create menu bar') m.Append(pullDownMenu.ID_NEW_MENU, 'Menu', 'Create menu') else: xxx = self.GetPyData(item) if xxx.__class__ == xxxMenuBar: m.Append(pullDownMenu.ID_NEW_MENU, 'Menu', 'Create menu') elif xxx.__class__ in [xxxMenu, xxxMenuItem]: SetMenu(m, pullDownMenu.menuControls) else: SetMenu(m, pullDownMenu.controls) if not (xxx.isSizer or \ xxx.parent and xxx.parent.isSizer): m.Enable(pullDownMenu.ID_NEW_SPACER, false) # Select correct label for create menu if item == self.GetRootItem(): menu.AppendMenu(wxNewId(), 'Create', m, 'Create top-level object') else: if not needInsert: menu.AppendMenu(wxNewId(), 'Create child', m, 'Create child object') else: menu.AppendMenu(wxNewId(), 'Create Sibling', m, 'Create sibling of selected object') menu.AppendSeparator() menu.Append(wxID_CUT, 'Cut', 'Cut to the clipboard') menu.Append(wxID_COPY, 'Copy', 'Copy to the clipboard') menu.Append(wxID_PASTE, 'Paste', 'Paste from the clipboard') menu.Append(pullDownMenu.ID_DELETE, 'Delete', 'Delete object') if item.IsOk() and self.ItemHasChildren(item): menu.AppendSeparator() menu.Append(pullDownMenu.ID_EXPAND, 'Expand', 'Expand subtree') self.PopupMenu(menu, evt.GetPosition()) menu.Destroy() # Clear tree def Clear(self): self.DeleteAllItems() # Add minimal structure root = self.AddRoot('XML tree', self.rootImage) self.Unselect() if self.dom: self.dom.unlink() self.dom = minidom.Document() self.dummyNode = self.dom.createComment('dummy node') # Create main node self.mainNode = self.dom.createElement('resource') self.dom.appendChild(self.mainNode) # Apply changes def Apply(self, xxx, item): if not xxx: return # !!! Save undo info if xxx.undo: xxx.undo.unlink() xxx.undo = xxx.element.cloneNode(false) if xxx.hasName: name = panel.page1.FindWindowByName('data_name').GetValue() if xxx.name != name: xxx.name = name xxx.element.setAttribute('name', name) self.SetItemText(item, xxx.treeName()) if xxx.hasChild: prefix = '_' else: prefix = '' for param, data in xxx.params.items(): value = panel.FindWindowByName(prefix + 'data_' + param).GetValue() if param == 'content': # If number if items is not the same, recreate children if len(value) != len(data): elem = xxx.element.getElementsByTagName('content')[0] for n in elem.childNodes: elem.removeChild(n) data = [] for str in value: itemElem = tree.dom.createElement('item') itemText = tree.dom.createTextNode(str) itemElem.appendChild(itemText) elem.appendChild(itemElem) data.append(itemText) xxx.params[param] = data else: for i in range(len(value)): data[i].data = value[i] elif param == 'font': data.updateXML(value) else: data.data = value if xxx.hasChild: self.Apply(xxx.child, item) else: # Change tree icon for sizers if isinstance(xxx, xxxBoxSizer): self.SetItemImage(item, xxx.treeImage()) # Set global modified state frame.modified = true class PullDownMenu: ID_NEW_PANEL = wxNewId() ID_NEW_DIALOG = wxNewId() ID_NEW_FRAME = wxNewId() ID_NEW_MENU_BAR = wxNewId() ID_NEW_MENU = wxNewId() ID_NEW_STATIC_TEXT = wxNewId() ID_NEW_TEXT_CTRL = wxNewId() ID_NEW_BUTTON = wxNewId() ID_NEW_BITMAP_BUTTON = wxNewId() ID_NEW_RADIO_BUTTON = wxNewId() ID_NEW_SPIN_BUTTON = wxNewId() ID_NEW_STATIC_BOX = wxNewId() ID_NEW_CHECK_BOX = wxNewId() ID_NEW_RADIO_BOX = wxNewId() ID_NEW_COMBO_BOX = wxNewId() ID_NEW_LIST_BOX = wxNewId() ID_NEW_STATIC_LINE = wxNewId() ID_NEW_CHOICE = wxNewId() ID_NEW_SLIDER = wxNewId() ID_NEW_GAUGE = wxNewId() ID_NEW_SCROLL_BAR = wxNewId() ID_NEW_TREE_CTRL = wxNewId() ID_NEW_LIST_CTRL = wxNewId() ID_NEW_CHECK_LIST = wxNewId() ID_NEW_NOTEBOOK = wxNewId() ID_NEW_HTML_WINDOW = wxNewId() ID_NEW_CALENDAR = wxNewId() ID_NEW_BOX_SIZER = wxNewId() ID_NEW_STATIC_BOX_SIZER = wxNewId() ID_NEW_GRID_SIZER = wxNewId() ID_NEW_FLEX_GRID_SIZER = wxNewId() ID_NEW_SPACER = wxNewId() ID_NEW_MENU = wxNewId() ID_NEW_MENU_ITEM = wxNewId() ID_NEW_SEPARATOR = wxNewId() ID_NEW_LAST = wxNewId() ID_EXPAND = wxNewId() def __init__(self, parent): self.ID_DELETE = parent.ID_DELETE EVT_MENU_RANGE(parent, self.ID_NEW_PANEL, self.ID_NEW_LAST, parent.OnCreate) EVT_MENU(parent, self.ID_EXPAND, parent.OnExpand) # We connect to tree, but process in frame EVT_MENU_HIGHLIGHT_ALL(tree, parent.OnPullDownHighlight) class Frame(wxFrame): def __init__(self, size): wxFrame.__init__(self, None, -1, '', size=size) self.CreateStatusBar() self.SetIcon(wxIconFromXPMData(images.getIconData())) # Make menus menuBar = wxMenuBar() menu = wxMenu() menu.Append(wxID_NEW, '&New\tCtrl-N', 'New file') menu.Append(wxID_OPEN, '&Open...\tCtrl-O', 'Open XRC file') menu.Append(wxID_SAVE, '&Save\tCtrl-S', 'Save XRC file') menu.Append(wxID_SAVEAS, 'Save &As...', 'Save XRC file under different name') menu.AppendSeparator() menu.Append(wxID_EXIT, '&Quit\tCtrl-Q', 'Exit application') menuBar.Append(menu, '&File') menu = wxMenu() menu.Append(wxID_UNDO, '&Undo\tCtrl-Z', 'Undo') menu.Append(wxID_REDO, '&Redo\tCtrl-R', 'Redo') menu.AppendSeparator() menu.Append(wxID_CUT, 'Cut\tCtrl-X', 'Cut to the clipboard') menu.Append(wxID_COPY, '&Copy\tCtrl-C', 'Copy to the clipboard') menu.Append(wxID_PASTE, '&Paste\tCtrl-V', 'Paste from the clipboard') self.ID_DELETE = wxNewId() menu.Append(self.ID_DELETE, '&Delete\tCtrl-D', 'Delete object') menuBar.Append(menu, '&Edit') menu = wxMenu() self.ID_REFRESH = wxNewId() menu.Append(self.ID_REFRESH, '&Refresh\tCtrl-R', 'Refresh view') self.ID_AUTO_REFRESH = wxNewId() menu.Append(self.ID_AUTO_REFRESH, '&Auto-refresh\tCtrl-A', 'Toggle auto-refresh mode', true) menu.Check(self.ID_AUTO_REFRESH, conf.autoRefresh) menuBar.Append(menu, '&View') menu = wxMenu() menu.Append(wxID_ABOUT, 'About...', 'About XCRed') if debug: self.ID_DEBUG_CMD = wxNewId() menu.Append(self.ID_DEBUG_CMD, 'CMD', 'Python command line') EVT_MENU(self, self.ID_DEBUG_CMD, self.OnDebugCMD) menuBar.Append(menu, '&Help') self.menuBar = menuBar self.SetMenuBar(menuBar) # Create toolbar tb = self.CreateToolBar(wxTB_HORIZONTAL | wxNO_BORDER | wxTB_FLAT) tb.SetToolBitmapSize((24, 23)) tb.AddSimpleTool(wxID_NEW, images.getNewBitmap(), 'New', 'New file') tb.AddSimpleTool(wxID_OPEN, images.getOpenBitmap(), 'Open', 'Open file') tb.AddSimpleTool(wxID_SAVE, images.getSaveBitmap(), 'Save', 'Save file') tb.AddSeparator() tb.AddSimpleTool(wxID_CUT, images.getCutBitmap(), 'Cut', 'Cut') tb.AddSimpleTool(wxID_COPY, images.getCopyBitmap(), 'Copy', 'Copy') tb.AddSimpleTool(wxID_PASTE, images.getPasteBitmap(), 'Paste', 'Paste') tb.AddSeparator() tb.AddSimpleTool(self.ID_REFRESH, images.getRefreshBitmap(), 'Refresh', 'Refresh view') tb.AddSimpleTool(self.ID_AUTO_REFRESH, images.getAutoRefreshBitmap(), 'Auto-refresh', 'Toggle auto-refresh mode', true) tb.ToggleTool(self.ID_AUTO_REFRESH, conf.autoRefresh) tb.Realize() self.tb = tb # File EVT_MENU(self, wxID_NEW, self.OnNew) EVT_MENU(self, wxID_OPEN, self.OnOpen) EVT_MENU(self, wxID_SAVE, self.OnSaveOrSaveAs) EVT_MENU(self, wxID_SAVEAS, self.OnSaveOrSaveAs) EVT_MENU(self, wxID_EXIT, self.OnExit) # Edit EVT_MENU(self, wxID_UNDO, self.OnUndo) EVT_MENU(self, wxID_REDO, self.OnRedo) EVT_MENU(self, wxID_CUT, self.OnCut) EVT_MENU(self, wxID_COPY, self.OnCopy) EVT_MENU(self, wxID_PASTE, self.OnPaste) EVT_MENU(self, self.ID_DELETE, self.OnDelete) # View EVT_MENU(self, self.ID_REFRESH, self.OnRefresh) EVT_MENU(self, self.ID_AUTO_REFRESH, self.OnAutoRefresh) # Help EVT_MENU(self, wxID_ABOUT, self.OnAbout) # Update events EVT_UPDATE_UI(self, wxID_CUT, self.OnUpdateUI) EVT_UPDATE_UI(self, wxID_COPY, self.OnUpdateUI) EVT_UPDATE_UI(self, wxID_PASTE, self.OnUpdateUI) EVT_UPDATE_UI(self, self.ID_DELETE, self.OnUpdateUI) # Build interface sizer = wxBoxSizer(wxVERTICAL) sizer.Add(wxStaticLine(self, -1), 0, wxEXPAND) splitter = wxSplitterWindow(self, -1, style=wxSP_3DSASH) splitter.SetMinimumPaneSize(100) # Create tree global tree tree = XML_Tree(splitter, -1) sys.modules['xxx'].tree = tree # Create panel for parameters global panel panel = Panel(splitter) # Set plitter windows splitter.SplitVertically(tree, panel, 200) sizer.Add(splitter, 1, wxEXPAND) self.SetAutoLayout(true) self.SetSizer(sizer) # Init pull-down menu data global pullDownMenu pullDownMenu = PullDownMenu(self) # Mapping from IDs to element names self.createMap = { pullDownMenu.ID_NEW_PANEL: 'wxPanel', pullDownMenu.ID_NEW_DIALOG: 'wxDialog', pullDownMenu.ID_NEW_FRAME: 'wxFrame', pullDownMenu.ID_NEW_MENU_BAR: 'wxMenuBar', pullDownMenu.ID_NEW_MENU: 'wxMenu', pullDownMenu.ID_NEW_STATIC_TEXT: 'wxStaticText', pullDownMenu.ID_NEW_TEXT_CTRL: 'wxTextCtrl', pullDownMenu.ID_NEW_BUTTON: 'wxButton', pullDownMenu.ID_NEW_BITMAP_BUTTON: 'wxBitmapButton', pullDownMenu.ID_NEW_RADIO_BUTTON: 'wxRadioButton', pullDownMenu.ID_NEW_SPIN_BUTTON: 'wxSpinButton', pullDownMenu.ID_NEW_STATIC_BOX: 'wxStaticBox', pullDownMenu.ID_NEW_CHECK_BOX: 'wxCheckBox', pullDownMenu.ID_NEW_RADIO_BOX: 'wxRadioBox', pullDownMenu.ID_NEW_COMBO_BOX: 'wxComboBox', pullDownMenu.ID_NEW_LIST_BOX: 'wxListBox', pullDownMenu.ID_NEW_STATIC_LINE: 'wxStaticLine', pullDownMenu.ID_NEW_CHOICE: 'wxChoice', pullDownMenu.ID_NEW_SLIDER: 'wxSlider', pullDownMenu.ID_NEW_GAUGE: 'wxGauge', pullDownMenu.ID_NEW_SCROLL_BAR: 'wxScrollBar', pullDownMenu.ID_NEW_TREE_CTRL: 'wxTreeCtrl', pullDownMenu.ID_NEW_LIST_CTRL: 'wxListCtrl', pullDownMenu.ID_NEW_CHECK_LIST: 'wxCheckList', pullDownMenu.ID_NEW_NOTEBOOK: 'wxNotebook', pullDownMenu.ID_NEW_HTML_WINDOW: 'wxHtmlWindow', pullDownMenu.ID_NEW_CALENDAR: 'wxCalendar', pullDownMenu.ID_NEW_BOX_SIZER: 'wxBoxSizer', pullDownMenu.ID_NEW_STATIC_BOX_SIZER: 'wxStaticBoxSizer', pullDownMenu.ID_NEW_GRID_SIZER: 'wxGridSizer', pullDownMenu.ID_NEW_FLEX_GRID_SIZER: 'wxFlexGridSizer', pullDownMenu.ID_NEW_SPACER: 'spacer', pullDownMenu.ID_NEW_MENU: 'wxMenu', pullDownMenu.ID_NEW_MENU_ITEM: 'wxMenuItem', pullDownMenu.ID_NEW_SEPARATOR: 'separator', } pullDownMenu.controls = [ ['control', 'Various controls', (pullDownMenu.ID_NEW_STATIC_TEXT, 'Label', 'Create static label'), (pullDownMenu.ID_NEW_STATIC_LINE, 'Line', 'Create static line'), (pullDownMenu.ID_NEW_TEXT_CTRL, 'TextBox', 'Create text box control'), (pullDownMenu.ID_NEW_CHOICE, 'Choice', 'Create choice control'), (pullDownMenu.ID_NEW_SLIDER, 'Slider', 'Create slider control'), (pullDownMenu.ID_NEW_GAUGE, 'Gauge', 'Create gauge control'), (pullDownMenu.ID_NEW_SCROLL_BAR, 'ScrollBar', 'Create scroll bar'), (pullDownMenu.ID_NEW_TREE_CTRL, 'TreeCtrl', 'Create tree control'), (pullDownMenu.ID_NEW_LIST_CTRL, 'ListCtrl', 'Create list control'), (pullDownMenu.ID_NEW_CHECK_LIST, 'CheckList', 'Create check list control'), (pullDownMenu.ID_NEW_HTML_WINDOW, 'HtmlWindow', 'Create HTML window'), (pullDownMenu.ID_NEW_CALENDAR, 'Calendar', 'Create calendar control'), (pullDownMenu.ID_NEW_PANEL, 'Panel', 'Create panel'), (pullDownMenu.ID_NEW_NOTEBOOK, 'Notebook', 'Create notebook control'), ], ['button', 'Buttons', (pullDownMenu.ID_NEW_BUTTON, 'Button', 'Create button'), (pullDownMenu.ID_NEW_BITMAP_BUTTON, 'BitmapButton', 'Create bitmap button'), (pullDownMenu.ID_NEW_RADIO_BUTTON, 'RadioButton', 'Create radio button'), (pullDownMenu.ID_NEW_SPIN_BUTTON, 'SpinButton', 'Create spin button'), ], ['box', 'Boxes', (pullDownMenu.ID_NEW_STATIC_BOX, 'StaticBox', 'Create static box'), (pullDownMenu.ID_NEW_CHECK_BOX, 'CheckBox', 'Create check box'), (pullDownMenu.ID_NEW_RADIO_BOX, 'RadioBox', 'Create radio box'), (pullDownMenu.ID_NEW_COMBO_BOX, 'ComboBox', 'Create combo box'), (pullDownMenu.ID_NEW_LIST_BOX, 'ListBox', 'Create list box'), ], ['sizer', 'Sizers', (pullDownMenu.ID_NEW_BOX_SIZER, 'BoxSizer', 'Create box sizer'), (pullDownMenu.ID_NEW_STATIC_BOX_SIZER, 'StaticBoxSizer', 'Create static box sizer'), (pullDownMenu.ID_NEW_GRID_SIZER, 'GridSizer', 'Create grid sizer'), (pullDownMenu.ID_NEW_FLEX_GRID_SIZER, 'FlexGridSizer', 'Create flexgrid sizer'), (pullDownMenu.ID_NEW_SPACER, 'Spacer', 'Create spacer'), ] ] pullDownMenu.menuControls = [ (pullDownMenu.ID_NEW_MENU, 'Menu', 'Create menu'), (pullDownMenu.ID_NEW_MENU_ITEM, 'MenuItem', 'Create menu item'), (pullDownMenu.ID_NEW_SEPARATOR, 'Separator', 'Create separator'), ] # Initialize self.Clear() # Other events EVT_IDLE(self, self.OnIdle) EVT_CLOSE(self, self.OnCloseWindow) def OnNew(self, evt): self.Clear() def OnOpen(self, evt): if not self.AskSave(): return dlg = wxFileDialog(self, 'Open', os.path.dirname(self.dataFile), '', '*.xrc', wxOPEN | wxCHANGE_DIR) if dlg.ShowModal() == wxID_OK: path = dlg.GetPath() self.SetStatusText('Loading...') wxYield() wxBeginBusyCursor() self.Open(path) wxEndBusyCursor() self.SetStatusText('Ready') dlg.Destroy() def OnSaveOrSaveAs(self, evt): if evt.GetId() == wxID_SAVEAS or not self.dataFile: if self.dataFile: defaultName = '' else: defaultName = 'UNTITLED.xrc' dlg = wxFileDialog(self, 'Save As', os.path.dirname(self.dataFile), defaultName, '*.xrc', wxSAVE | wxOVERWRITE_PROMPT | wxCHANGE_DIR) if dlg.ShowModal() == wxID_OK: path = dlg.GetPath() dlg.Destroy() else: dlg.Destroy() return else: path = self.dataFile self.SetStatusText('Saving...') wxYield() wxBeginBusyCursor() self.Save(path) self.dataFile = path wxEndBusyCursor() self.SetStatusText('Ready') def OnExit(self, evt): self.Close() def OnUndo(self, evt): print '*** being implemented' print self.lastOp, self.undo if self.lastOp == 'DELETE': parent, prev, elem = self.undo if prev.IsOk(): xxx = MakeXXXFromDOM(tree.GetPyData(parent).treeObject(), elem) item = tree.InsertItem( parent, prev, xxx.treeObject().className, data=wxTreeItemData(xxx) ) def OnRedo(self, evt): print '*** being implemented' def OnCut(self, evt): selected = tree.GetSelection() # Undo info self.lastOp = 'CUT' self.undo = [tree.GetItemParent(selected), tree.GetPrevSibling(selected)] # Delete testWin? global testWin if testWin: # If deleting top-level item, delete testWin if selected == testWin.item: testWin.Destroy() testWin = None else: # Remove highlight, update testWin if tree.GetItemAncestor(selected) == testWin.item: if testWin.highLight: testWin.highLight.Remove() tree.needUpdate = true self.clipboard = tree.RemoveLeaf(selected) tree.pendingHighLight = None tree.Unselect() panel.Clear() self.modified = true def OnCopy(self, evt): selected = tree.GetSelection() xxx = tree.GetPyData(selected) self.clipboard = xxx.element.cloneNode(true) def OnPaste(self, evt): selected = tree.GetSelection() appendChild = not tree.NeedInsert(selected) xxx = tree.GetPyData(selected) if not appendChild: # If has next item, insert, else append to parent nextItem = tree.GetNextSibling(selected) if nextItem.IsOk(): # Insert before nextItem parentLeaf = tree.GetItemParent(selected) else: # last child: change selected to parent appendChild = true selected = tree.GetItemParent(selected) # Expanded container (must have children) elif tree.IsExpanded(selected) and tree.ItemHasChildren(selected): appendChild = false nextItem = tree.GetFirstChild(selected, 0)[0] parentLeaf = selected # Parent should be tree element or None if appendChild: parent = tree.GetPyData(selected) else: parent = tree.GetPyData(parentLeaf) if parent and parent.hasChild: parent = parent.child # Create a copy of clipboard element elem = self.clipboard.cloneNode(true) # Tempopary xxx object to test things xxx = MakeXXXFromDOM(parent, elem) className = xxx.treeObject().className # Check parent and child relationships # Parent is sizer or notebook, child is of wrong class or # parent is normal window, child is child container: detach child isChildContainer = isinstance(xxx, xxxChildContainer) if not parent and isChildContainer or \ (parent.isSizer and not isinstance(xxx, xxxSizerItem)) or \ (isinstance(parent, xxxNotebook) and not isinstance(xxx, xxxNotebookPage)) or \ (not parent.isSizer and not isinstance(parent, xxxNotebook) and \ isChildContainer): if isChildContainer: elem.removeChild(xxx.child.element) # detach child elem.unlink() # delete child container elem = xxx.child.element # replace # This should help garbage collection (!!! maybe not needed?) xxx.child.parent = None xxx.child = None if parent: # Parent is sizer or notebook, child is not child container if parent.isSizer and not isChildContainer and \ not isinstance(xxx, xxxSpacer): # Create sizer item element sizerItemElem = MakeEmptyDOM('sizeritem') sizerItemElem.appendChild(elem) elem = sizerItemElem elif isinstance(parent, xxxNotebook) and not isChildContainer: pageElem = MakeEmptyDOM('notebookpage') pageElem.appendChild(elem) elem = pageElem xxx = MakeXXXFromDOM(parent, elem) # Figure out if we must append a new child or sibling if appendChild: if parent: node = parent.element else: node = tree.mainNode node.appendChild(elem) newItem = tree.AppendItem(selected, xxx.treeName(), image=xxx.treeImage(), data=wxTreeItemData(xxx)) else: node = tree.GetPyData(nextItem).element if parent: elemParent = parent.element else: elemParent = tree.mainNode elemParent.insertBefore(elem, node) # Inserting before is difficult, se we insert after or first child newItem = tree.InsertItem(parentLeaf, selected, xxx.treeName(), image=xxx.treeImage(), data=wxTreeItemData(xxx)) # Add children items if xxx.hasChildren: treeObj = xxx.treeObject() for n in treeObj.element.childNodes: if IsObject(n): tree.AddNode(newItem, treeObj, n) # Scroll to show new item tree.EnsureVisible(newItem) tree.SelectItem(newItem) if not tree.IsVisible(newItem): tree.ScrollTo(newItem) tree.Refresh() # Update view? if testWin and tree.GetItemAncestor(newItem) == testWin.item: if conf.autoRefresh: tree.needUpdate = true tree.pendingHighLight = newItem else: tree.pendingHighLight = None self.modified = true def OnDelete(self, evt): selected = tree.GetSelection() # Undo info self.lastOp = 'DELETE' self.undo = [tree.GetItemParent(selected), tree.GetPrevSibling(selected)] # Delete testWin? global testWin if testWin: # If deleting top-level item, delete testWin if selected == testWin.item: testWin.Destroy() testWin = None else: # Remove highlight, update testWin if tree.GetItemAncestor(selected) == testWin.item: if testWin.highLight: testWin.highLight.Remove() tree.needUpdate = true xnode = tree.RemoveLeaf(selected) self.undo.append(xnode.cloneNode(true)) xnode.unlink() tree.pendingHighLight = None tree.Unselect() panel.Clear() self.modified = true def OnRefresh(self, evt): # If modified, apply first selection = tree.GetSelection() if selection.IsOk(): xxx = tree.GetPyData(selection) if xxx and panel.IsModified(): tree.Apply(xxx, selection) if testWin: # (re)create tree.CreateTestWin(testWin.item) tree.needUpdate = false def OnAutoRefresh(self, evt): conf.autoRefresh = evt.IsChecked() self.menuBar.Check(self.ID_AUTO_REFRESH, conf.autoRefresh) self.tb.ToggleTool(self.ID_AUTO_REFRESH, conf.autoRefresh) def OnAbout(self, evt): wxMessageDialog(self, '%s %s\n\nRoman Rolinsky ' % \ (progname, version), 'About %s' % progname, wxOK | wxCENTRE).ShowModal() # Simple emulation of python command line def OnDebugCMD(self, evt): while 1: try: exec raw_input('C:\> ') except EOFError: print '^D' break except: (etype, value, tb) =sys.exc_info() tblist =traceback.extract_tb(tb)[1:] msg =string.join(traceback.format_exception_only(etype, value) +traceback.format_list(tblist)) print msg def OnCreate(self, evt): selected = tree.GetSelection() appendChild = not tree.NeedInsert(selected) xxx = tree.GetPyData(selected) if not appendChild: # If has next item, insert, else append to parent nextItem = tree.GetNextSibling(selected) if nextItem.IsOk(): # Insert before nextItem parentLeaf = tree.GetItemParent(selected) else: # last child: change selected to parent appendChild = true selected = tree.GetItemParent(selected) # Expanded container (must have children) elif tree.IsExpanded(selected) and tree.ItemHasChildren(selected): appendChild = false nextItem = tree.GetFirstChild(selected, 0)[0] parentLeaf = selected # Parent should be tree element or None if appendChild: parent = tree.GetPyData(selected) else: parent = tree.GetPyData(parentLeaf) if parent and parent.hasChild: parent = parent.child # Create element className = self.createMap[evt.GetId()] xxx = MakeEmptyXXX(parent, className) # Figure out if we must append a new child or sibling elem = xxx.element if appendChild: if parent: node = parent.element else: node = tree.mainNode # Insert newline for debug purposes node.appendChild(tree.dom.createTextNode('\n')) node.appendChild(elem) newItem = tree.AppendItem(selected, xxx.treeName(), image=xxx.treeImage(), data=wxTreeItemData(xxx)) else: node = tree.GetPyData(nextItem).element if parent: elemParent = parent.element else: elemParent = tree.mainNode elemParent.insertBefore(tree.dom.createTextNode('\n'), node) elemParent.insertBefore(elem, node) # Inserting before is difficult, se we insert after or first child newItem = tree.InsertItem(parentLeaf, selected, xxx.treeName(), image=xxx.treeImage(), data=wxTreeItemData(xxx)) tree.EnsureVisible(newItem) tree.SelectItem(newItem) if not tree.IsVisible(newItem): tree.ScrollTo(newItem) tree.Refresh() # Update view? if testWin and tree.GetItemAncestor(newItem) == testWin.item: if conf.autoRefresh: tree.needUpdate = true tree.pendingHighLight = newItem else: tree.pendingHighLight = None def OnExpand(self, evt): if tree.GetSelection().IsOk(): tree.ExpandAll(tree.GetSelection()) else: tree.ExpandAll(tree.GetRootItem()) def OnPullDownHighlight(self, evt): menuId = evt.GetMenuId() if menuId != -1: menu = evt.GetEventObject() help = menu.GetHelpString(menuId) self.SetStatusText(help) else: self.SetStatusText('') def OnUpdateUI(self, evt): if evt.GetId() in [wxID_CUT, wxID_COPY, self.ID_DELETE]: enable = tree.GetSelection().IsOk() and \ tree.GetSelection() != tree.GetRootItem() evt.Enable(enable) elif evt.GetId() == wxID_PASTE: enable = self.clipboard != None evt.Enable(enable) def OnIdle(self, evt): if tree.needUpdate: if conf.autoRefresh: if testWin: # (re)create tree.CreateTestWin(testWin.item) tree.needUpdate = false elif tree.pendingHighLight: tree.HighLight(tree.pendingHighLight) evt.Skip() def OnCloseWindow(self, evt): if not self.AskSave(): return if testWin: testWin.Destroy() conf.width, conf.height = self.GetSize() evt.Skip() def Clear(self): self.dataFile = '' self.clipboard = None self.modified = false panel.SetModified(false) panel.Clear() tree.Clear() global testWin if testWin: testWin.Destroy() testWin = None self.SetTitle(progname) def Open(self, path): # Try to read the file try: open(path) self.Clear() # Build wx tree dom = minidom.parse(path) tree.SetData(dom) self.dataFile = path self.SetTitle(progname + ': ' + os.path.basename(path)) except: wxLogError('Error reading file: ' + path) raise def Save(self, path): try: self.OnRefresh(wxCommandEvent()) memFile = MemoryFile(path) tree.dom.writexml(memFile) memFile.close() self.modified = false panel.SetModified(false) except: wxLogError('Error writing file: ' + path) raise def AskSave(self): if not (self.modified or panel.IsModified()): return true flags = wxICON_EXCLAMATION | wxYES_NO | wxCANCEL | wxCENTRE say = wxMessageDialog( self, 'File is modified. Save before exit?', 'Save before too late?', flags ).ShowModal() if say == wxID_YES: self.OnSaveOrSaveAs(wxCommandEvent(wxID_SAVE)) # If save was successful, modified flag is unset if not self.modified: return true elif say == wxID_NO: self.modified = false panel.SetModified(false) return true return false ################################################################################ class App(wxApp): def OnInit(self): self.SetAppName("xrced") # Settings global conf # !!! wxConfigBase_Get doesn't seem to work conf = wxConfig(style=wxCONFIG_USE_LOCAL_FILE) conf.autoRefresh = conf.ReadInt('autorefresh', true) size = conf.ReadInt('width', 800), conf.ReadInt('height', 600) # Add handlers wxFileSystem_AddHandler(wxMemoryFSHandler()) wxInitAllImageHandlers() # Create main frame global frame frame = self.frame = Frame(size) self.frame.Show(true) # Load resources from XRC file (!!! should be transformed to .py later) sys.modules['params'].frame = frame frame.res = wxXmlResource('') frame.res.Load(os.path.join(sys.path[0], 'xrced.xrc')) return true def OnExit(self): # Write config wc = wxConfigBase_Get() wc.WriteInt('autorefresh', conf.autoRefresh) wc.WriteInt('width', conf.width) wc.WriteInt('height', conf.height) wc.Flush() def main(): app = App() app.MainLoop() app.OnExit() if __name__ == '__main__': main()