don't crash on unterminated CDATA elements (patch #1117629)

[wxWidgets.git] / wxPython / samples / frogedit / FrogEdit.py

#!/usr/bin/env python

# simple text editor
#
# Copyright 2001 Adam Feuer and Steve Howell
#
# License: Python

import  os
import  re
import  string
import  sys

import  wx

from StatusBar           import *
from FrogEditor          import FrogEditor


ABOUT_TEXT = """FrogEdit : Copyright 2001 Adam Feuer and Steve Howell
wxEditor component : Copyright 1999 - 2001 Dirk Holtwic, Robin Dunn, Adam Feuer, Steve Howell

FrogEdit was built using Python, wxPython, and wxWindows."""


##---------------------------------------------------------------------

def chomp(line):
    line = string.split(line,'\n')[0]
    return string.split(line,'\r')[0]

##---------------------------------------------------------------------

class OutlinerPanel(wx.Panel):

    def Close(self, event):
        self.parent.Close()
        wx.Panel.Close(self)

##----------------------------------------------------------------------


class FrogEditFrame(wx.Frame):
    def __init__(self, parent, ID, title, pos=wx.DefaultPosition,
                 size=wx.DefaultSize, style=wx.DEFAULT_FRAME_STYLE):

        wx.Frame.__init__(self, parent, ID, title, pos, size, style)

        splitter = wx.SplitterWindow(self, -1, style=wx.NO_3D|wx.SP_3D)
        win = OutlinerPanel(splitter, -1, style=wx.CLIP_CHILDREN)
        win.parent = self
        log = self.MakeLogWindow(splitter)

        self.MakeStatusbar()
        self.MakeEditorWindow(win, log)
        self.SetUpSplitter(splitter, win, log)
        self.MakeMenus()
        self.MakeMainWindow(win)
        self.RegisterEventHandlers()
        self.InitVariables()

        win.Layout()


##------------- Init Misc

    def RegisterEventHandlers(self):
        self.Bind(wx.EVT_CLOSE,self.OnCloseWindow)

    def InitVariables(self):
        self.fileName = None
        self.edl.UnTouchBuffer()

    def MakeMenus(self):
        self.MainMenu = wx.MenuBar()
        self.AddMenus(self.MainMenu)
        self.SetMenuBar(self.MainMenu)

##------------- Init Subwindows

    def MakeEditorWindow(self, win, log):
        self.edl = FrogEditor(win, -1, style=wx.SUNKEN_BORDER, statusBar = self.sb)
        self.edl.SetControlFuncs = self.SetControlFuncs
        self.edl.SetAltFuncs = self.SetAltFuncs
        self.edl.SetStatus(log)

    def MakeStatusbar(self):
        self.sb = CustomStatusBar(self)
        self.SetStatusBar(self.sb)

    def MakeLogWindow(self, container):
        log = wx.TextCtrl(container, -1,
                         style = wx.TE_MULTILINE|wx.TE_READONLY|wx.HSCROLL)
        wx.Log_SetActiveTarget(wx.LogTextCtrl(log))
        wx.LogMessage('window handle: %s' % self.GetHandle())
        return log

    def SetUpSplitter(self, splitter, win, log):
        splitter.SplitHorizontally(win, log)
        splitter.SetSashPosition(360, True)
        splitter.SetMinimumPaneSize(40)

    def MakeToolbar(self, win):
        toolbarBox = wx.BoxSizer(wx.HORIZONTAL)
        self.AddButtons(win, toolbarBox)
        return toolbarBox

    def MakeMainWindow(self, win):
        mainBox = wx.BoxSizer(wx.VERTICAL)
        mainBox.Add(self.MakeToolbar(win))
        borderWidth = 5
        mainBox.Add(self.edl, 1, wx.ALL|wx.GROW, borderWidth)
        win.SetSizer(mainBox)
        win.SetAutoLayout(True)

##-------------- Init Menus

    # override this to add more menus
    def AddMenus(self, menu):
        self.AddFileMenu(menu)
        self.AddEditMenu(menu)
        self.AddHelpMenu(menu)

    def AddMenuItem(self, menu, itemText, itemDescription, itemHandler):
        menuId = wx.NewId()
        menu.Append(menuId, itemText, itemDescription)
        self.Bind(wx.EVT_MENU, itemHandler, id=menuId)
        return menuId

    def AddFileMenu(self, menu):
        fileMenu = wx.Menu()
        self.AddMenuItem(fileMenu, '&New File\tCtrl-N', 'New File', self.OnNewFile)
        self.AddMenuItem(fileMenu, '&Open File\tCtrl-O', 'Open File', self.OnOpenFile)
        self.AddMenuItem(fileMenu, '&Save File\tCtrl-S', 'Save File', self.OnSaveFile)
        self.AddMenuItem(fileMenu, 'Save File &As\tCtrl-A', 'Save File As',self.OnSaveFileAs)
        self.AddMenuItem(fileMenu, 'E&xit\tAlt-X', 'Exit', self.OnFileExit)
        menu.Append(fileMenu, 'File')

    def AddEditMenu(self, menu):
        editMenu = wx.Menu()
        self.AddMenuItem(editMenu, 'Cut\tCtrl-X', 'Cut', self.edl.OnCutSelection)
        self.AddMenuItem(editMenu, '&Copy\tCtrl-C', 'Copy', self.edl.OnCopySelection)
        self.AddMenuItem(editMenu, 'Paste\tCtrl-V', 'Paste', self.edl.OnPaste)
        self.AddMenuItem(editMenu, 'Edit preferences', 'Edit Preferences', self.OnEditPreferences)
        menu.Append(editMenu, 'Edit')

    def AddHelpMenu(self, menu):
        helpMenu = wx.Menu()
        self.AddMenuItem(helpMenu, 'About', 'About the program', self.OnHelpAbout)
        menu.Append(helpMenu, 'Help')

##---------------- Init Buttons


    def NewButton(self, window, container, name, pos, size, handler):
        buttonId = wx.NewId()
        if pos == None or size == None:
            container.Add(wx.Button(window, buttonId, name), 0, 0)
        else:
            container.Add(wx.Button(window, buttonId, name, pos, size), 0, 0)
        self.Bind(wx.EVT_BUTTON, handler, id=buttonId)
        return buttonId

    # override this to make more buttons
    def AddButtons(self, window, container):
        buttonPos = None
        buttonSize = None
        self.NewButton(window, container, "New", buttonPos, buttonSize, self.OnNewFile)
        self.NewButton(window, container, "Open", buttonPos, buttonSize, self.OnOpenFile)
        self.NewButton(window, container, "Save", buttonPos, buttonSize, self.OnSaveFile)


##-------------- Init Dialogs

    def MessageDialog(self, text, title):
        messageDialog = wx.MessageDialog(self, text, title, wx.OK | wx.ICON_INFORMATION)
        messageDialog.ShowModal()
        messageDialog.Destroy()

    def OkCancelDialog(self, text, title):
        dialog = wx.MessageDialog(self, text, title, wx.OK | wx.CANCEL | wx.ICON_INFORMATION)
        result = dialog.ShowModal()
        dialog.Destroy()
        if result == wx.ID_OK:
            return True
        else:
            return False

    def SelectFileDialog(self, defaultDir=None, defaultFile=None, wildCard=None):
        if defaultDir == None:
            defaultDir = "."
        if defaultFile == None:
            defaultFile = ""
        if wildCard == None:
            wildCard = "*.*"
        fileName = None
        fileDialog = wx.FileDialog(self, "Choose a file", defaultDir, defaultFile, wildCard, wx.OPEN|wx.MULTIPLE)
        result = fileDialog.ShowModal()
        if result == wx.ID_OK:
            fileName = fileDialog.GetPath()
            wx.LogMessage('You selected: %s\n' % fileName)
        fileDialog.Destroy()
        return fileName

    def OpenFileError(self, fileName):
        wx.LogMessage('Open file error.')
        self.MessageDialog("Error opening file '%s'!" % fileName, "Error")


    def SaveFileError(self, fileName):
        wx.LogMessage('Save file error.')
        self.MessageDialog("Error saving file '%s'!" % fileName, "Error")

##---------------- Utility functions


    def SetControlFuncs(self, action):
        "for overriding editor's keys"
        FrogEditor.SetControlFuncs(self.edl, action)
        action['a'] = self.OnSaveFileAs
        action['o'] = self.OnOpenFile
        action['n'] = self.OnNewFile
        action['s'] = self.OnSaveFile

    def SetAltFuncs(self, action):
        FrogEditor.SetAltFuncs(self.edl, action)
        action['x'] = self.OnFileExit

    def GetCurrentDir(self):
        if self.fileName is not None:
            return os.path.split(self.fileName)[0]
        return "."

    def GetFileName(self):
        if self.fileName is not None:
            return os.path.split(self.fileName)[1]
        return ""

    def NewFile(self):
        self.edl.SetText([""])
        self.fileName = None
        self.sb.setFileName("")

    def SaveFile(self, fileName):
        try:
            contents = string.join(self.edl.GetText(), '\n')
            f = open(fileName, 'w')
            f.write(contents)
            f.close()
            self.edl.UnTouchBuffer()
            self.sb.setFileName(fileName)
            return True
        except:
            return False

    def OpenFile(self, fileName):
        try:
            f = open(fileName, 'r')
            contents = f.readlines()
            f.close()
            contents = [chomp(line) for line in contents]
            if len(contents) == 0:
                contents = [""]
            self.edl.SetText(contents)
            self.fileName = fileName
            self.sb.setFileName(fileName)
            return True
        except:
            return False



##---------------- Event handlers


    def OnCloseWindow(self, event):
        self.edl.OnCloseWindow(event)
        self.Destroy()

    def OnNewFile(self, event):
        if self.edl.BufferWasTouched():
            if not self.OkCancelDialog("New file - abandon changes?", "New File"):
                return
        self.NewFile()
        self.edl.SetFocus()

    def OnOpenFile(self, event):
        if self.edl.BufferWasTouched():
            if not self.OkCancelDialog("Open file - abandon changes?", "Open File"):
                return
        fileName = self.SelectFileDialog(self.GetCurrentDir())
        if fileName is not None:
            if self.OpenFile(fileName) is False:
                self.OpenFileError(fileName)
        self.edl.SetFocus()

    def OnSaveFile(self, event):
        if self.fileName is None:
            return self.OnSaveFileAs(event)
        wx.LogMessage("Saving %s..." % self.fileName)
        if self.SaveFile(self.fileName) is not True:
            self.SaveFileError(self.fileName)
        self.edl.SetFocus()

    def OnSaveFileAs(self, event):
        fileName = self.SelectFileDialog(self.GetCurrentDir(),self.GetFileName())
        if fileName is not None:
            self.fileName = fileName
            wx.LogMessage("Saving %s..." % self.fileName)
            if self.SaveFile(self.fileName) is not True:
                self.SaveFileError(self.fileName)
        self.edl.SetFocus()

    def OnFileExit(self, event):
        if self.edl.BufferWasTouched():
            if not self.OkCancelDialog("Exit program - abandon changes?", "Exit"):
                return
        self.OnCloseWindow(event)

    def OnEditPreferences(self, event):
        self.MessageDialog("Edit preferences is not implemented yet.", "Not implemented.")
        pass

    def OnHelpAbout(self, event):
        self.MessageDialog(ABOUT_TEXT, "About FrogEdit")
        pass

    def Show(self, show):
        wx.Frame.Show(self, show)
        self.edl.SetFocus()

##------------- Startup stuff

    def LoadInitialFile(self, fileName):
        if fileName is not None:
            if self.OpenFile(fileName) is False:
                self.OpenFileError(fileName)




##-------------- Application Launcher utility class

class FrogEditLauncher:

    def MakeAppFrame(self):
        return FrogEditFrame(None, -1, "FrogEdit", size=(640, 480),
                             style=wx.DEFAULT_FRAME_STYLE|wx.NO_FULL_REPAINT_ON_RESIZE)

    def GetArgvFilename(self):
        if len(sys.argv) > 1:
            return sys.argv[1]
        else:
            return None

    def Main(self):
        app = wx.PySimpleApp()
        win = self.MakeAppFrame()
        win.Show(True)
        win.LoadInitialFile(self.GetArgvFilename())
        app.MainLoop()


##-------------- Main program


if __name__ == '__main__':

    launcher = FrogEditLauncher()
    launcher.Main()