Tools/idle/ClassBrowser.py

"""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])