Python code coverage for Lib/idlelib/PathBrowser.py

#	count	content
1	n/a	import os
2	n/a	import sys
3	n/a	import importlib.machinery
4	n/a
5	n/a	from idlelib.TreeWidget import TreeItem
6	n/a	from idlelib.ClassBrowser import ClassBrowser, ModuleBrowserTreeItem
7	n/a
8	n/a	class PathBrowser(ClassBrowser):
9	n/a
10	n/a	def __init__(self, flist):
11	n/a	self.init(flist)
12	n/a
13	n/a	def settitle(self):
14	n/a	self.top.wm_title("Path Browser")
15	n/a	self.top.wm_iconname("Path Browser")
16	n/a
17	n/a	def rootnode(self):
18	n/a	return PathBrowserTreeItem()
19	n/a
20	n/a	class PathBrowserTreeItem(TreeItem):
21	n/a
22	n/a	def GetText(self):
23	n/a	return "sys.path"
24	n/a
25	n/a	def GetSubList(self):
26	n/a	sublist = []
27	n/a	for dir in sys.path:
28	n/a	item = DirBrowserTreeItem(dir)
29	n/a	sublist.append(item)
30	n/a	return sublist
31	n/a
32	n/a	class DirBrowserTreeItem(TreeItem):
33	n/a
34	n/a	def __init__(self, dir, packages=[]):
35	n/a	self.dir = dir
36	n/a	self.packages = packages
37	n/a
38	n/a	def GetText(self):
39	n/a	if not self.packages:
40	n/a	return self.dir
41	n/a	else:
42	n/a	return self.packages[-1] + ": package"
43	n/a
44	n/a	def GetSubList(self):
45	n/a	try:
46	n/a	names = os.listdir(self.dir or os.curdir)
47	n/a	except OSError:
48	n/a	return []
49	n/a	packages = []
50	n/a	for name in names:
51	n/a	file = os.path.join(self.dir, name)
52	n/a	if self.ispackagedir(file):
53	n/a	nn = os.path.normcase(name)
54	n/a	packages.append((nn, name, file))
55	n/a	packages.sort()
56	n/a	sublist = []
57	n/a	for nn, name, file in packages:
58	n/a	item = DirBrowserTreeItem(file, self.packages + [name])
59	n/a	sublist.append(item)
60	n/a	for nn, name in self.listmodules(names):
61	n/a	item = ModuleBrowserTreeItem(os.path.join(self.dir, name))
62	n/a	sublist.append(item)
63	n/a	return sublist
64	n/a
65	n/a	def ispackagedir(self, file):
66	n/a	if not os.path.isdir(file):
67	n/a	return 0
68	n/a	init = os.path.join(file, "__init__.py")
69	n/a	return os.path.exists(init)
70	n/a
71	n/a	def listmodules(self, allnames):
72	n/a	modules = {}
73	n/a	suffixes = importlib.machinery.EXTENSION_SUFFIXES[:]
74	n/a	suffixes += importlib.machinery.SOURCE_SUFFIXES[:]
75	n/a	suffixes += importlib.machinery.BYTECODE_SUFFIXES[:]
76	n/a	sorted = []
77	n/a	for suff in suffixes:
78	n/a	i = -len(suff)
79	n/a	for name in allnames[:]:
80	n/a	normed_name = os.path.normcase(name)
81	n/a	if normed_name[i:] == suff:
82	n/a	mod_name = name[:i]
83	n/a	if mod_name not in modules:
84	n/a	modules[mod_name] = None
85	n/a	sorted.append((normed_name, name))
86	n/a	allnames.remove(name)
87	n/a	sorted.sort()
88	n/a	return sorted
89	n/a
90	n/a	def main():
91	n/a	from idlelib import PyShell
92	n/a	PathBrowser(PyShell.flist)
93	n/a	if sys.stdin is sys.__stdin__:
94	n/a	mainloop()
95	n/a
96	n/a	if __name__ == "__main__":
97	n/a	from unittest import main
98	n/a	main('idlelib.idle_test.test_pathbrowser', verbosity=2, exit=False)