David Scherer | 7aced17 | 2000-08-15 01:13:23 +0000 | [diff] [blame] | 1 | import os |
| 2 | import sys |
| 3 | import imp |
Brett Cannon | cb66eb0 | 2012-05-11 12:58:42 -0400 | [diff] [blame] | 4 | import importlib.machinery |
David Scherer | 7aced17 | 2000-08-15 01:13:23 +0000 | [diff] [blame] | 5 | |
Kurt B. Kaiser | 2d7f6a0 | 2007-08-22 23:01:33 +0000 | [diff] [blame] | 6 | from idlelib.TreeWidget import TreeItem |
| 7 | from idlelib.ClassBrowser import ClassBrowser, ModuleBrowserTreeItem |
David Scherer | 7aced17 | 2000-08-15 01:13:23 +0000 | [diff] [blame] | 8 | |
| 9 | class PathBrowser(ClassBrowser): |
Kurt B. Kaiser | bec5e5c | 2001-07-13 19:57:00 +0000 | [diff] [blame] | 10 | |
David Scherer | 7aced17 | 2000-08-15 01:13:23 +0000 | [diff] [blame] | 11 | def __init__(self, flist): |
| 12 | self.init(flist) |
| 13 | |
| 14 | def settitle(self): |
| 15 | self.top.wm_title("Path Browser") |
| 16 | self.top.wm_iconname("Path Browser") |
| 17 | |
| 18 | def rootnode(self): |
| 19 | return PathBrowserTreeItem() |
| 20 | |
| 21 | class PathBrowserTreeItem(TreeItem): |
| 22 | |
| 23 | def GetText(self): |
| 24 | return "sys.path" |
| 25 | |
| 26 | def GetSubList(self): |
| 27 | sublist = [] |
| 28 | for dir in sys.path: |
| 29 | item = DirBrowserTreeItem(dir) |
| 30 | sublist.append(item) |
| 31 | return sublist |
| 32 | |
| 33 | class DirBrowserTreeItem(TreeItem): |
| 34 | |
| 35 | def __init__(self, dir, packages=[]): |
| 36 | self.dir = dir |
| 37 | self.packages = packages |
| 38 | |
| 39 | def GetText(self): |
| 40 | if not self.packages: |
| 41 | return self.dir |
| 42 | else: |
| 43 | return self.packages[-1] + ": package" |
| 44 | |
| 45 | def GetSubList(self): |
| 46 | try: |
| 47 | names = os.listdir(self.dir or os.curdir) |
Andrew Svetlov | 786fbd8 | 2012-12-17 19:51:15 +0200 | [diff] [blame] | 48 | except OSError: |
David Scherer | 7aced17 | 2000-08-15 01:13:23 +0000 | [diff] [blame] | 49 | return [] |
| 50 | packages = [] |
| 51 | for name in names: |
| 52 | file = os.path.join(self.dir, name) |
| 53 | if self.ispackagedir(file): |
| 54 | nn = os.path.normcase(name) |
| 55 | packages.append((nn, name, file)) |
| 56 | packages.sort() |
| 57 | sublist = [] |
| 58 | for nn, name, file in packages: |
| 59 | item = DirBrowserTreeItem(file, self.packages + [name]) |
| 60 | sublist.append(item) |
| 61 | for nn, name in self.listmodules(names): |
| 62 | item = ModuleBrowserTreeItem(os.path.join(self.dir, name)) |
| 63 | sublist.append(item) |
| 64 | return sublist |
| 65 | |
| 66 | def ispackagedir(self, file): |
| 67 | if not os.path.isdir(file): |
| 68 | return 0 |
| 69 | init = os.path.join(file, "__init__.py") |
| 70 | return os.path.exists(init) |
| 71 | |
| 72 | def listmodules(self, allnames): |
| 73 | modules = {} |
Brett Cannon | cb66eb0 | 2012-05-11 12:58:42 -0400 | [diff] [blame] | 74 | suffixes = importlib.machinery.EXTENSION_SUFFIXES[:] |
| 75 | suffixes += importlib.machinery.SOURCE_SUFFIXES[:] |
| 76 | suffixes += importlib.machinery.BYTECODE_SUFFIXES[:] |
David Scherer | 7aced17 | 2000-08-15 01:13:23 +0000 | [diff] [blame] | 77 | sorted = [] |
Ned Deily | bfbf3f1 | 2012-10-18 18:16:44 -0700 | [diff] [blame] | 78 | for suff in suffixes: |
David Scherer | 7aced17 | 2000-08-15 01:13:23 +0000 | [diff] [blame] | 79 | i = -len(suff) |
| 80 | for name in allnames[:]: |
| 81 | normed_name = os.path.normcase(name) |
| 82 | if normed_name[i:] == suff: |
| 83 | mod_name = name[:i] |
Guido van Rossum | 811c4e0 | 2006-08-22 15:45:46 +0000 | [diff] [blame] | 84 | if mod_name not in modules: |
David Scherer | 7aced17 | 2000-08-15 01:13:23 +0000 | [diff] [blame] | 85 | modules[mod_name] = None |
| 86 | sorted.append((normed_name, name)) |
| 87 | allnames.remove(name) |
| 88 | sorted.sort() |
| 89 | return sorted |
| 90 | |
| 91 | def main(): |
Kurt B. Kaiser | 2d7f6a0 | 2007-08-22 23:01:33 +0000 | [diff] [blame] | 92 | from idlelib import PyShell |
David Scherer | 7aced17 | 2000-08-15 01:13:23 +0000 | [diff] [blame] | 93 | PathBrowser(PyShell.flist) |
| 94 | if sys.stdin is sys.__stdin__: |
| 95 | mainloop() |
| 96 | |
| 97 | if __name__ == "__main__": |
| 98 | main() |