import wx
from configClass import *
Config = ConfigNanny()
class SrcBrowser(wx.TreeCtrl):
def __init__(self,TargetFile, nb, text_id, parent):
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)
if TargetFile != "" and TargetFile != "New Document":
br_file = open(TargetFile, "r")
n=0
root = self.AddRoot("Top")
for line in br_file.readlines():
n+=1
if "class " in line:
root2 = self.AppendItem(root,line+" "+str(n))
elif " def " in line:
self.AppendItem(root2, line.strip(" def ")+" "+str(n))
elif "def " in line and line[0] != " ":
self.AppendItem(root, line.strip("def ")+" "+str(n))
self.Bind(wx.EVT_TREE_ITEM_RIGHT_CLICK, lambda event: self.OnTreeClick(event, text_id))
def RefreshTree(self,text_id, file_br):
CurrentTree = wx.FindWindowById(text_id+2000)
CurrentTree.DeleteAllItems()
if file_br != "" and file_br != "New Document":
br_file = open(file_br, "r")
n=0
root = CurrentTree.AddRoot("Top")
for line in br_file.readlines():
n+=1
if "class " in line:
root2 = CurrentTree.AppendItem(root,line+" "+str(n))
elif " def " in line:
CurrentTree.AppendItem(root2, line.strip(" def ")+" "+str(n))
elif "def " in line and line[0] != " ":
CurrentTree.AppendItem(root, line.strip("def ")+" "+str(n))
def OnTreeClick(self,event,text_id):
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)