gEcrit Code

#!/usr/bin/python
# -*- coding: utf-8 -*-

import wx
from configClass import *


class SrcBrowser(wx.TreeCtrl):
    """
    SrcBrowser

    Provides the necessary functions for collecting data and
    displays the date using a TreeCtrl.

    Used to display the classes and functions in the current
    file.
    """
    def __init__(self, TargetFile, nb, text_id, parent):
        """
        __init__

        Initializes the TreeCtrl object, gathers data and displays it
        in the TreeCtrl
        """
        wx.TreeCtrl.__init__(self, parent, text_id + 2000, pos=(0, 70),
                             size=(100, 100), name="Source Browser",
                             style=wx.TR_HIDE_ROOT | wx.TR_HAS_BUTTONS |
                             wx.TR_HAS_VARIABLE_ROW_HEIGHT)

        self.parent = parent
        if TargetFile != "" and TargetFile != "New Document":
            try:
                br_file = open(TargetFile, "r")
                n = 0
                root = self.AddRoot("Top")
                for line in br_file.readlines():
                    n += 1
                    if "class " in line and ":" in line:
                        root2 = self.AppendItem(root, line + " " + str(n))
                    elif " def " in line and ":" in line:
                        self.AppendItem(root2, line.strip(" def ") + " " +
                                str(n))
                    elif "def " in line and line[0] != " " and ":" in line:
                        self.AppendItem(root, line.strip("def ") + " " +
                                str(n))
                br_file.close()
            except:
                pass
        self.Bind(wx.EVT_TREE_ITEM_RIGHT_CLICK, lambda event: self.OnTreeClick(event,
                  text_id))

    def RefreshTree(self, text_id, file_br):
        """
        RefreshTree

        Finds the current document, gathers data and displays
        it in the TreeCtrl.
        """
        CurrentTree = wx.FindWindowById(text_id + 2000)
        CurrentTree.DeleteAllItems()

        if file_br != "" and file_br != "New Document":
            try:
                br_file = open(file_br, "r")
                n = 0
                root = CurrentTree.AddRoot("Top")
                for line in br_file.readlines():
                    n += 1
                    if "class " in line and ":" in line:
                        root2 = CurrentTree.AppendItem(root, line + " " +
                                str(n))
                    elif " def " in line and ":" in line:
                        CurrentTree.AppendItem(root2, line.strip(" def ") +
                                " " + str(n))
                    elif "def " in line and ":" in line and line[0] != " ":
                        CurrentTree.AppendItem(root, line.strip("def ") +
                                " " + str(n))
            except:
                pass

    def OnTreeClick(self, event, text_id):
        """
        OnTreeClick

        Scrolls the editor to the appropriate line upon right click.
        """
        TextWidget = wx.FindWindowById(text_id)
        TreeWidget = wx.FindWindowById(text_id + 2000)
        id = TreeWidget.GetSelection()

        text = TreeWidget.GetItemText(id)
        TextWidget.ScrollToLine(int(text.split(" ")[-1]) - 1)