162 lines
4.8 KiB
Python
162 lines
4.8 KiB
Python
"""Primitive class browser.
|
|
|
|
XXX TO DO:
|
|
|
|
- generalize the scrolling listbox with some behavior into a base class
|
|
- add popup menu with more options (e.g. doc strings, base classes, imports)
|
|
- show function argument list (have to do pattern matching on source)
|
|
- should the classes and methods lists also be in the module's menu bar?
|
|
|
|
"""
|
|
|
|
import os
|
|
import string
|
|
import pyclbr
|
|
from Tkinter import *
|
|
import tkMessageBox
|
|
|
|
from ScrolledList import ScrolledList
|
|
|
|
|
|
class ClassBrowser:
|
|
|
|
def __init__(self, flist, name):
|
|
root = flist.root
|
|
try:
|
|
dict = pyclbr.readmodule(name)
|
|
except ImportError, msg:
|
|
tkMessageBox.showerror("Import error", str(msg), parent=root)
|
|
return
|
|
if not dict:
|
|
tkMessageBox.showerror("Nothing to browse",
|
|
"Module %s defines no classes" % name, parent=root)
|
|
return
|
|
self.flist = flist
|
|
self.dict = dict
|
|
self.root = root
|
|
self.top = top = Toplevel(root)
|
|
self.top.protocol("WM_DELETE_WINDOW", self.close)
|
|
top.wm_title("Class browser")
|
|
self.leftframe = leftframe = Frame(top)
|
|
self.leftframe.pack(side="left", fill="both", expand=1)
|
|
# Create help label
|
|
self.helplabel = Label(leftframe, text="Module %s" % name,
|
|
relief="groove", borderwidth=2)
|
|
self.helplabel.pack(fill="x")
|
|
# Create top frame, with scrollbar and listbox
|
|
self.classviewer = ClassViewer(
|
|
self.leftframe, self.flist, self)
|
|
# Load the classes
|
|
self.load_classes(dict, name)
|
|
|
|
def close(self):
|
|
self.classviewer = None
|
|
self.methodviewer = None
|
|
self.top.destroy()
|
|
|
|
def load_classes(self, dict, module):
|
|
self.classviewer.load_classes(dict, module)
|
|
if self.botframe:
|
|
self.botframe.destroy()
|
|
self.botframe = None
|
|
self.methodviewer = None
|
|
|
|
botframe = None
|
|
methodhelplabel = None
|
|
methodviewer = None
|
|
|
|
def show_methods(self, cl):
|
|
if not self.botframe:
|
|
self.botframe = Frame(self.top)
|
|
self.botframe.pack(side="right", expand=1, fill="both")
|
|
self.methodhelplabel = Label(self.botframe,
|
|
relief="groove", borderwidth=2)
|
|
self.methodhelplabel.pack(fill="x")
|
|
self.methodviewer = MethodViewer(self.botframe, self.flist)
|
|
self.methodhelplabel.config(text="Class %s" % cl.name)
|
|
self.methodviewer.load_methods(cl)
|
|
|
|
|
|
class ClassViewer(ScrolledList):
|
|
|
|
def __init__(self, master, flist, browser):
|
|
ScrolledList.__init__(self, master)
|
|
self.flist = flist
|
|
self.browser = browser
|
|
|
|
def load_classes(self, dict, module):
|
|
self.clear()
|
|
self.dict = dict
|
|
items = []
|
|
for key, value in dict.items():
|
|
if value.module == module:
|
|
items.append((value.lineno, key, value))
|
|
items.sort()
|
|
for lineno, key, value in items:
|
|
s = key
|
|
if value.super:
|
|
super = []
|
|
for sup in value.super:
|
|
name = sup.name
|
|
if sup.module != value.module:
|
|
name = "%s.%s" % (sup.module, name)
|
|
super.append(name)
|
|
s = s + "(%s)" % string.join(super, ", ")
|
|
self.append(s)
|
|
|
|
def getname(self, index):
|
|
name = self.listbox.get(index)
|
|
i = string.find(name, '(')
|
|
if i >= 0:
|
|
name = name[:i]
|
|
return name
|
|
|
|
def getclass(self, index):
|
|
return self.dict[self.getname(index)]
|
|
|
|
def on_select(self, index):
|
|
self.show_methods(index)
|
|
|
|
def on_double(self, index):
|
|
self.show_source(index)
|
|
|
|
def show_methods(self, index):
|
|
cl = self.getclass(index)
|
|
self.browser.show_methods(cl)
|
|
|
|
def show_source(self, index):
|
|
cl = self.getclass(index)
|
|
if os.path.isfile(cl.file):
|
|
edit = self.flist.open(cl.file)
|
|
edit.gotoline(cl.lineno)
|
|
|
|
|
|
class MethodViewer(ScrolledList):
|
|
|
|
def __init__(self, master, flist):
|
|
ScrolledList.__init__(self, master)
|
|
self.flist = flist
|
|
|
|
classinfo = None
|
|
|
|
def load_methods(self, cl):
|
|
self.classinfo = cl
|
|
self.clear()
|
|
items = []
|
|
for name, lineno in cl.methods.items():
|
|
items.append((lineno, name))
|
|
items.sort()
|
|
for item, name in items:
|
|
self.append(name)
|
|
|
|
def click_event(self, event):
|
|
pass
|
|
|
|
def on_double(self, index):
|
|
self.show_source(self.get(index))
|
|
|
|
def show_source(self, name):
|
|
if os.path.isfile(self.classinfo.file):
|
|
edit = self.flist.open(self.classinfo.file)
|
|
edit.gotoline(self.classinfo.methods[name])
|