1

import wx

     2

import wx.grid

     3

import wx.stc  as  stc

     4

import keyword

     5

     6

from editors.ConfTreeNodeEditor import ConfTreeNodeEditor

     7

     8

if wx.Platform == '__WXMSW__':

     9

    faces = { 'times': 'Times New Roman',

    10

              'mono' : 'Courier New',

    11

              'helv' : 'Arial',

    12

              'other': 'Comic Sans MS',

    13

              'size' : 10,

    14

              'size2': 8,

    15

             }

    16

elif wx.Platform == '__WXMAC__':

    17

    faces = { 'times': 'Times New Roman',

    18

              'mono' : 'Monaco',

    19

              'helv' : 'Arial',

    20

              'other': 'Comic Sans MS',

    21

              'size' : 12,

    22

              'size2': 10,

    23

             }

    24

else:

    25

    faces = { 'times': 'Times',

    26

              'mono' : 'Courier',

    27

              'helv' : 'Helvetica',

    28

              'other': 'new century schoolbook',

    29

              'size' : 12,

    30

              'size2': 10,

    31

             }

    32

    33

[ID_PYTHONEDITOR,

    34

] = [wx.NewId() for _init_ctrls in range(1)]

    35

    36

def GetCursorPos(old, new):

    37

    old_length = len(old)

    38

    new_length = len(new)

    39

    common_length = min(old_length, new_length)

    40

    i = 0

    41

    for i in xrange(common_length):

    42

        if old[i] != new[i]:

    43

            break

    44

    if old_length < new_length:

    45

        if common_length > 0 and old[i] != new[i]:

    46

            return i + new_length - old_length

    47

        else:

    48

            return i + new_length - old_length + 1

    49

    elif old_length > new_length or i < min(old_length, new_length) - 1:

    50

        if common_length > 0 and old[i] != new[i]:

    51

            return i

    52

        else:

    53

            return i + 1

    54

    else:

    55

        return None

    56

    57

class PythonEditor(ConfTreeNodeEditor):

    58

    59

    fold_symbols = 3

    60

    CONFNODEEDITOR_TABS = [

    61

        (_("Python code"), "_create_PythonCodeEditor")]

    62

    63

    def _create_PythonCodeEditor(self, prnt):

    64

        self.PythonCodeEditor = stc.StyledTextCtrl(id=ID_PYTHONEDITOR, parent=prnt,

    65

                 name="TextViewer", pos=wx.DefaultPosition, 

    66

                 size=wx.DefaultSize, style=0)

    67

        self.PythonCodeEditor.ParentWindow = self

    68

    69

        self.PythonCodeEditor.CmdKeyAssign(ord('B'), stc.STC_SCMOD_CTRL, stc.STC_CMD_ZOOMIN)

    70

        self.PythonCodeEditor.CmdKeyAssign(ord('N'), stc.STC_SCMOD_CTRL, stc.STC_CMD_ZOOMOUT)

    71

    72

        self.PythonCodeEditor.SetLexer(stc.STC_LEX_PYTHON)

    73

        self.PythonCodeEditor.SetKeyWords(0, " ".join(keyword.kwlist))

    74

    75

        self.PythonCodeEditor.SetProperty("fold", "1")

    76

        self.PythonCodeEditor.SetProperty("tab.timmy.whinge.level", "1")

    77

        self.PythonCodeEditor.SetMargins(0,0)

    78

    79

        self.PythonCodeEditor.SetViewWhiteSpace(False)

    80

    81

        self.PythonCodeEditor.SetEdgeMode(stc.STC_EDGE_BACKGROUND)

    82

        self.PythonCodeEditor.SetEdgeColumn(78)

    83

    84

        # Set up the numbers in the margin for margin #1

    85

        self.PythonCodeEditor.SetMarginType(1, wx.stc.STC_MARGIN_NUMBER)

    86

        # Reasonable value for, say, 4-5 digits using a mono font (40 pix)

    87

        self.PythonCodeEditor.SetMarginWidth(1, 40)

    88

    89

        # Setup a margin to hold fold markers

    90

        self.PythonCodeEditor.SetMarginType(2, stc.STC_MARGIN_SYMBOL)

    91

        self.PythonCodeEditor.SetMarginMask(2, stc.STC_MASK_FOLDERS)

    92

        self.PythonCodeEditor.SetMarginSensitive(2, True)

    93

        self.PythonCodeEditor.SetMarginWidth(2, 12)

    94

    95

        if self.fold_symbols == 0:

    96

            # Arrow pointing right for contracted folders, arrow pointing down for expanded

    97

            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDEROPEN,    stc.STC_MARK_ARROWDOWN, "black", "black")

    98

            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDER,        stc.STC_MARK_ARROW, "black", "black")

    99

            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDERSUB,     stc.STC_MARK_EMPTY, "black", "black")

   100

            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDERTAIL,    stc.STC_MARK_EMPTY, "black", "black")

   101

            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDEREND,     stc.STC_MARK_EMPTY,     "white", "black")

   102

            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDEROPENMID, stc.STC_MARK_EMPTY,     "white", "black")

   103

            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDERMIDTAIL, stc.STC_MARK_EMPTY,     "white", "black")

   104

   105

        elif self.fold_symbols == 1:

   106

            # Plus for contracted folders, minus for expanded

   107

            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDEROPEN,    stc.STC_MARK_MINUS, "white", "black")

   108

            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDER,        stc.STC_MARK_PLUS,  "white", "black")

   109

            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDERSUB,     stc.STC_MARK_EMPTY, "white", "black")

   110

            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDERTAIL,    stc.STC_MARK_EMPTY, "white", "black")

   111

            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDEREND,     stc.STC_MARK_EMPTY, "white", "black")

   112

            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDEROPENMID, stc.STC_MARK_EMPTY, "white", "black")

   113

            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDERMIDTAIL, stc.STC_MARK_EMPTY, "white", "black")

   114

   115

        elif self.fold_symbols == 2:

   116

            # Like a flattened tree control using circular headers and curved joins

   117

            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDEROPEN,    stc.STC_MARK_CIRCLEMINUS,          "white", "#404040")

   118

            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDER,        stc.STC_MARK_CIRCLEPLUS,           "white", "#404040")

   119

            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDERSUB,     stc.STC_MARK_VLINE,                "white", "#404040")

   120

            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDERTAIL,    stc.STC_MARK_LCORNERCURVE,         "white", "#404040")

   121

            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDEREND,     stc.STC_MARK_CIRCLEPLUSCONNECTED,  "white", "#404040")

   122

            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDEROPENMID, stc.STC_MARK_CIRCLEMINUSCONNECTED, "white", "#404040")

   123

            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDERMIDTAIL, stc.STC_MARK_TCORNERCURVE,         "white", "#404040")

   124

   125

        elif self.fold_symbols == 3:

   126

            # Like a flattened tree control using square headers

   127

            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDEROPEN,    stc.STC_MARK_BOXMINUS,          "white", "#808080")

   128

            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDER,        stc.STC_MARK_BOXPLUS,           "white", "#808080")

   129

            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDERSUB,     stc.STC_MARK_VLINE,             "white", "#808080")

   130

            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDERTAIL,    stc.STC_MARK_LCORNER,           "white", "#808080")

   131

            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDEREND,     stc.STC_MARK_BOXPLUSCONNECTED,  "white", "#808080")

   132

            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDEROPENMID, stc.STC_MARK_BOXMINUSCONNECTED, "white", "#808080")

   133

            self.PythonCodeEditor.MarkerDefine(stc.STC_MARKNUM_FOLDERMIDTAIL, stc.STC_MARK_TCORNER,           "white", "#808080")

   134

   135

   136

        self.PythonCodeEditor.Bind(stc.EVT_STC_UPDATEUI, self.OnUpdateUI)

   137

        self.PythonCodeEditor.Bind(stc.EVT_STC_MARGINCLICK, self.OnMarginClick)

   138

        self.PythonCodeEditor.Bind(wx.EVT_KEY_DOWN, self.OnKeyPressed)

   139

   140

        # Global default style

   141

        if wx.Platform == '__WXMSW__':

   142

            self.PythonCodeEditor.StyleSetSpec(stc.STC_STYLE_DEFAULT, 'fore:#000000,back:#FFFFFF,face:Courier New')

   143

        elif wx.Platform == '__WXMAC__':

   144

            # TODO: if this looks fine on Linux too, remove the Mac-specific case 

   145

            # and use this whenever OS != MSW.

   146

            self.PythonCodeEditor.StyleSetSpec(stc.STC_STYLE_DEFAULT, 'fore:#000000,back:#FFFFFF,face:Monaco')

   147

        else:

   148

            defsize = wx.SystemSettings.GetFont(wx.SYS_ANSI_FIXED_FONT).GetPointSize()

   149

            self.PythonCodeEditor.StyleSetSpec(stc.STC_STYLE_DEFAULT, 'fore:#000000,back:#FFFFFF,face:Courier,size:%d'%defsize)

   150

   151

        # Clear styles and revert to default.

   152

        self.PythonCodeEditor.StyleClearAll()

   153

   154

        # Following style specs only indicate differences from default.

   155

        # The rest remains unchanged.

   156

   157

        # Line numbers in margin

   158

        self.PythonCodeEditor.StyleSetSpec(wx.stc.STC_STYLE_LINENUMBER,'fore:#000000,back:#99A9C2')    

   159

        # Highlighted brace

   160

        self.PythonCodeEditor.StyleSetSpec(wx.stc.STC_STYLE_BRACELIGHT,'fore:#00009D,back:#FFFF00')

   161

        # Unmatched brace

   162

        self.PythonCodeEditor.StyleSetSpec(wx.stc.STC_STYLE_BRACEBAD,'fore:#00009D,back:#FF0000')

   163

        # Indentation guide

   164

        self.PythonCodeEditor.StyleSetSpec(wx.stc.STC_STYLE_INDENTGUIDE, "fore:#CDCDCD")

   165

   166

        # Python styles

   167

        self.PythonCodeEditor.StyleSetSpec(wx.stc.STC_P_DEFAULT, 'fore:#000000')

   168

        # Comments

   169

        self.PythonCodeEditor.StyleSetSpec(wx.stc.STC_P_COMMENTLINE,  'fore:#008000,back:#F0FFF0')

   170

        self.PythonCodeEditor.StyleSetSpec(wx.stc.STC_P_COMMENTBLOCK, 'fore:#008000,back:#F0FFF0')

   171

        # Numbers

   172

        self.PythonCodeEditor.StyleSetSpec(wx.stc.STC_P_NUMBER, 'fore:#008080')

   173

        # Strings and characters

   174

        self.PythonCodeEditor.StyleSetSpec(wx.stc.STC_P_STRING, 'fore:#800080')

   175

        self.PythonCodeEditor.StyleSetSpec(wx.stc.STC_P_CHARACTER, 'fore:#800080')

   176

        # Keywords

   177

        self.PythonCodeEditor.StyleSetSpec(wx.stc.STC_P_WORD, 'fore:#000080,bold')

   178

        # Triple quotes

   179

        self.PythonCodeEditor.StyleSetSpec(wx.stc.STC_P_TRIPLE, 'fore:#800080,back:#FFFFEA')

   180

        self.PythonCodeEditor.StyleSetSpec(wx.stc.STC_P_TRIPLEDOUBLE, 'fore:#800080,back:#FFFFEA')

   181

        # Class names

   182

        self.PythonCodeEditor.StyleSetSpec(wx.stc.STC_P_CLASSNAME, 'fore:#0000FF,bold')

   183

        # Function names

   184

        self.PythonCodeEditor.StyleSetSpec(wx.stc.STC_P_DEFNAME, 'fore:#008080,bold')

   185

        # Operators

   186

        self.PythonCodeEditor.StyleSetSpec(wx.stc.STC_P_OPERATOR, 'fore:#800000,bold')

   187

        # Identifiers. I leave this as not bold because everything seems

   188

        # to be an identifier if it doesn't match the above criterae

   189

        self.PythonCodeEditor.StyleSetSpec(wx.stc.STC_P_IDENTIFIER, 'fore:#000000')

   190

   191

        # Caret color

   192

        self.PythonCodeEditor.SetCaretForeground("BLUE")

   193

        # Selection background

   194

        self.PythonCodeEditor.SetSelBackground(1, '#66CCFF')

   195

   196

        self.PythonCodeEditor.SetSelBackground(True, wx.SystemSettings_GetColour(wx.SYS_COLOUR_HIGHLIGHT))

   197

        self.PythonCodeEditor.SetSelForeground(True, wx.SystemSettings_GetColour(wx.SYS_COLOUR_HIGHLIGHTTEXT))

   198

   199

        # register some images for use in the AutoComplete box.

   200

        #self.RegisterImage(1, images.getSmilesBitmap())

   201

        self.PythonCodeEditor.RegisterImage(1, 

   202

            wx.ArtProvider.GetBitmap(wx.ART_DELETE, size=(16,16)))

   203

        self.PythonCodeEditor.RegisterImage(2, 

   204

            wx.ArtProvider.GetBitmap(wx.ART_NEW, size=(16,16)))

   205

        self.PythonCodeEditor.RegisterImage(3, 

   206

            wx.ArtProvider.GetBitmap(wx.ART_COPY, size=(16,16)))

   207

   208

        # Indentation and tab stuff

   209

        self.PythonCodeEditor.SetIndent(4)               # Proscribed indent size for wx

   210

        self.PythonCodeEditor.SetIndentationGuides(True) # Show indent guides

   211

        self.PythonCodeEditor.SetBackSpaceUnIndents(True)# Backspace unindents rather than delete 1 space

   212

        self.PythonCodeEditor.SetTabIndents(True)        # Tab key indents

   213

        self.PythonCodeEditor.SetTabWidth(4)             # Proscribed tab size for wx

   214

        self.PythonCodeEditor.SetUseTabs(False)          # Use spaces rather than tabs, or

   215

                                        # TabTimmy will complain!    

   216

        # White space

   217

        self.PythonCodeEditor.SetViewWhiteSpace(False)   # Don't view white space

   218

   219

        # EOL: Since we are loading/saving ourselves, and the

   220

        # strings will always have \n's in them, set the STC to

   221

        # edit them that way.            

   222

        self.PythonCodeEditor.SetEOLMode(wx.stc.STC_EOL_LF)

   223

        self.PythonCodeEditor.SetViewEOL(False)

   224

   225

        # No right-edge mode indicator

   226

        self.PythonCodeEditor.SetEdgeMode(stc.STC_EDGE_NONE)

   227

   228

        self.PythonCodeEditor.SetModEventMask(wx.stc.STC_MOD_BEFOREINSERT|wx.stc.STC_MOD_BEFOREDELETE)

   229

   230

        self.PythonCodeEditor.Bind(wx.stc.EVT_STC_DO_DROP, self.OnDoDrop, id=ID_PYTHONEDITOR)

   231

        self.PythonCodeEditor.Bind(wx.EVT_KILL_FOCUS, self.OnKillFocus)

   232

        self.PythonCodeEditor.Bind(wx.stc.EVT_STC_MODIFIED, self.OnModification, id=ID_PYTHONEDITOR)

   233

   234

        return self.PythonCodeEditor

   235

   236

    def __init__(self, parent, controler, window):

   237

        ConfTreeNodeEditor.__init__(self, parent, controler, window)

   238

   239

        self.DisableEvents = False

   240

        self.CurrentAction = None

   241

   242

    def GetBufferState(self):

   243

        return self.Controler.GetBufferState()

   244

   245

    def Undo(self):

   246

        self.Controler.LoadPrevious()

   247

        self.RefreshView()

   248

   249

    def Redo(self):

   250

        self.Controler.LoadNext()

   251

        self.RefreshView()

   252

   253

    def OnModification(self, event):

   254

        if not self.DisableEvents:

   255

            mod_type = event.GetModificationType()

   256

            if not (mod_type&wx.stc.STC_PERFORMED_UNDO or mod_type&wx.stc.STC_PERFORMED_REDO):

   257

                if mod_type&wx.stc.STC_MOD_BEFOREINSERT:

   258

                    if self.CurrentAction is None:

   259

                        self.StartBuffering()

   260

                    elif self.CurrentAction[0] != "Add" or self.CurrentAction[1] != event.GetPosition() - 1:

   261

                        self.Controler.EndBuffering()

   262

                        self.StartBuffering()

   263

                    self.CurrentAction = ("Add", event.GetPosition())

   264

                    wx.CallAfter(self.RefreshModel)

   265

                elif mod_type&wx.stc.STC_MOD_BEFOREDELETE:

   266

                    if self.CurrentAction == None:

   267

                        self.StartBuffering()

   268

                    elif self.CurrentAction[0] != "Delete" or self.CurrentAction[1] != event.GetPosition() + 1:

   269

                        self.Controler.EndBuffering()

   270

                        self.StartBuffering()

   271

                    self.CurrentAction = ("Delete", event.GetPosition())

   272

                    wx.CallAfter(self.RefreshModel)

   273

        event.Skip()

   274

   275

    def OnDoDrop(self, event):

   276

        self.ResetBuffer()

   277

        wx.CallAfter(self.RefreshModel)

   278

        event.Skip()

   279

   280

    # Buffer the last model state

   281

    def RefreshBuffer(self):

   282

        self.Controler.BufferPython()

   283

        if self.ParentWindow is not None:

   284

            self.ParentWindow.RefreshTitle()

   285

            self.ParentWindow.RefreshFileMenu()

   286

            self.ParentWindow.RefreshEditMenu()

   287

            self.ParentWindow.RefreshPageTitles()

   288

   289

    def StartBuffering(self):

   290

        self.Controler.StartBuffering()

   291

        if self.ParentWindow is not None:

   292

            self.ParentWindow.RefreshTitle()

   293

            self.ParentWindow.RefreshFileMenu()

   294

            self.ParentWindow.RefreshEditMenu()

   295

            self.ParentWindow.RefreshPageTitles()

   296

   297

    def ResetBuffer(self):

   298

        if self.CurrentAction != None:

   299

            self.Controler.EndBuffering()

   300

            self.CurrentAction = None

   301

   302

    def RefreshView(self):

   303

        ConfTreeNodeEditor.RefreshView(self)

   304

   305

        self.ResetBuffer()

   306

        self.DisableEvents = True

   307

        old_cursor_pos = self.PythonCodeEditor.GetCurrentPos()

   308

        old_text = self.PythonCodeEditor.GetText()

   309

        new_text = self.Controler.GetPythonCode()

   310

        self.PythonCodeEditor.SetText(new_text)

   311

        new_cursor_pos = GetCursorPos(old_text, new_text)

   312

        if new_cursor_pos != None:

   313

            self.PythonCodeEditor.GotoPos(new_cursor_pos)

   314

        else:

   315

            self.PythonCodeEditor.GotoPos(old_cursor_pos)

   316

        self.PythonCodeEditor.ScrollToColumn(0)

   317

        self.PythonCodeEditor.EmptyUndoBuffer()

   318

        self.DisableEvents = False

   319

   320

        self.PythonCodeEditor.Colourise(0, -1)

   321

   322

    def RefreshModel(self):

   323

        self.Controler.SetPythonCode(self.PythonCodeEditor.GetText())

   324

   325

    def OnKeyPressed(self, event):

   326

        if self.PythonCodeEditor.CallTipActive():

   327

            self.PythonCodeEditor.CallTipCancel()

   328

        key = event.GetKeyCode()

   329

   330

        if key == 32 and event.ControlDown():

   331

            pos = self.PythonCodeEditor.GetCurrentPos()

   332

   333

            # Code completion

   334

            if not event.ShiftDown():

   335

                self.PythonCodeEditor.AutoCompSetIgnoreCase(False)  # so this needs to match

   336

   337

                # Images are specified with a appended "?type"

   338

                self.PythonCodeEditor.AutoCompShow(0, " ".join([word + "?1" for word in keyword.kwlist]))

   339

        else:

   340

            event.Skip()

   341

   342

    def OnKillFocus(self, event):

   343

        self.PythonCodeEditor.AutoCompCancel()

   344

        event.Skip()

   345

   346

    def OnUpdateUI(self, evt):

   347

        # check for matching braces

   348

        braceAtCaret = -1

   349

        braceOpposite = -1

   350

        charBefore = None

   351

        caretPos = self.PythonCodeEditor.GetCurrentPos()

   352

   353

        if caretPos > 0:

   354

            charBefore = self.PythonCodeEditor.GetCharAt(caretPos - 1)

   355

            styleBefore = self.PythonCodeEditor.GetStyleAt(caretPos - 1)

   356

   357

        # check before

   358

        if charBefore and chr(charBefore) in "[]{}()" and styleBefore == stc.STC_P_OPERATOR:

   359

            braceAtCaret = caretPos - 1

   360

   361

        # check after

   362

        if braceAtCaret < 0:

   363

            charAfter = self.PythonCodeEditor.GetCharAt(caretPos)

   364

            styleAfter = self.PythonCodeEditor.GetStyleAt(caretPos)

   365

   366

            if charAfter and chr(charAfter) in "[]{}()" and styleAfter == stc.STC_P_OPERATOR:

   367

                braceAtCaret = caretPos

   368

   369

        if braceAtCaret >= 0:

   370

            braceOpposite = self.PythonCodeEditor.BraceMatch(braceAtCaret)

   371

   372

        if braceAtCaret != -1  and braceOpposite == -1:

   373

            self.PythonCodeEditor.BraceBadLight(braceAtCaret)

   374

        else:

   375

            self.PythonCodeEditor.BraceHighlight(braceAtCaret, braceOpposite)

   376

   377

    def OnMarginClick(self, evt):

   378

        # fold and unfold as needed

   379

        if evt.GetMargin() == 2:

   380

            if evt.GetShift() and evt.GetControl():

   381

                self.FoldAll()

   382

            else:

   383

                lineClicked = self.PythonCodeEditor.LineFromPosition(evt.GetPosition())

   384

   385

                if self.PythonCodeEditor.GetFoldLevel(lineClicked) & stc.STC_FOLDLEVELHEADERFLAG:

   386

                    if evt.GetShift():

   387

                        self.PythonCodeEditor.SetFoldExpanded(lineClicked, True)

   388

                        self.Expand(lineClicked, True, True, 1)

   389

                    elif evt.GetControl():

   390

                        if self.PythonCodeEditor.GetFoldExpanded(lineClicked):

   391

                            self.PythonCodeEditor.SetFoldExpanded(lineClicked, False)

   392

                            self.Expand(lineClicked, False, True, 0)

   393

                        else:

   394

                            self.PythonCodeEditor.SetFoldExpanded(lineClicked, True)

   395

                            self.Expand(lineClicked, True, True, 100)

   396

                    else:

   397

                        self.PythonCodeEditor.ToggleFold(lineClicked)

   398

   399

   400

    def FoldAll(self):

   401

        lineCount = self.PythonCodeEditor.GetLineCount()

   402

        expanding = True

   403

   404

        # find out if we are folding or unfolding

   405

        for lineNum in range(lineCount):

   406

            if self.PythonCodeEditor.GetFoldLevel(lineNum) & stc.STC_FOLDLEVELHEADERFLAG:

   407

                expanding = not self.PythonCodeEditor.GetFoldExpanded(lineNum)

   408

                break

   409

   410

        lineNum = 0

   411

   412

        while lineNum < lineCount:

   413

            level = self.PythonCodeEditor.GetFoldLevel(lineNum)

   414

            if level & stc.STC_FOLDLEVELHEADERFLAG and \

   415

               (level & stc.STC_FOLDLEVELNUMBERMASK) == stc.STC_FOLDLEVELBASE:

   416

   417

                if expanding:

   418

                    self.PythonCodeEditor.SetFoldExpanded(lineNum, True)

   419

                    lineNum = self.Expand(lineNum, True)

   420

                    lineNum = lineNum - 1

   421

                else:

   422

                    lastChild = self.PythonCodeEditor.GetLastChild(lineNum, -1)

   423

                    self.PythonCodeEditor.SetFoldExpanded(lineNum, False)

   424

   425

                    if lastChild > lineNum:

   426

                        self.PythonCodeEditor.HideLines(lineNum+1, lastChild)

   427

   428

            lineNum = lineNum + 1

   429

   430

   431

   432

    def Expand(self, line, doExpand, force=False, visLevels=0, level=-1):

   433

        lastChild = self.PythonCodeEditor.GetLastChild(line, level)

   434

        line = line + 1

   435

   436

        while line <= lastChild:

   437

            if force:

   438

                if visLevels > 0:

   439

                    self.PythonCodeEditor.ShowLines(line, line)

   440

                else:

   441

                    self.PythonCodeEditor.HideLines(line, line)

   442

            else:

   443

                if doExpand:

   444

                    self.PythonCodeEditor.ShowLines(line, line)

   445

   446

            if level == -1:

   447

                level = self.PythonCodeEditor.GetFoldLevel(line)

   448

   449

            if level & stc.STC_FOLDLEVELHEADERFLAG:

   450

                if force:

   451

                    if visLevels > 1:

   452

                        self.PythonCodeEditor.SetFoldExpanded(line, True)

   453

                    else:

   454

                        self.PythonCodeEditor.SetFoldExpanded(line, False)

   455

   456

                    line = self.Expand(line, doExpand, force, visLevels-1)

   457

   458

                else:

   459

                    if doExpand and self.PythonCodeEditor.GetFoldExpanded(line):

   460

                        line = self.Expand(line, True, force, visLevels-1)

   461

                    else:

   462

                        line = self.Expand(line, False, force, visLevels-1)

   463

            else:

   464

                line = line + 1

   465

   466

        return line

   467

   468

    def Cut(self):

   469

        self.ResetBuffer()

   470

        self.DisableEvents = True

   471

        self.PythonCodeEditor.CmdKeyExecute(wx.stc.STC_CMD_CUT)

   472

        self.DisableEvents = False

   473

        self.RefreshModel()

   474

        self.RefreshBuffer()

   475

   476

    def Copy(self):

   477

        self.PythonCodeEditor.CmdKeyExecute(wx.stc.STC_CMD_COPY)

   478

   479

    def Paste(self):

   480

        self.ResetBuffer()

   481

        self.DisableEvents = True

   482

        self.PythonCodeEditor.CmdKeyExecute(wx.stc.STC_CMD_PASTE)

   483

        self.DisableEvents = False

   484

        self.RefreshModel()

   485

        self.RefreshBuffer()