mirror of
https://github.com/python/cpython.git
synced 2025-07-12 13:55:34 +00:00
a361e89d5a
Match Window on the main menu and remove last plural module name. Change imports, test, and attribute references to match new name.
248 lines
8.1 KiB
Python
248 lines
8.1 KiB
Python
"""Module browser.
|
|
|
|
XXX TO DO:
|
|
|
|
- reparse when source changed (maybe just a button would be OK?)
|
|
(or recheck on window popup)
|
|
- add popup menu with more options (e.g. doc strings, base classes, imports)
|
|
- add base classes to class browser tree
|
|
- finish removing limitation to x.py files (ModuleBrowserTreeItem)
|
|
"""
|
|
|
|
import os
|
|
import pyclbr
|
|
import sys
|
|
|
|
from idlelib.config import idleConf
|
|
from idlelib import pyshell
|
|
from idlelib.tree import TreeNode, TreeItem, ScrolledCanvas
|
|
from idlelib.window import ListedToplevel
|
|
|
|
|
|
file_open = None # Method...Item and Class...Item use this.
|
|
# Normally pyshell.flist.open, but there is no pyshell.flist for htest.
|
|
|
|
|
|
def transform_children(child_dict, modname=None):
|
|
"""Transform a child dictionary to an ordered sequence of objects.
|
|
|
|
The dictionary maps names to pyclbr information objects.
|
|
Filter out imported objects.
|
|
Augment class names with bases.
|
|
Sort objects by line number.
|
|
|
|
The current tree only calls this once per child_dic as it saves
|
|
TreeItems once created. A future tree and tests might violate this,
|
|
so a check prevents multiple in-place augmentations.
|
|
"""
|
|
obs = [] # Use list since values should already be sorted.
|
|
for key, obj in child_dict.items():
|
|
if modname is None or obj.module == modname:
|
|
if hasattr(obj, 'super') and obj.super and obj.name == key:
|
|
# If obj.name != key, it has already been suffixed.
|
|
supers = []
|
|
for sup in obj.super:
|
|
if type(sup) is type(''):
|
|
sname = sup
|
|
else:
|
|
sname = sup.name
|
|
if sup.module != obj.module:
|
|
sname = f'{sup.module}.{sname}'
|
|
supers.append(sname)
|
|
obj.name += '({})'.format(', '.join(supers))
|
|
obs.append(obj)
|
|
return sorted(obs, key=lambda o: o.lineno)
|
|
|
|
|
|
class ModuleBrowser:
|
|
"""Browse module classes and functions in IDLE.
|
|
"""
|
|
# This class is also the base class for pathbrowser.PathBrowser.
|
|
# Init and close are inherited, other methods are overridden.
|
|
# PathBrowser.__init__ does not call __init__ below.
|
|
|
|
def __init__(self, master, path, *, _htest=False, _utest=False):
|
|
"""Create a window for browsing a module's structure.
|
|
|
|
Args:
|
|
master: parent for widgets.
|
|
path: full path of file to browse.
|
|
_htest - bool; change box location when running htest.
|
|
-utest - bool; suppress contents when running unittest.
|
|
|
|
Global variables:
|
|
file_open: Function used for opening a file.
|
|
|
|
Instance variables:
|
|
name: Module name.
|
|
file: Full path and module with .py extension. Used in
|
|
creating ModuleBrowserTreeItem as the rootnode for
|
|
the tree and subsequently in the children.
|
|
"""
|
|
self.master = master
|
|
self.path = path
|
|
self._htest = _htest
|
|
self._utest = _utest
|
|
self.init()
|
|
|
|
def close(self, event=None):
|
|
"Dismiss the window and the tree nodes."
|
|
self.top.destroy()
|
|
self.node.destroy()
|
|
|
|
def init(self):
|
|
"Create browser tkinter widgets, including the tree."
|
|
global file_open
|
|
root = self.master
|
|
flist = (pyshell.flist if not (self._htest or self._utest)
|
|
else pyshell.PyShellFileList(root))
|
|
file_open = flist.open
|
|
pyclbr._modules.clear()
|
|
|
|
# create top
|
|
self.top = top = ListedToplevel(root)
|
|
top.protocol("WM_DELETE_WINDOW", self.close)
|
|
top.bind("<Escape>", self.close)
|
|
if self._htest: # place dialog below parent if running htest
|
|
top.geometry("+%d+%d" %
|
|
(root.winfo_rootx(), root.winfo_rooty() + 200))
|
|
self.settitle()
|
|
top.focus_set()
|
|
|
|
# create scrolled canvas
|
|
theme = idleConf.CurrentTheme()
|
|
background = idleConf.GetHighlight(theme, 'normal')['background']
|
|
sc = ScrolledCanvas(top, bg=background, highlightthickness=0,
|
|
takefocus=1)
|
|
sc.frame.pack(expand=1, fill="both")
|
|
item = self.rootnode()
|
|
self.node = node = TreeNode(sc.canvas, None, item)
|
|
if not self._utest:
|
|
node.update()
|
|
node.expand()
|
|
|
|
def settitle(self):
|
|
"Set the window title."
|
|
self.top.wm_title("Module Browser - " + os.path.basename(self.path))
|
|
self.top.wm_iconname("Module Browser")
|
|
|
|
def rootnode(self):
|
|
"Return a ModuleBrowserTreeItem as the root of the tree."
|
|
return ModuleBrowserTreeItem(self.path)
|
|
|
|
|
|
class ModuleBrowserTreeItem(TreeItem):
|
|
"""Browser tree for Python module.
|
|
|
|
Uses TreeItem as the basis for the structure of the tree.
|
|
Used by both browsers.
|
|
"""
|
|
|
|
def __init__(self, file):
|
|
"""Create a TreeItem for the file.
|
|
|
|
Args:
|
|
file: Full path and module name.
|
|
"""
|
|
self.file = file
|
|
|
|
def GetText(self):
|
|
"Return the module name as the text string to display."
|
|
return os.path.basename(self.file)
|
|
|
|
def GetIconName(self):
|
|
"Return the name of the icon to display."
|
|
return "python"
|
|
|
|
def GetSubList(self):
|
|
"Return ChildBrowserTreeItems for children."
|
|
return [ChildBrowserTreeItem(obj) for obj in self.listchildren()]
|
|
|
|
def OnDoubleClick(self):
|
|
"Open a module in an editor window when double clicked."
|
|
if os.path.normcase(self.file[-3:]) != ".py":
|
|
return
|
|
if not os.path.exists(self.file):
|
|
return
|
|
file_open(self.file)
|
|
|
|
def IsExpandable(self):
|
|
"Return True if Python (.py) file."
|
|
return os.path.normcase(self.file[-3:]) == ".py"
|
|
|
|
def listchildren(self):
|
|
"Return sequenced classes and functions in the module."
|
|
dir, base = os.path.split(self.file)
|
|
name, ext = os.path.splitext(base)
|
|
if os.path.normcase(ext) != ".py":
|
|
return []
|
|
try:
|
|
tree = pyclbr.readmodule_ex(name, [dir] + sys.path)
|
|
except ImportError:
|
|
return []
|
|
return transform_children(tree, name)
|
|
|
|
|
|
class ChildBrowserTreeItem(TreeItem):
|
|
"""Browser tree for child nodes within the module.
|
|
|
|
Uses TreeItem as the basis for the structure of the tree.
|
|
"""
|
|
|
|
def __init__(self, obj):
|
|
"Create a TreeItem for a pyclbr class/function object."
|
|
self.obj = obj
|
|
self.name = obj.name
|
|
self.isfunction = isinstance(obj, pyclbr.Function)
|
|
|
|
def GetText(self):
|
|
"Return the name of the function/class to display."
|
|
name = self.name
|
|
if self.isfunction:
|
|
return "def " + name + "(...)"
|
|
else:
|
|
return "class " + name
|
|
|
|
def GetIconName(self):
|
|
"Return the name of the icon to display."
|
|
if self.isfunction:
|
|
return "python"
|
|
else:
|
|
return "folder"
|
|
|
|
def IsExpandable(self):
|
|
"Return True if self.obj has nested objects."
|
|
return self.obj.children != {}
|
|
|
|
def GetSubList(self):
|
|
"Return ChildBrowserTreeItems for children."
|
|
return [ChildBrowserTreeItem(obj)
|
|
for obj in transform_children(self.obj.children)]
|
|
|
|
def OnDoubleClick(self):
|
|
"Open module with file_open and position to lineno."
|
|
try:
|
|
edit = file_open(self.obj.file)
|
|
edit.gotoline(self.obj.lineno)
|
|
except (OSError, AttributeError):
|
|
pass
|
|
|
|
|
|
def _module_browser(parent): # htest #
|
|
if len(sys.argv) > 1: # If pass file on command line.
|
|
file = sys.argv[1]
|
|
else:
|
|
file = __file__
|
|
# Add nested objects for htest.
|
|
class Nested_in_func(TreeNode):
|
|
def nested_in_class(): pass
|
|
def closure():
|
|
class Nested_in_closure: pass
|
|
ModuleBrowser(parent, file, _htest=True)
|
|
|
|
if __name__ == "__main__":
|
|
if len(sys.argv) == 1: # If pass file on command line, unittest fails.
|
|
from unittest import main
|
|
main('idlelib.idle_test.test_browser', verbosity=2, exit=False)
|
|
from idlelib.idle_test.htest import run
|
|
run(_module_browser)
|