wxPython/samples/frogedit/FrogEdit.py

   1 #!/usr/bin/env python
   2
   3 # simple text editor
   4 #
   5 # Copyright 2001 Adam Feuer and Steve Howell
   6 #
   7 # License: Python
   8
   9 import os, string, re
  10 import sys
  11 from wxPython.wx         import *
  12 from StatusBar           import *
  13 from FrogEditor          import FrogEditor
  14
  15 TRUE  = 1
  16 FALSE = 0
  17
  18 ABOUT_TEXT = """FrogEdit : Copyright 2001 Adam Feuer and Steve Howell
  19 wxEditor component : Copyright 1999 - 2001 Dirk Holtwic, Robin Dunn, Adam Feuer, Steve Howell
  20
  21 FrogEdit was built using Python, wxPython, and wxWindows."""
  22
  23
  24 ##---------------------------------------------------------------------
  25
  26 def chomp(line):
  27     line = string.split(line,'\n')[0]
  28     return string.split(line,'\r')[0]
  29
  30 ##---------------------------------------------------------------------
  31
  32 class OutlinerPanel(wxPanel):
  33
  34     def Close(self, event):
  35         self.parent.Close()
  36         wxPanel.Close(self)
  37
  38 ##----------------------------------------------------------------------
  39
  40
  41 class FrogEditFrame(wxFrame):
  42     def __init__(self, parent, ID, title, pos=wxDefaultPosition,
  43                  size=wxDefaultSize, style=wxDEFAULT_FRAME_STYLE):
  44
  45         wxFrame.__init__(self, parent, ID, title, pos, size, style)
  46
  47         splitter = wxSplitterWindow(self, -1, style=wxNO_3D|wxSP_3D)
  48         win = OutlinerPanel(splitter, -1, style=wxCLIP_CHILDREN)
  49         win.parent = self
  50         log = self.MakeLogWindow(splitter)
  51
  52         self.MakeStatusbar()
  53         self.MakeEditorWindow(win, log)
  54         self.SetUpSplitter(splitter, win, log)
  55         self.MakeMenus()
  56         self.MakeMainWindow(win)
  57         self.RegisterEventHandlers()
  58         self.InitVariables()
  59
  60         win.Layout()
  61
  62
  63 ##------------- Init Misc
  64
  65     def RegisterEventHandlers(self):
  66         EVT_CLOSE(self,self.OnCloseWindow)
  67
  68     def InitVariables(self):
  69         self.fileName = None
  70         self.edl.UnTouchBuffer()
  71
  72     def MakeMenus(self):
  73         self.MainMenu = wxMenuBar()
  74         self.AddMenus(self.MainMenu)
  75         self.SetMenuBar(self.MainMenu)
  76
  77 ##------------- Init Subwindows
  78
  79     def MakeEditorWindow(self, win, log):
  80         self.edl = FrogEditor(win, -1, style=wxSUNKEN_BORDER, statusBar = self.sb)
  81         self.edl.SetControlFuncs = self.SetControlFuncs
  82         self.edl.SetAltFuncs = self.SetAltFuncs
  83         self.edl.SetStatus(log)
  84
  85     def MakeStatusbar(self):
  86         self.sb = CustomStatusBar(self)
  87         self.SetStatusBar(self.sb)
  88
  89     def MakeLogWindow(self, container):
  90         log = wxTextCtrl(container, -1,
  91                          style = wxTE_MULTILINE|wxTE_READONLY|wxHSCROLL)
  92         wxLog_SetActiveTarget(wxLogTextCtrl(log))
  93         wxLogMessage('window handle: %s' % self.GetHandle())
  94         return log
  95
  96     def SetUpSplitter(self, splitter, win, log):
  97         splitter.SplitHorizontally(win, log)
  98         splitter.SetSashPosition(360, true)
  99         splitter.SetMinimumPaneSize(40)
 100
 101     def MakeToolbar(self, win):
 102         toolbarBox = wxBoxSizer(wxHORIZONTAL)
 103         self.AddButtons(win, toolbarBox)
 104         return toolbarBox
 105
 106     def MakeMainWindow(self, win):
 107         mainBox = wxBoxSizer(wxVERTICAL)
 108         mainBox.Add(self.MakeToolbar(win))
 109         borderWidth = 5
 110         mainBox.Add(self.edl, 1, wxALL|wxGROW, borderWidth)
 111         win.SetSizer(mainBox)
 112         win.SetAutoLayout(true)
 113
 114 ##-------------- Init Menus
 115
 116     # override this to add more menus
 117     def AddMenus(self, menu):
 118         self.AddFileMenu(menu)
 119         self.AddEditMenu(menu)
 120         self.AddHelpMenu(menu)
 121
 122     def AddMenuItem(self, menu, itemText, itemDescription, itemHandler):
 123         menuId = wxNewId()
 124         menu.Append(menuId, itemText, itemDescription)
 125         EVT_MENU(self, menuId, itemHandler)
 126         return menuId
 127
 128     def AddFileMenu(self, menu):
 129         fileMenu = wxMenu()
 130         self.AddMenuItem(fileMenu, '&New File\tCtrl-N', 'New File', self.OnNewFile)
 131         self.AddMenuItem(fileMenu, '&Open File\tCtrl-O', 'Open File', self.OnOpenFile)
 132         self.AddMenuItem(fileMenu, '&Save File\tCtrl-S', 'Save File', self.OnSaveFile)
 133         self.AddMenuItem(fileMenu, 'Save File &As\tCtrl-A', 'Save File As',self.OnSaveFileAs)
 134         self.AddMenuItem(fileMenu, 'E&xit\tAlt-X', 'Exit', self.OnFileExit)
 135         menu.Append(fileMenu, 'File')
 136
 137     def AddEditMenu(self, menu):
 138         editMenu = wxMenu()
 139         self.AddMenuItem(editMenu, 'Cut\tCtrl-X', 'Cut', self.edl.OnCutSelection)
 140         self.AddMenuItem(editMenu, '&Copy\tCtrl-C', 'Copy', self.edl.OnCopySelection)
 141         self.AddMenuItem(editMenu, 'Paste\tCtrl-V', 'Paste', self.edl.OnPaste)
 142         self.AddMenuItem(editMenu, 'Edit preferences', 'Edit Preferences', self.OnEditPreferences)
 143         menu.Append(editMenu, 'Edit')
 144
 145     def AddHelpMenu(self, menu):
 146         helpMenu = wxMenu()
 147         self.AddMenuItem(helpMenu, 'About', 'About the program', self.OnHelpAbout)
 148         menu.Append(helpMenu, 'Help')
 149
 150 ##---------------- Init Buttons
 151
 152
 153     def NewButton(self, window, container, name, pos, size, handler):
 154         buttonId = wxNewId()
 155         if pos == None or size == None:
 156             container.Add(wxButton(window, buttonId, name), 0, 0)
 157         else:
 158             container.Add(wxButton(window, buttonId, name, pos, size), 0, 0)
 159         EVT_BUTTON(self, buttonId, handler)
 160         return buttonId
 161
 162     # override this to make more buttons
 163     def AddButtons(self, window, container):
 164         buttonPos = None
 165         buttonSize = None
 166         self.NewButton(window, container, "New", buttonPos, buttonSize, self.OnNewFile)
 167         self.NewButton(window, container, "Open", buttonPos, buttonSize, self.OnOpenFile)
 168         self.NewButton(window, container, "Save", buttonPos, buttonSize, self.OnSaveFile)
 169
 170
 171 ##-------------- Init Dialogs
 172
 173     def MessageDialog(self, text, title):
 174         messageDialog = wxMessageDialog(self, text, title, wxOK | wxICON_INFORMATION)
 175         messageDialog.ShowModal()
 176         messageDialog.Destroy()
 177
 178     def OkCancelDialog(self, text, title):
 179         dialog = wxMessageDialog(self, text, title, wxOK | wxCANCEL | wxICON_INFORMATION)
 180         result = dialog.ShowModal()
 181         dialog.Destroy()
 182         if result == wxID_OK:
 183             return TRUE
 184         else:
 185             return FALSE
 186
 187     def SelectFileDialog(self, defaultDir=None, defaultFile=None, wildCard=None):
 188         if defaultDir == None:
 189             defaultDir = "."
 190         if defaultFile == None:
 191             defaultFile = ""
 192         if wildCard == None:
 193             wildCard = "*.*"
 194         fileName = None
 195         fileDialog = wxFileDialog(self, "Choose a file", defaultDir, defaultFile, wildCard, wxOPEN|wxMULTIPLE)
 196         result = fileDialog.ShowModal()
 197         if result == wxID_OK:
 198             fileName = fileDialog.GetPath()
 199             wxLogMessage('You selected: %s\n' % fileName)
 200         fileDialog.Destroy()
 201         return fileName
 202
 203     def OpenFileError(self, fileName):
 204         wxLogMessage('Open file error.')
 205         self.MessageDialog("Error opening file '%s'!" % fileName, "Error")
 206
 207
 208     def SaveFileError(self, fileName):
 209         wxLogMessage('Save file error.')
 210         self.MessageDialog("Error saving file '%s'!" % fileName, "Error")
 211
 212 ##---------------- Utility functions
 213
 214
 215     def SetControlFuncs(self, action):
 216         "for overriding editor's keys"
 217         FrogEditor.SetControlFuncs(self.edl, action)
 218         action['a'] = self.OnSaveFileAs
 219         action['o'] = self.OnOpenFile
 220         action['n'] = self.OnNewFile
 221         action['s'] = self.OnSaveFile
 222
 223     def SetAltFuncs(self, action):
 224         FrogEditor.SetAltFuncs(self.edl, action)
 225         action['x'] = self.OnFileExit
 226
 227     def GetCurrentDir(self):
 228         if self.fileName is not None:
 229             return os.path.split(self.fileName)[0]
 230         return "."
 231
 232     def GetFileName(self):
 233         if self.fileName is not None:
 234             return os.path.split(self.fileName)[1]
 235         return ""
 236
 237     def NewFile(self):
 238         self.edl.SetText([""])
 239         self.fileName = None
 240         self.sb.setFileName("")
 241
 242     def SaveFile(self, fileName):
 243         try:
 244             contents = string.join(self.edl.GetText(), '\n')
 245             f = open(fileName, 'w')
 246             f.write(contents)
 247             f.close()
 248             self.edl.UnTouchBuffer()
 249             self.sb.setFileName(fileName)
 250             return TRUE
 251         except:
 252             return FALSE
 253
 254     def OpenFile(self, fileName):
 255         try:
 256             f = open(fileName, 'r')
 257             contents = f.readlines()
 258             f.close()
 259             contents = [chomp(line) for line in contents]
 260             if len(contents) == 0:
 261                 contents = [""]
 262             self.edl.SetText(contents)
 263             self.fileName = fileName
 264             self.sb.setFileName(fileName)
 265             return TRUE
 266         except:
 267             return FALSE
 268
 269
 270
 271 ##---------------- Event handlers
 272
 273
 274     def OnCloseWindow(self, event):
 275         self.edl.OnCloseWindow(event)
 276         self.Destroy()
 277
 278     def OnNewFile(self, event):
 279         if self.edl.BufferWasTouched():
 280             if not self.OkCancelDialog("New file - abandon changes?", "New File"):
 281                 return
 282         self.NewFile()
 283         self.edl.SetFocus()
 284
 285     def OnOpenFile(self, event):
 286         if self.edl.BufferWasTouched():
 287             if not self.OkCancelDialog("Open file - abandon changes?", "Open File"):
 288                 return
 289         fileName = self.SelectFileDialog(self.GetCurrentDir())
 290         if fileName is not None:
 291             if self.OpenFile(fileName) is FALSE:
 292                 self.OpenFileError(fileName)
 293         self.edl.SetFocus()
 294
 295     def OnSaveFile(self, event):
 296         if self.fileName is None:
 297             return self.OnSaveFileAs(event)
 298         wxLogMessage("Saving %s..." % self.fileName)
 299         if self.SaveFile(self.fileName) is not TRUE:
 300             self.SaveFileError(self.fileName)
 301         self.edl.SetFocus()
 302
 303     def OnSaveFileAs(self, event):
 304         fileName = self.SelectFileDialog(self.GetCurrentDir(),self.GetFileName())
 305         if fileName is not None:
 306             self.fileName = fileName
 307             wxLogMessage("Saving %s..." % self.fileName)
 308             if self.SaveFile(self.fileName) is not TRUE:
 309                 self.SaveFileError(self.fileName)
 310         self.edl.SetFocus()
 311
 312     def OnFileExit(self, event):
 313         if self.edl.BufferWasTouched():
 314             if not self.OkCancelDialog("Exit program - abandon changes?", "Exit"):
 315                 return
 316         self.OnCloseWindow(event)
 317
 318     def OnEditPreferences(self, event):
 319         self.MessageDialog("Edit preferences is not implemented yet.", "Not implemented.")
 320         pass
 321
 322     def OnHelpAbout(self, event):
 323         self.MessageDialog(ABOUT_TEXT, "About FrogEdit")
 324         pass
 325
 326     def Show(self, show):
 327         wxFrame.Show(self, show)
 328         self.edl.SetFocus()
 329
 330 ##------------- Startup stuff
 331
 332     def LoadInitialFile(self, fileName):
 333         if fileName is not None:
 334             if self.OpenFile(fileName) is FALSE:
 335                 self.OpenFileError(fileName)
 336
 337
 338
 339
 340 ##-------------- Application Launcher utility class
 341
 342 class FrogEditLauncher:
 343
 344     def MakeAppFrame(self):
 345         return FrogEditFrame(None, -1, "FrogEdit", size=(640, 480),
 346                              style=wxDEFAULT_FRAME_STYLE|wxNO_FULL_REPAINT_ON_RESIZE)
 347
 348     def GetArgvFilename(self):
 349         if len(sys.argv) > 1:
 350             return sys.argv[1]
 351         else:
 352             return None
 353
 354     def Main(self):
 355         app = wxPySimpleApp()
 356         win = self.MakeAppFrame()
 357         win.Show(true)
 358         win.LoadInitialFile(self.GetArgvFilename())
 359         app.MainLoop()
 360
 361
 362 ##-------------- Main program
 363
 364
 365 if __name__ == '__main__':
 366
 367     launcher = FrogEditLauncher()
 368     launcher.Main()