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