Fixed Python editor adding variable panel
authorLaurent Bessard
Wed, 08 May 2013 22:52:55 +0200
changeset 1097 233681f2a00e
parent 1096 c9ace6a881c9
child 1098 0ead830daf26
Fixed Python editor adding variable panel
c_ext/CFileEditor.py
c_ext/c_ext.py
editors/CodeFileEditor.py
py_ext/PythonEditor.py
py_ext/PythonFileCTNMixin.py
--- a/c_ext/CFileEditor.py	Wed May 08 21:37:31 2013 +0200
+++ b/c_ext/CFileEditor.py	Wed May 08 22:52:55 2013 +0200
@@ -48,3 +48,6 @@
         CodeFileEditor.RefreshView(self)
         
         self.CCodeEditor.RefreshView()
+
+    def Find(self, direction, search_params):
+        self.CCodeEditor.Find(direction, search_params)
--- a/c_ext/c_ext.py	Wed May 08 21:37:31 2013 +0200
+++ b/c_ext/c_ext.py	Wed May 08 22:52:55 2013 +0200
@@ -50,7 +50,6 @@
         text += '#include "iec_types_all.h"'
 
         # Adding variables
-        base_types = self.GetCTRoot().GetBaseTypes()
         config = self.GetCTRoot().GetProjectConfigNames()[0]
         text += "/* User variables reference */\n"
         for variable in self.CodeFile.variables.variable:
--- a/editors/CodeFileEditor.py	Wed May 08 21:37:31 2013 +0200
+++ b/editors/CodeFileEditor.py	Wed May 08 22:52:55 2013 +0200
@@ -5,14 +5,23 @@
 import wx.stc as stc
 import wx.lib.buttons
 
+from plcopen.plcopen import TestTextElement
 from controls import CustomGrid, CustomTable
 from editors.ConfTreeNodeEditor import ConfTreeNodeEditor
 from util.BitmapLibrary import GetBitmap
 from controls.CustomStyledTextCtrl import CustomStyledTextCtrl, faces, GetCursorPos
+from graphics.GraphicCommons import ERROR_HIGHLIGHT, SEARCH_RESULT_HIGHLIGHT, REFRESH_HIGHLIGHT_PERIOD
 
 SECTIONS_NAMES = ["Includes", "Globals", "Init",
                   "CleanUp", "Retrieve", "Publish"]
 
+[STC_CODE_ERROR, STC_CODE_SEARCH_RESULT] = range(15, 17)
+
+HIGHLIGHT_TYPES = {
+    ERROR_HIGHLIGHT: STC_CODE_ERROR,
+    SEARCH_RESULT_HIGHLIGHT: STC_CODE_SEARCH_RESULT,
+}
+
 class CodeEditor(CustomStyledTextCtrl):
     
     KEYWORDS = []
@@ -71,6 +80,10 @@
         self.StyleSetSpec(stc.STC_STYLE_BRACELIGHT,  "fore:#FFFFFF,back:#0000FF,bold")
         self.StyleSetSpec(stc.STC_STYLE_BRACEBAD,    "fore:#000000,back:#FF0000,bold")
         
+        # Highlighting styles
+        self.StyleSetSpec(STC_CODE_ERROR, 'fore:#FF0000,back:#FFFF00,size:%(size)d' % faces)
+        self.StyleSetSpec(STC_CODE_SEARCH_RESULT, 'fore:#FFFFFF,back:#FFA500,size:%(size)d' % faces)
+        
         # register some images for use in the AutoComplete box.
         #self.RegisterImage(1, images.getSmilesBitmap())
         self.RegisterImage(1, 
@@ -93,6 +106,14 @@
         self.DisableEvents = True
         self.CurrentAction = None
         
+        self.Highlights = []
+        self.SearchParams = None
+        self.SearchResults = None
+        self.CurrentFindHighlight = None
+        
+        self.RefreshHighlightsTimer = wx.Timer(self, -1)
+        self.Bind(wx.EVT_TIMER, self.OnRefreshHighlightsTimer, self.RefreshHighlightsTimer)
+        
         self.SectionsComments = {}
         for section in SECTIONS_NAMES:
             section_start_comment = "%s %s section\n" % (self.COMMENT_HEADER, section)
@@ -194,6 +215,8 @@
         self.DisableEvents = False
         
         self.Colourise(0, -1)
+        
+        self.ShowHighlights()
 
     def DoGetBestSize(self):
         return self.ParentWindow.GetPanelBestSize()
@@ -375,6 +398,91 @@
         self.RefreshModel()
         self.RefreshBuffer()
 
+    def Find(self, direction, search_params):
+        if self.SearchParams != search_params:
+            self.ClearHighlights(SEARCH_RESULT_HIGHLIGHT)
+            
+            self.SearchParams = search_params
+            criteria = {
+                "raw_pattern": search_params["find_pattern"], 
+                "pattern": re.compile(search_params["find_pattern"]),
+                "case_sensitive": search_params["case_sensitive"],
+                "regular_expression": search_params["regular_expression"],
+                "filter": "all"}
+            
+            self.SearchResults = [
+                (start, end, SEARCH_RESULT_HIGHLIGHT)
+                for start, end, text in 
+                TestTextElement(self.GetText(), criteria)]
+            self.CurrentFindHighlight = None
+        
+        if len(self.SearchResults) > 0:
+            if self.CurrentFindHighlight is not None:
+                old_idx = self.SearchResults.index(self.CurrentFindHighlight)
+                if self.SearchParams["wrap"]:
+                    idx = (old_idx + direction) % len(self.SearchResults)
+                else:
+                    idx = max(0, min(old_idx + direction, len(self.SearchResults) - 1))
+                if idx != old_idx:
+                    self.RemoveHighlight(*self.CurrentFindHighlight)
+                    self.CurrentFindHighlight = self.SearchResults[idx]
+                    self.AddHighlight(*self.CurrentFindHighlight)
+            else:
+                self.CurrentFindHighlight = self.SearchResults[0]
+                self.AddHighlight(*self.CurrentFindHighlight)
+            
+        else:
+            if self.CurrentFindHighlight is not None:
+                self.RemoveHighlight(*self.CurrentFindHighlight)
+            self.CurrentFindHighlight = None
+
+#-------------------------------------------------------------------------------
+#                        Highlights showing functions
+#-------------------------------------------------------------------------------
+
+    def OnRefreshHighlightsTimer(self, event):
+        self.RefreshView()
+        event.Skip()
+
+    def ClearHighlights(self, highlight_type=None):
+        if highlight_type is None:
+            self.Highlights = []
+        else:
+            highlight_type = HIGHLIGHT_TYPES.get(highlight_type, None)
+            if highlight_type is not None:
+                self.Highlights = [(start, end, highlight) for (start, end, highlight) in self.Highlights if highlight != highlight_type]
+        self.RefreshView()
+
+    def AddHighlight(self, start, end, highlight_type):
+        highlight_type = HIGHLIGHT_TYPES.get(highlight_type, None)
+        if highlight_type is not None:
+            self.Highlights.append((start, end, highlight_type))
+            self.GotoPos(self.PositionFromLine(start[0]) + start[1])
+            self.RefreshHighlightsTimer.Start(int(REFRESH_HIGHLIGHT_PERIOD * 1000), oneShot=True)
+            self.RefreshView()
+
+    def RemoveHighlight(self, start, end, highlight_type):
+        highlight_type = HIGHLIGHT_TYPES.get(highlight_type, None)
+        if (highlight_type is not None and 
+            (start, end, highlight_type) in self.Highlights):
+            self.Highlights.remove((start, end, highlight_type))
+            self.RefreshHighlightsTimer.Start(int(REFRESH_HIGHLIGHT_PERIOD * 1000), oneShot=True)
+    
+    def ShowHighlights(self):
+        for start, end, highlight_type in self.Highlights:
+            if start[0] == 0:
+                highlight_start_pos = start[1]
+            else:
+                highlight_start_pos = self.GetLineEndPosition(start[0] - 1) + start[1] + 1
+            if end[0] == 0:
+                highlight_end_pos = end[1] - indent + 1
+            else:
+                highlight_end_pos = self.GetLineEndPosition(end[0] - 1) + end[1] + 2
+            self.StartStyling(highlight_start_pos, 0xff)
+            self.SetStyling(highlight_end_pos - highlight_start_pos, highlight_type)
+            self.StartStyling(highlight_start_pos, 0x00)
+            self.SetStyling(len(self.GetText()) - highlight_end_pos, stc.STC_STYLE_DEFAULT)
+
 
 #-------------------------------------------------------------------------------
 #                         Helper for VariablesGrid values
@@ -429,13 +537,7 @@
         
         main_sizer = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=4)
         main_sizer.AddGrowableCol(0)
-        main_sizer.AddGrowableRow(0)
-        
-        self.VariablesGrid = CustomGrid(self, size=wx.Size(-1, 300), style=wx.VSCROLL)
-        self.VariablesGrid.Bind(wx.grid.EVT_GRID_CELL_CHANGE, self.OnVariablesGridCellChange)
-        self.VariablesGrid.Bind(wx.grid.EVT_GRID_CELL_LEFT_CLICK, self.OnVariablesGridCellLeftClick)
-        self.VariablesGrid.Bind(wx.grid.EVT_GRID_EDITOR_SHOWN, self.OnVariablesGridEditorShown)
-        main_sizer.AddWindow(self.VariablesGrid, flag=wx.GROW)
+        main_sizer.AddGrowableRow(1)
         
         controls_sizer = wx.BoxSizer(wx.HORIZONTAL)
         main_sizer.AddSizer(controls_sizer, border=5, flag=wx.TOP|wx.ALIGN_RIGHT)
@@ -449,7 +551,13 @@
                   size=wx.Size(28, 28), style=wx.NO_BORDER)
             button.SetToolTipString(help)
             setattr(self, name, button)
-            controls_sizer.AddWindow(button, border=5, flag=wx.LEFT)
+            controls_sizer.AddWindow(button, border=5, flag=wx.RIGHT)
+        
+        self.VariablesGrid = CustomGrid(self, size=wx.Size(-1, 300), style=wx.VSCROLL)
+        self.VariablesGrid.Bind(wx.grid.EVT_GRID_CELL_CHANGE, self.OnVariablesGridCellChange)
+        self.VariablesGrid.Bind(wx.grid.EVT_GRID_CELL_LEFT_CLICK, self.OnVariablesGridCellLeftClick)
+        self.VariablesGrid.Bind(wx.grid.EVT_GRID_EDITOR_SHOWN, self.OnVariablesGridEditorShown)
+        main_sizer.AddWindow(self.VariablesGrid, flag=wx.GROW)
         
         self.SetSizer(main_sizer)
                 
@@ -531,7 +639,7 @@
                 self.Bind(wx.EVT_MENU, self.GetVariableTypeFunction(base_type), id=new_id)
             type_menu.AppendMenu(wx.NewId(), "Base Types", base_menu)
             datatype_menu = wx.Menu(title='')
-            for datatype in self.Controler.GetDataTypes(basetypes=False, only_locatables=True):
+            for datatype in self.Controler.GetDataTypes():
                 new_id = wx.NewId()
                 datatype_menu.Append(help='', id=new_id, kind=wx.ITEM_NORMAL, text=datatype)
                 self.Bind(wx.EVT_MENU, self.GetVariableTypeFunction(datatype), id=new_id)
--- a/py_ext/PythonEditor.py	Wed May 08 21:37:31 2013 +0200
+++ b/py_ext/PythonEditor.py	Wed May 08 22:52:55 2013 +0200
@@ -1,554 +1,70 @@
-import re
 import keyword
-
-import wx
-import wx.grid
 import wx.stc as stc
 
-from plcopen.plcopen import TestTextElement
-from graphics.GraphicCommons import ERROR_HIGHLIGHT, SEARCH_RESULT_HIGHLIGHT, REFRESH_HIGHLIGHT_PERIOD
-from editors.ConfTreeNodeEditor import ConfTreeNodeEditor
-from controls.CustomStyledTextCtrl import CustomStyledTextCtrl, faces, GetCursorPos
+from controls.CustomStyledTextCtrl import faces
+from editors.CodeFileEditor import CodeFileEditor, CodeEditor
 
-[STC_PYTHON_ERROR, STC_PYTHON_SEARCH_RESULT] = range(15, 17)
+class PythonCodeEditor(CodeEditor):
 
-HIGHLIGHT_TYPES = {
-    ERROR_HIGHLIGHT: STC_PYTHON_ERROR,
-    SEARCH_RESULT_HIGHLIGHT: STC_PYTHON_SEARCH_RESULT,
-}
+    KEYWORDS = keyword.kwlist
+    COMMENT_HEADER = "##"
+    
+    def SetCodeLexer(self):
+        self.SetLexer(stc.STC_LEX_PYTHON)
+        
+        # Line numbers in margin
+        self.StyleSetSpec(stc.STC_STYLE_LINENUMBER,'fore:#000000,back:#99A9C2,size:%(size)d' % faces)    
+        # Highlighted brace
+        self.StyleSetSpec(stc.STC_STYLE_BRACELIGHT,'fore:#00009D,back:#FFFF00,size:%(size)d' % faces)
+        # Unmatched brace
+        self.StyleSetSpec(stc.STC_STYLE_BRACEBAD,'fore:#00009D,back:#FF0000,size:%(size)d' % faces)
+        # Indentation guide
+        self.StyleSetSpec(stc.STC_STYLE_INDENTGUIDE, 'fore:#CDCDCD,size:%(size)d' % faces)
 
-[ID_PYTHONEDITOR,
-] = [wx.NewId() for _init_ctrls in range(1)]
+        # Python styles
+        self.StyleSetSpec(stc.STC_P_DEFAULT, 'fore:#000000,size:%(size)d' % faces)
+        # Comments
+        self.StyleSetSpec(stc.STC_P_COMMENTLINE,  'fore:#008000,back:#F0FFF0,size:%(size)d' % faces)
+        self.StyleSetSpec(stc.STC_P_COMMENTBLOCK, 'fore:#008000,back:#F0FFF0,size:%(size)d' % faces)
+        # Numbers
+        self.StyleSetSpec(stc.STC_P_NUMBER, 'fore:#008080,size:%(size)d' % faces)
+        # Strings and characters
+        self.StyleSetSpec(stc.STC_P_STRING, 'fore:#800080,size:%(size)d' % faces)
+        self.StyleSetSpec(stc.STC_P_CHARACTER, 'fore:#800080,size:%(size)d' % faces)
+        # Keywords
+        self.StyleSetSpec(stc.STC_P_WORD, 'fore:#000080,bold,size:%(size)d' % faces)
+        # Triple quotes
+        self.StyleSetSpec(stc.STC_P_TRIPLE, 'fore:#800080,back:#FFFFEA,size:%(size)d' % faces)
+        self.StyleSetSpec(stc.STC_P_TRIPLEDOUBLE, 'fore:#800080,back:#FFFFEA,size:%(size)d' % faces)
+        # Class names
+        self.StyleSetSpec(stc.STC_P_CLASSNAME, 'fore:#0000FF,bold,size:%(size)d' % faces)
+        # Function names
+        self.StyleSetSpec(stc.STC_P_DEFNAME, 'fore:#008080,bold,size:%(size)d' % faces)
+        # Operators
+        self.StyleSetSpec(stc.STC_P_OPERATOR, 'fore:#800000,bold,size:%(size)d' % faces)
+        # Identifiers. I leave this as not bold because everything seems
+        # to be an identifier if it doesn't match the above criterae
+        self.StyleSetSpec(stc.STC_P_IDENTIFIER, 'fore:#000000,size:%(size)d' % faces)
+        
 
-class PythonEditor(ConfTreeNodeEditor):
+#-------------------------------------------------------------------------------
+#                          CFileEditor Main Frame Class
+#-------------------------------------------------------------------------------
 
-    fold_symbols = 3
-    CONFNODEEDITOR_TABS = [
+class PythonEditor(CodeFileEditor):
+    
+    CONFNODEEDITOR_TABS = CodeFileEditor.CONFNODEEDITOR_TABS + [
         (_("Python code"), "_create_PythonCodeEditor")]
     
     def _create_PythonCodeEditor(self, prnt):
-        self.PythonCodeEditor = CustomStyledTextCtrl(id=ID_PYTHONEDITOR, parent=prnt,
-                 name="TextViewer", pos=wx.DefaultPosition, 
-                 size=wx.DefaultSize, style=0)
-        self.PythonCodeEditor.ParentWindow = self
-        
-        self.PythonCodeEditor.CmdKeyAssign(ord('B'), stc.STC_SCMOD_CTRL, stc.STC_CMD_ZOOMIN)
-        self.PythonCodeEditor.CmdKeyAssign(ord('N'), stc.STC_SCMOD_CTRL, stc.STC_CMD_ZOOMOUT)
-
-        self.PythonCodeEditor.SetLexer(stc.STC_LEX_PYTHON)
-        self.PythonCodeEditor.SetKeyWords(0, " ".join(keyword.kwlist))
-
-        self.PythonCodeEditor.SetProperty("fold", "1")
-        self.PythonCodeEditor.SetProperty("tab.timmy.whinge.level", "1")
-        self.PythonCodeEditor.SetMargins(0,0)
-
-        self.PythonCodeEditor.SetViewWhiteSpace(False)
-        
-        self.PythonCodeEditor.SetEdgeMode(stc.STC_EDGE_BACKGROUND)
-        self.PythonCodeEditor.SetEdgeColumn(78)
-
-        # Set up the numbers in the margin for margin #1
-        self.PythonCodeEditor.SetMarginType(1, wx.stc.STC_MARGIN_NUMBER)
-        # Reasonable value for, say, 4-5 digits using a mono font (40 pix)
-        self.PythonCodeEditor.SetMarginWidth(1, 40)
-
-        # Setup a margin to hold fold markers
-        self.PythonCodeEditor.SetMarginType(2, stc.STC_MARGIN_SYMBOL)
-        self.PythonCodeEditor.SetMarginMask(2, stc.STC_MASK_FOLDERS)
-        self.PythonCodeEditor.SetMarginSensitive(2, True)
-        self.PythonCodeEditor.SetMarginWidth(2, 12)
-
-        if self.fold_symbols == 0:
-            # Arrow pointing right for contracted folders, arrow pointing down for expanded
-            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDEROPEN,    stc.STC_MARK_ARROWDOWN, "black", "black")
-            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDER,        stc.STC_MARK_ARROW, "black", "black")
-            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDERSUB,     stc.STC_MARK_EMPTY, "black", "black")
-            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDERTAIL,    stc.STC_MARK_EMPTY, "black", "black")
-            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDEREND,     stc.STC_MARK_EMPTY,     "white", "black")
-            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDEROPENMID, stc.STC_MARK_EMPTY,     "white", "black")
-            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDERMIDTAIL, stc.STC_MARK_EMPTY,     "white", "black")
-            
-        elif self.fold_symbols == 1:
-            # Plus for contracted folders, minus for expanded
-            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDEROPEN,    stc.STC_MARK_MINUS, "white", "black")
-            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDER,        stc.STC_MARK_PLUS,  "white", "black")
-            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDERSUB,     stc.STC_MARK_EMPTY, "white", "black")
-            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDERTAIL,    stc.STC_MARK_EMPTY, "white", "black")
-            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDEREND,     stc.STC_MARK_EMPTY, "white", "black")
-            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDEROPENMID, stc.STC_MARK_EMPTY, "white", "black")
-            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDERMIDTAIL, stc.STC_MARK_EMPTY, "white", "black")
-
-        elif self.fold_symbols == 2:
-            # Like a flattened tree control using circular headers and curved joins
-            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDEROPEN,    stc.STC_MARK_CIRCLEMINUS,          "white", "#404040")
-            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDER,        stc.STC_MARK_CIRCLEPLUS,           "white", "#404040")
-            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDERSUB,     stc.STC_MARK_VLINE,                "white", "#404040")
-            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDERTAIL,    stc.STC_MARK_LCORNERCURVE,         "white", "#404040")
-            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDEREND,     stc.STC_MARK_CIRCLEPLUSCONNECTED,  "white", "#404040")
-            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDEROPENMID, stc.STC_MARK_CIRCLEMINUSCONNECTED, "white", "#404040")
-            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDERMIDTAIL, stc.STC_MARK_TCORNERCURVE,         "white", "#404040")
-
-        elif self.fold_symbols == 3:
-            # Like a flattened tree control using square headers
-            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDEROPEN,    stc.STC_MARK_BOXMINUS,          "white", "#808080")
-            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDER,        stc.STC_MARK_BOXPLUS,           "white", "#808080")
-            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDERSUB,     stc.STC_MARK_VLINE,             "white", "#808080")
-            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDERTAIL,    stc.STC_MARK_LCORNER,           "white", "#808080")
-            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDEREND,     stc.STC_MARK_BOXPLUSCONNECTED,  "white", "#808080")
-            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDEROPENMID, stc.STC_MARK_BOXMINUSCONNECTED, "white", "#808080")
-            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDERMIDTAIL, stc.STC_MARK_TCORNER,           "white", "#808080")
-
-
-        self.PythonCodeEditor.Bind(stc.EVT_STC_UPDATEUI, self.OnUpdateUI)
-        self.PythonCodeEditor.Bind(stc.EVT_STC_MARGINCLICK, self.OnMarginClick)
-        self.PythonCodeEditor.Bind(wx.EVT_KEY_DOWN, self.OnKeyPressed)
-        
-        # Global default style
-        if wx.Platform == '__WXMSW__':
-            self.PythonCodeEditor.StyleSetSpec(stc.STC_STYLE_DEFAULT, 'fore:#000000,back:#FFFFFF,face:Courier New')
-        elif wx.Platform == '__WXMAC__':
-            # TODO: if this looks fine on Linux too, remove the Mac-specific case 
-            # and use this whenever OS != MSW.
-            self.PythonCodeEditor.StyleSetSpec(stc.STC_STYLE_DEFAULT, 'fore:#000000,back:#FFFFFF,face:Monaco')
-        else:
-            defsize = wx.SystemSettings.GetFont(wx.SYS_ANSI_FIXED_FONT).GetPointSize()
-            self.PythonCodeEditor.StyleSetSpec(stc.STC_STYLE_DEFAULT, 'fore:#000000,back:#FFFFFF,face:Courier,size:%d'%defsize)
-
-        # Clear styles and revert to default.
-        self.PythonCodeEditor.StyleClearAll()
-
-        # Following style specs only indicate differences from default.
-        # The rest remains unchanged.
-
-        # Line numbers in margin
-        self.PythonCodeEditor.StyleSetSpec(stc.STC_STYLE_LINENUMBER,'fore:#000000,back:#99A9C2,size:%(size)d' % faces)    
-        # Highlighted brace
-        self.PythonCodeEditor.StyleSetSpec(stc.STC_STYLE_BRACELIGHT,'fore:#00009D,back:#FFFF00,size:%(size)d' % faces)
-        # Unmatched brace
-        self.PythonCodeEditor.StyleSetSpec(stc.STC_STYLE_BRACEBAD,'fore:#00009D,back:#FF0000,size:%(size)d' % faces)
-        # Indentation guide
-        self.PythonCodeEditor.StyleSetSpec(stc.STC_STYLE_INDENTGUIDE, 'fore:#CDCDCD,size:%(size)d' % faces)
-
-        # Python styles
-        self.PythonCodeEditor.StyleSetSpec(stc.STC_P_DEFAULT, 'fore:#000000,size:%(size)d' % faces)
-        # Comments
-        self.PythonCodeEditor.StyleSetSpec(stc.STC_P_COMMENTLINE,  'fore:#008000,back:#F0FFF0,size:%(size)d' % faces)
-        self.PythonCodeEditor.StyleSetSpec(stc.STC_P_COMMENTBLOCK, 'fore:#008000,back:#F0FFF0,size:%(size)d' % faces)
-        # Numbers
-        self.PythonCodeEditor.StyleSetSpec(stc.STC_P_NUMBER, 'fore:#008080,size:%(size)d' % faces)
-        # Strings and characters
-        self.PythonCodeEditor.StyleSetSpec(stc.STC_P_STRING, 'fore:#800080,size:%(size)d' % faces)
-        self.PythonCodeEditor.StyleSetSpec(stc.STC_P_CHARACTER, 'fore:#800080,size:%(size)d' % faces)
-        # Keywords
-        self.PythonCodeEditor.StyleSetSpec(stc.STC_P_WORD, 'fore:#000080,bold,size:%(size)d' % faces)
-        # Triple quotes
-        self.PythonCodeEditor.StyleSetSpec(stc.STC_P_TRIPLE, 'fore:#800080,back:#FFFFEA,size:%(size)d' % faces)
-        self.PythonCodeEditor.StyleSetSpec(stc.STC_P_TRIPLEDOUBLE, 'fore:#800080,back:#FFFFEA,size:%(size)d' % faces)
-        # Class names
-        self.PythonCodeEditor.StyleSetSpec(stc.STC_P_CLASSNAME, 'fore:#0000FF,bold,size:%(size)d' % faces)
-        # Function names
-        self.PythonCodeEditor.StyleSetSpec(stc.STC_P_DEFNAME, 'fore:#008080,bold,size:%(size)d' % faces)
-        # Operators
-        self.PythonCodeEditor.StyleSetSpec(stc.STC_P_OPERATOR, 'fore:#800000,bold,size:%(size)d' % faces)
-        # Identifiers. I leave this as not bold because everything seems
-        # to be an identifier if it doesn't match the above criterae
-        self.PythonCodeEditor.StyleSetSpec(stc.STC_P_IDENTIFIER, 'fore:#000000,size:%(size)d' % faces)
-        
-        # Highlighting styles
-        self.PythonCodeEditor.StyleSetSpec(STC_PYTHON_ERROR, 'fore:#FF0000,back:#FFFF00,size:%(size)d' % faces)
-        self.PythonCodeEditor.StyleSetSpec(STC_PYTHON_SEARCH_RESULT, 'fore:#FFFFFF,back:#FFA500,size:%(size)d' % faces)
-        
-        # Caret color
-        self.PythonCodeEditor.SetCaretForeground("BLUE")
-        # Selection background
-        self.PythonCodeEditor.SetSelBackground(1, '#66CCFF')
-
-        self.PythonCodeEditor.SetSelBackground(True, wx.SystemSettings_GetColour(wx.SYS_COLOUR_HIGHLIGHT))
-        self.PythonCodeEditor.SetSelForeground(True, wx.SystemSettings_GetColour(wx.SYS_COLOUR_HIGHLIGHTTEXT))
-        
-        # register some images for use in the AutoComplete box.
-        #self.RegisterImage(1, images.getSmilesBitmap())
-        self.PythonCodeEditor.RegisterImage(1, 
-            wx.ArtProvider.GetBitmap(wx.ART_DELETE, size=(16,16)))
-        self.PythonCodeEditor.RegisterImage(2, 
-            wx.ArtProvider.GetBitmap(wx.ART_NEW, size=(16,16)))
-        self.PythonCodeEditor.RegisterImage(3, 
-            wx.ArtProvider.GetBitmap(wx.ART_COPY, size=(16,16)))
-
-        # Indentation and tab stuff
-        self.PythonCodeEditor.SetIndent(4)               # Proscribed indent size for wx
-        self.PythonCodeEditor.SetIndentationGuides(True) # Show indent guides
-        self.PythonCodeEditor.SetBackSpaceUnIndents(True)# Backspace unindents rather than delete 1 space
-        self.PythonCodeEditor.SetTabIndents(True)        # Tab key indents
-        self.PythonCodeEditor.SetTabWidth(4)             # Proscribed tab size for wx
-        self.PythonCodeEditor.SetUseTabs(False)          # Use spaces rather than tabs, or
-                                        # TabTimmy will complain!    
-        # White space
-        self.PythonCodeEditor.SetViewWhiteSpace(False)   # Don't view white space
-
-        # EOL: Since we are loading/saving ourselves, and the
-        # strings will always have \n's in them, set the STC to
-        # edit them that way.            
-        self.PythonCodeEditor.SetEOLMode(stc.STC_EOL_LF)
-        self.PythonCodeEditor.SetViewEOL(False)
-        
-        # No right-edge mode indicator
-        self.PythonCodeEditor.SetEdgeMode(stc.STC_EDGE_NONE)
-        
-        self.PythonCodeEditor.SetModEventMask(stc.STC_MOD_BEFOREINSERT|stc.STC_MOD_BEFOREDELETE)
-
-        self.PythonCodeEditor.Bind(wx.stc.EVT_STC_DO_DROP, self.OnDoDrop, id=ID_PYTHONEDITOR)
-        self.PythonCodeEditor.Bind(wx.EVT_KILL_FOCUS, self.OnKillFocus)
-        self.PythonCodeEditor.Bind(wx.stc.EVT_STC_MODIFIED, self.OnModification, id=ID_PYTHONEDITOR)
+        self.PythonCodeEditor = PythonCodeEditor(prnt, self.ParentWindow, self.Controler)
         
         return self.PythonCodeEditor
 
-    def __init__(self, parent, controler, window):
-        ConfTreeNodeEditor.__init__(self, parent, controler, window)
+    def RefreshView(self):
+        CodeFileEditor.RefreshView(self)
         
-        self.DisableEvents = False
-        self.CurrentAction = None
-    
-        self.Highlights = []
-        self.SearchParams = None
-        self.SearchResults = None
-        self.CurrentFindHighlight = None
-        
-        self.RefreshHighlightsTimer = wx.Timer(self, -1)
-        self.Bind(wx.EVT_TIMER, self.OnRefreshHighlightsTimer, self.RefreshHighlightsTimer)
-    
-    def GetBufferState(self):
-        return self.Controler.GetBufferState()
-        
-    def Undo(self):
-        self.Controler.LoadPrevious()
-        self.RefreshView()
-            
-    def Redo(self):
-        self.Controler.LoadNext()
-        self.RefreshView()
-    
-    def OnModification(self, event):
-        if not self.DisableEvents:
-            mod_type = event.GetModificationType()
-            if not (mod_type&wx.stc.STC_PERFORMED_UNDO or mod_type&wx.stc.STC_PERFORMED_REDO):
-                if mod_type&wx.stc.STC_MOD_BEFOREINSERT:
-                    if self.CurrentAction is None:
-                        self.StartBuffering()
-                    elif self.CurrentAction[0] != "Add" or self.CurrentAction[1] != event.GetPosition() - 1:
-                        self.Controler.EndBuffering()
-                        self.StartBuffering()
-                    self.CurrentAction = ("Add", event.GetPosition())
-                    wx.CallAfter(self.RefreshModel)
-                elif mod_type&wx.stc.STC_MOD_BEFOREDELETE:
-                    if self.CurrentAction == None:
-                        self.StartBuffering()
-                    elif self.CurrentAction[0] != "Delete" or self.CurrentAction[1] != event.GetPosition() + 1:
-                        self.Controler.EndBuffering()
-                        self.StartBuffering()
-                    self.CurrentAction = ("Delete", event.GetPosition())
-                    wx.CallAfter(self.RefreshModel)
-        event.Skip()
-    
-    def OnDoDrop(self, event):
-        self.ResetBuffer()
-        wx.CallAfter(self.RefreshModel)
-        event.Skip()
-
-    # Buffer the last model state
-    def RefreshBuffer(self):
-        self.Controler.BufferPython()
-        if self.ParentWindow is not None:
-            self.ParentWindow.RefreshTitle()
-            self.ParentWindow.RefreshFileMenu()
-            self.ParentWindow.RefreshEditMenu()
-            self.ParentWindow.RefreshPageTitles()
-    
-    def StartBuffering(self):
-        self.Controler.StartBuffering()
-        if self.ParentWindow is not None:
-            self.ParentWindow.RefreshTitle()
-            self.ParentWindow.RefreshFileMenu()
-            self.ParentWindow.RefreshEditMenu()
-            self.ParentWindow.RefreshPageTitles()
-    
-    def ResetBuffer(self):
-        if self.CurrentAction != None:
-            self.Controler.EndBuffering()
-            self.CurrentAction = None
-
-    def RefreshView(self):
-        ConfTreeNodeEditor.RefreshView(self)
-        
-        self.ResetBuffer()
-        self.DisableEvents = True
-        old_cursor_pos = self.PythonCodeEditor.GetCurrentPos()
-        line = self.PythonCodeEditor.GetFirstVisibleLine()
-        column = self.PythonCodeEditor.GetXOffset()
-        old_text = self.PythonCodeEditor.GetText()
-        new_text = self.Controler.GetPythonCode()
-        if old_text != new_text:
-            self.PythonCodeEditor.SetText(new_text)
-            new_cursor_pos = GetCursorPos(old_text, new_text)
-            self.PythonCodeEditor.LineScroll(column, line)
-            if new_cursor_pos != None:
-                self.PythonCodeEditor.GotoPos(new_cursor_pos)
-            else:
-                self.PythonCodeEditor.GotoPos(old_cursor_pos)
-            self.PythonCodeEditor.EmptyUndoBuffer()
-        self.DisableEvents = False
-        
-        self.PythonCodeEditor.Colourise(0, -1)
-        
-        self.ShowHighlights()
-
-    def RefreshModel(self):
-        self.Controler.SetPythonCode(self.PythonCodeEditor.GetText())
-
-    def OnKeyPressed(self, event):
-        if self.PythonCodeEditor.CallTipActive():
-            self.PythonCodeEditor.CallTipCancel()
-        key = event.GetKeyCode()
-
-        if key == 32 and event.ControlDown():
-            pos = self.PythonCodeEditor.GetCurrentPos()
-
-            # Code completion
-            if not event.ShiftDown():
-                self.PythonCodeEditor.AutoCompSetIgnoreCase(False)  # so this needs to match
-
-                # Images are specified with a appended "?type"
-                self.PythonCodeEditor.AutoCompShow(0, " ".join([word + "?1" for word in keyword.kwlist]))
-        else:
-            event.Skip()
-    
-    def OnKillFocus(self, event):
-        self.PythonCodeEditor.AutoCompCancel()
-        event.Skip()
-
-    def OnUpdateUI(self, evt):
-        # check for matching braces
-        braceAtCaret = -1
-        braceOpposite = -1
-        charBefore = None
-        caretPos = self.PythonCodeEditor.GetCurrentPos()
-
-        if caretPos > 0:
-            charBefore = self.PythonCodeEditor.GetCharAt(caretPos - 1)
-            styleBefore = self.PythonCodeEditor.GetStyleAt(caretPos - 1)
-
-        # check before
-        if charBefore and chr(charBefore) in "[]{}()" and styleBefore == stc.STC_P_OPERATOR:
-            braceAtCaret = caretPos - 1
-
-        # check after
-        if braceAtCaret < 0:
-            charAfter = self.PythonCodeEditor.GetCharAt(caretPos)
-            styleAfter = self.PythonCodeEditor.GetStyleAt(caretPos)
-
-            if charAfter and chr(charAfter) in "[]{}()" and styleAfter == stc.STC_P_OPERATOR:
-                braceAtCaret = caretPos
-
-        if braceAtCaret >= 0:
-            braceOpposite = self.PythonCodeEditor.BraceMatch(braceAtCaret)
-
-        if braceAtCaret != -1  and braceOpposite == -1:
-            self.PythonCodeEditor.BraceBadLight(braceAtCaret)
-        else:
-            self.PythonCodeEditor.BraceHighlight(braceAtCaret, braceOpposite)
-
-    def OnMarginClick(self, evt):
-        # fold and unfold as needed
-        if evt.GetMargin() == 2:
-            if evt.GetShift() and evt.GetControl():
-                self.FoldAll()
-            else:
-                lineClicked = self.PythonCodeEditor.LineFromPosition(evt.GetPosition())
-
-                if self.PythonCodeEditor.GetFoldLevel(lineClicked) & stc.STC_FOLDLEVELHEADERFLAG:
-                    if evt.GetShift():
-                        self.PythonCodeEditor.SetFoldExpanded(lineClicked, True)
-                        self.Expand(lineClicked, True, True, 1)
-                    elif evt.GetControl():
-                        if self.PythonCodeEditor.GetFoldExpanded(lineClicked):
-                            self.PythonCodeEditor.SetFoldExpanded(lineClicked, False)
-                            self.Expand(lineClicked, False, True, 0)
-                        else:
-                            self.PythonCodeEditor.SetFoldExpanded(lineClicked, True)
-                            self.Expand(lineClicked, True, True, 100)
-                    else:
-                        self.PythonCodeEditor.ToggleFold(lineClicked)
-
-
-    def FoldAll(self):
-        lineCount = self.PythonCodeEditor.GetLineCount()
-        expanding = True
-
-        # find out if we are folding or unfolding
-        for lineNum in range(lineCount):
-            if self.PythonCodeEditor.GetFoldLevel(lineNum) & stc.STC_FOLDLEVELHEADERFLAG:
-                expanding = not self.PythonCodeEditor.GetFoldExpanded(lineNum)
-                break
-
-        lineNum = 0
-
-        while lineNum < lineCount:
-            level = self.PythonCodeEditor.GetFoldLevel(lineNum)
-            if level & stc.STC_FOLDLEVELHEADERFLAG and \
-               (level & stc.STC_FOLDLEVELNUMBERMASK) == stc.STC_FOLDLEVELBASE:
-
-                if expanding:
-                    self.PythonCodeEditor.SetFoldExpanded(lineNum, True)
-                    lineNum = self.Expand(lineNum, True)
-                    lineNum = lineNum - 1
-                else:
-                    lastChild = self.PythonCodeEditor.GetLastChild(lineNum, -1)
-                    self.PythonCodeEditor.SetFoldExpanded(lineNum, False)
-
-                    if lastChild > lineNum:
-                        self.PythonCodeEditor.HideLines(lineNum+1, lastChild)
-
-            lineNum = lineNum + 1
-
-
-
-    def Expand(self, line, doExpand, force=False, visLevels=0, level=-1):
-        lastChild = self.PythonCodeEditor.GetLastChild(line, level)
-        line = line + 1
-
-        while line <= lastChild:
-            if force:
-                if visLevels > 0:
-                    self.PythonCodeEditor.ShowLines(line, line)
-                else:
-                    self.PythonCodeEditor.HideLines(line, line)
-            else:
-                if doExpand:
-                    self.PythonCodeEditor.ShowLines(line, line)
-
-            if level == -1:
-                level = self.PythonCodeEditor.GetFoldLevel(line)
-
-            if level & stc.STC_FOLDLEVELHEADERFLAG:
-                if force:
-                    if visLevels > 1:
-                        self.PythonCodeEditor.SetFoldExpanded(line, True)
-                    else:
-                        self.PythonCodeEditor.SetFoldExpanded(line, False)
-
-                    line = self.Expand(line, doExpand, force, visLevels-1)
-
-                else:
-                    if doExpand and self.PythonCodeEditor.GetFoldExpanded(line):
-                        line = self.Expand(line, True, force, visLevels-1)
-                    else:
-                        line = self.Expand(line, False, force, visLevels-1)
-            else:
-                line = line + 1
-
-        return line
-
-    def Cut(self):
-        self.ResetBuffer()
-        self.DisableEvents = True
-        self.PythonCodeEditor.CmdKeyExecute(wx.stc.STC_CMD_CUT)
-        self.DisableEvents = False
-        self.RefreshModel()
-        self.RefreshBuffer()
-    
-    def Copy(self):
-        self.PythonCodeEditor.CmdKeyExecute(wx.stc.STC_CMD_COPY)
-    
-    def Paste(self):
-        self.ResetBuffer()
-        self.DisableEvents = True
-        self.PythonCodeEditor.CmdKeyExecute(wx.stc.STC_CMD_PASTE)
-        self.DisableEvents = False
-        self.RefreshModel()
-        self.RefreshBuffer()
+        self.PythonCodeEditor.RefreshView()
 
     def Find(self, direction, search_params):
-        if self.SearchParams != search_params:
-            self.ClearHighlights(SEARCH_RESULT_HIGHLIGHT)
-            
-            self.SearchParams = search_params
-            criteria = {
-                "raw_pattern": search_params["find_pattern"], 
-                "pattern": re.compile(search_params["find_pattern"]),
-                "case_sensitive": search_params["case_sensitive"],
-                "regular_expression": search_params["regular_expression"],
-                "filter": "all"}
-            
-            self.SearchResults = [
-                (start, end, SEARCH_RESULT_HIGHLIGHT)
-                for start, end, text in 
-                TestTextElement(self.PythonCodeEditor.GetText(), criteria)]
-            self.CurrentFindHighlight = None
-        
-        if len(self.SearchResults) > 0:
-            if self.CurrentFindHighlight is not None:
-                old_idx = self.SearchResults.index(self.CurrentFindHighlight)
-                if self.SearchParams["wrap"]:
-                    idx = (old_idx + direction) % len(self.SearchResults)
-                else:
-                    idx = max(0, min(old_idx + direction, len(self.SearchResults) - 1))
-                if idx != old_idx:
-                    self.RemoveHighlight(*self.CurrentFindHighlight)
-                    self.CurrentFindHighlight = self.SearchResults[idx]
-                    self.AddHighlight(*self.CurrentFindHighlight)
-            else:
-                self.CurrentFindHighlight = self.SearchResults[0]
-                self.AddHighlight(*self.CurrentFindHighlight)
-            
-        else:
-            if self.CurrentFindHighlight is not None:
-                self.RemoveHighlight(*self.CurrentFindHighlight)
-            self.CurrentFindHighlight = None
-
-#-------------------------------------------------------------------------------
-#                        Highlights showing functions
-#-------------------------------------------------------------------------------
-
-    def OnRefreshHighlightsTimer(self, event):
-        self.RefreshView()
-        event.Skip()
-
-    def ClearHighlights(self, highlight_type=None):
-        if highlight_type is None:
-            self.Highlights = []
-        else:
-            highlight_type = HIGHLIGHT_TYPES.get(highlight_type, None)
-            if highlight_type is not None:
-                self.Highlights = [(start, end, highlight) for (start, end, highlight) in self.Highlights if highlight != highlight_type]
-        self.RefreshView()
-
-    def AddHighlight(self, start, end, highlight_type):
-        highlight_type = HIGHLIGHT_TYPES.get(highlight_type, None)
-        if highlight_type is not None:
-            self.Highlights.append((start, end, highlight_type))
-            self.PythonCodeEditor.GotoPos(self.PythonCodeEditor.PositionFromLine(start[0]) + start[1])
-            self.RefreshHighlightsTimer.Start(int(REFRESH_HIGHLIGHT_PERIOD * 1000), oneShot=True)
-            self.RefreshView()
-
-    def RemoveHighlight(self, start, end, highlight_type):
-        highlight_type = HIGHLIGHT_TYPES.get(highlight_type, None)
-        if (highlight_type is not None and 
-            (start, end, highlight_type) in self.Highlights):
-            self.Highlights.remove((start, end, highlight_type))
-            self.RefreshHighlightsTimer.Start(int(REFRESH_HIGHLIGHT_PERIOD * 1000), oneShot=True)
-    
-    def ShowHighlights(self):
-        for start, end, highlight_type in self.Highlights:
-            if start[0] == 0:
-                highlight_start_pos = start[1]
-            else:
-                highlight_start_pos = self.PythonCodeEditor.GetLineEndPosition(start[0] - 1) + start[1] + 1
-            if end[0] == 0:
-                highlight_end_pos = end[1] - indent + 1
-            else:
-                highlight_end_pos = self.PythonCodeEditor.GetLineEndPosition(end[0] - 1) + end[1] + 2
-            self.PythonCodeEditor.StartStyling(highlight_start_pos, 0xff)
-            self.PythonCodeEditor.SetStyling(highlight_end_pos - highlight_start_pos, highlight_type)
-            self.PythonCodeEditor.StartStyling(highlight_start_pos, 0x00)
-            self.PythonCodeEditor.SetStyling(len(self.PythonCodeEditor.GetText()) - highlight_end_pos, stc.STC_STYLE_DEFAULT)
-                
+        self.PythonCodeEditor.Find(direction, search_params)
--- a/py_ext/PythonFileCTNMixin.py	Wed May 08 21:37:31 2013 +0200
+++ b/py_ext/PythonFileCTNMixin.py	Wed May 08 22:52:55 2013 +0200
@@ -6,17 +6,20 @@
 from xmlclass import *
 import cPickle
 
+from CodeFileTreeNode import CodeFile
+
 PythonClasses = GenerateClassesFromXSD(os.path.join(os.path.dirname(__file__), "py_ext_xsd.xsd")) 
 
-class PythonFileCTNMixin:
+class PythonFileCTNMixin(CodeFile):
     
     EditorType = PythonEditor
     
     def __init__(self):
+        CodeFile.__init__(self)
         
         filepath = self.PythonFileName()
         
-        self.PythonCode = PythonClasses["Python"]()
+        python_code = PythonClasses["Python"]()
         if os.path.isfile(filepath):
             xmlfile = open(filepath, 'r')
             tree = minidom.parse(xmlfile)
@@ -24,87 +27,58 @@
             
             for child in tree.childNodes:
                 if child.nodeType == tree.ELEMENT_NODE and child.nodeName == "Python":
-                    self.PythonCode.loadXMLTree(child, ["xmlns", "xmlns:xsi", "xsi:schemaLocation"])
-                    self.CreatePythonBuffer(True)
-        else:
-            self.CreatePythonBuffer(False)
-            self.OnCTNSave()
-
+                    python_code.loadXMLTree(child, ["xmlns", "xmlns:xsi", "xsi:schemaLocation"])
+                    self.CodeFile.globals.settext(python_code.gettext())
+                    os.remove(filepath)
+                    self.CreateCodeFileBuffer(False)
+                    self.OnCTNSave()
+    
+    def CodeFileName(self):
+        return os.path.join(self.CTNPath(), "pyfile.xml")
+    
     def PythonFileName(self):
         return os.path.join(self.CTNPath(), "py_ext.xml")
 
-    def GetFilename(self):
-        if self.PythonBuffer.IsCurrentSaved():
-            return "py_ext"
-        else:
-            return "~py_ext~"
+    def GetPythonCode(self):
+        current_location = self.GetCurrentLocation()
+        # define a unique name for the generated C file
+        location_str = "_".join(map(str, current_location))
+        
+        text = "## Code generated by Beremiz python mixin confnode\n\n"
+        
+        # Adding includes
+        text += "## User includes\n"
+        text += self.CodeFile.includes.gettext()
+        text += "\n"
+        
+        # Adding variables
+        text += "## User variables reference\n"
+        for variable in self.CodeFile.variables.variable:
+            text += "%s = %s\n" % (variable.getname(),
+                                   str(variable.getinitial()))
+        text += "\n"
+        
+        # Adding user global variables and routines
+        text += "## User internal user variables and routines\n"
+        text += self.CodeFile.globals.gettext()
+        text += "\n"
+        
+        # Adding Beremiz confnode functions
+        text += "## Beremiz confnode functions\n"
+        for func, args, return_code, code_object in [
+            ("__init_", "*args, **kwargs", 
+             "return 0", self.CodeFile.initFunction),
+            ("__cleanup_", "", "", self.CodeFile.cleanUpFunction),
+            ("__retrieve_", "", "", self.CodeFile.retrieveFunction),
+            ("__publish_", "", "", self.CodeFile.publishFunction),]:
+            text += "def %s%s(%s):\n" % (func, location_str, args)
+            lines = code_object.gettext().splitlines()
+            if len(lines) > 0 or return_code != "":
+                for line in code_object.gettext().splitlines():
+                    text += "    " + line
+                text += "    " + return_code + "\n\n"
+            else:
+                text += "    pass\n\n"
+        
+        return text
 
-    def SetPythonCode(self, text):
-        self.PythonCode.settext(text)
-        
-    def GetPythonCode(self):
-        return self.PythonCode.gettext()
-    
-    def CTNTestModified(self):
-        return self.ChangesToSave or not self.PythonIsSaved()
-    
-    def OnCTNSave(self, from_project_path=None):
-        filepath = self.PythonFileName()
-        
-        text = "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\"?>\n"
-        extras = {"xmlns":"http://www.w3.org/2001/XMLSchema",
-                  "xmlns:xsi":"http://www.w3.org/2001/XMLSchema-instance",
-                  "xsi:schemaLocation" : "py_ext_xsd.xsd"}
-        text += self.PythonCode.generateXMLText("Python", 0, extras)
-
-        xmlfile = open(filepath,"w")
-        xmlfile.write(text.encode("utf-8"))
-        xmlfile.close()
-        
-        self.MarkPythonAsSaved()
-        return True
-        
-#-------------------------------------------------------------------------------
-#                      Current Buffering Management Functions
-#-------------------------------------------------------------------------------
-
-    """
-    Return a copy of the project
-    """
-    def Copy(self, model):
-        return cPickle.loads(cPickle.dumps(model))
-
-    def CreatePythonBuffer(self, saved):
-        self.Buffering = False
-        self.PythonBuffer = UndoBuffer(cPickle.dumps(self.PythonCode), saved)
-
-    def BufferPython(self):
-        self.PythonBuffer.Buffering(cPickle.dumps(self.PythonCode))
-    
-    def StartBuffering(self):
-        self.Buffering = True
-        
-    def EndBuffering(self):
-        if self.Buffering:
-            self.PythonBuffer.Buffering(cPickle.dumps(self.PythonCode))
-            self.Buffering = False
-    
-    def MarkPythonAsSaved(self):
-        self.EndBuffering()
-        self.PythonBuffer.CurrentSaved()
-    
-    def PythonIsSaved(self):
-        return self.PythonBuffer.IsCurrentSaved() and not self.Buffering
-        
-    def LoadPrevious(self):
-        self.EndBuffering()
-        self.PythonCode = cPickle.loads(self.PythonBuffer.Previous())
-    
-    def LoadNext(self):
-        self.PythonCode = cPickle.loads(self.PythonBuffer.Next())
-    
-    def GetBufferState(self):
-        first = self.PythonBuffer.IsFirst() and not self.Buffering
-        last = self.PythonBuffer.IsLast()
-        return not first, not last
-