Terry Jan Reedy | bfbaa6b | 2016-08-31 00:50:55 -0400 | [diff] [blame] | 1 | import importlib.machinery |
David Scherer | 7aced17 | 2000-08-15 01:13:23 +0000 | [diff] [blame] | 2 | import os |
| 3 | import sys |
David Scherer | 7aced17 | 2000-08-15 01:13:23 +0000 | [diff] [blame] | 4 | |
Cheryl Sabella | cd99e79 | 2017-09-23 16:46:01 -0400 | [diff] [blame] | 5 | from idlelib.browser import ModuleBrowser, ModuleBrowserTreeItem |
Terry Jan Reedy | bfbaa6b | 2016-08-31 00:50:55 -0400 | [diff] [blame] | 6 | from idlelib.tree import TreeItem |
Terry Jan Reedy | 1b392ff | 2014-05-24 18:48:18 -0400 | [diff] [blame] | 7 | |
David Scherer | 7aced17 | 2000-08-15 01:13:23 +0000 | [diff] [blame] | 8 | |
Cheryl Sabella | cd99e79 | 2017-09-23 16:46:01 -0400 | [diff] [blame] | 9 | class PathBrowser(ModuleBrowser): |
Kurt B. Kaiser | bec5e5c | 2001-07-13 19:57:00 +0000 | [diff] [blame] | 10 | |
Cheryl Sabella | 20d48a4 | 2017-11-22 19:05:25 -0500 | [diff] [blame] | 11 | def __init__(self, master, *, _htest=False, _utest=False): |
Terry Jan Reedy | 1b392ff | 2014-05-24 18:48:18 -0400 | [diff] [blame] | 12 | """ |
| 13 | _htest - bool, change box location when running htest |
| 14 | """ |
Cheryl Sabella | 20d48a4 | 2017-11-22 19:05:25 -0500 | [diff] [blame] | 15 | self.master = master |
Terry Jan Reedy | 1b392ff | 2014-05-24 18:48:18 -0400 | [diff] [blame] | 16 | self._htest = _htest |
Cheryl Sabella | 058de11 | 2017-09-22 16:08:44 -0400 | [diff] [blame] | 17 | self._utest = _utest |
Cheryl Sabella | 20d48a4 | 2017-11-22 19:05:25 -0500 | [diff] [blame] | 18 | self.init() |
David Scherer | 7aced17 | 2000-08-15 01:13:23 +0000 | [diff] [blame] | 19 | |
| 20 | def settitle(self): |
Terry Jan Reedy | 19c1a87 | 2015-07-20 17:44:59 -0400 | [diff] [blame] | 21 | "Set window titles." |
David Scherer | 7aced17 | 2000-08-15 01:13:23 +0000 | [diff] [blame] | 22 | self.top.wm_title("Path Browser") |
| 23 | self.top.wm_iconname("Path Browser") |
| 24 | |
| 25 | def rootnode(self): |
| 26 | return PathBrowserTreeItem() |
| 27 | |
Terry Jan Reedy | bfbaa6b | 2016-08-31 00:50:55 -0400 | [diff] [blame] | 28 | |
David Scherer | 7aced17 | 2000-08-15 01:13:23 +0000 | [diff] [blame] | 29 | class PathBrowserTreeItem(TreeItem): |
| 30 | |
| 31 | def GetText(self): |
| 32 | return "sys.path" |
| 33 | |
| 34 | def GetSubList(self): |
| 35 | sublist = [] |
| 36 | for dir in sys.path: |
| 37 | item = DirBrowserTreeItem(dir) |
| 38 | sublist.append(item) |
| 39 | return sublist |
| 40 | |
Terry Jan Reedy | bfbaa6b | 2016-08-31 00:50:55 -0400 | [diff] [blame] | 41 | |
David Scherer | 7aced17 | 2000-08-15 01:13:23 +0000 | [diff] [blame] | 42 | class DirBrowserTreeItem(TreeItem): |
| 43 | |
| 44 | def __init__(self, dir, packages=[]): |
| 45 | self.dir = dir |
| 46 | self.packages = packages |
| 47 | |
| 48 | def GetText(self): |
| 49 | if not self.packages: |
| 50 | return self.dir |
| 51 | else: |
| 52 | return self.packages[-1] + ": package" |
| 53 | |
| 54 | def GetSubList(self): |
| 55 | try: |
| 56 | names = os.listdir(self.dir or os.curdir) |
Andrew Svetlov | 786fbd8 | 2012-12-17 19:51:15 +0200 | [diff] [blame] | 57 | except OSError: |
David Scherer | 7aced17 | 2000-08-15 01:13:23 +0000 | [diff] [blame] | 58 | return [] |
| 59 | packages = [] |
| 60 | for name in names: |
| 61 | file = os.path.join(self.dir, name) |
| 62 | if self.ispackagedir(file): |
| 63 | nn = os.path.normcase(name) |
| 64 | packages.append((nn, name, file)) |
| 65 | packages.sort() |
| 66 | sublist = [] |
| 67 | for nn, name, file in packages: |
| 68 | item = DirBrowserTreeItem(file, self.packages + [name]) |
| 69 | sublist.append(item) |
| 70 | for nn, name in self.listmodules(names): |
| 71 | item = ModuleBrowserTreeItem(os.path.join(self.dir, name)) |
| 72 | sublist.append(item) |
| 73 | return sublist |
| 74 | |
| 75 | def ispackagedir(self, file): |
Terry Jan Reedy | 19c1a87 | 2015-07-20 17:44:59 -0400 | [diff] [blame] | 76 | " Return true for directories that are packages." |
David Scherer | 7aced17 | 2000-08-15 01:13:23 +0000 | [diff] [blame] | 77 | if not os.path.isdir(file): |
Terry Jan Reedy | 19c1a87 | 2015-07-20 17:44:59 -0400 | [diff] [blame] | 78 | return False |
David Scherer | 7aced17 | 2000-08-15 01:13:23 +0000 | [diff] [blame] | 79 | init = os.path.join(file, "__init__.py") |
| 80 | return os.path.exists(init) |
| 81 | |
| 82 | def listmodules(self, allnames): |
| 83 | modules = {} |
Brett Cannon | cb66eb0 | 2012-05-11 12:58:42 -0400 | [diff] [blame] | 84 | suffixes = importlib.machinery.EXTENSION_SUFFIXES[:] |
Terry Jan Reedy | 19c1a87 | 2015-07-20 17:44:59 -0400 | [diff] [blame] | 85 | suffixes += importlib.machinery.SOURCE_SUFFIXES |
| 86 | suffixes += importlib.machinery.BYTECODE_SUFFIXES |
David Scherer | 7aced17 | 2000-08-15 01:13:23 +0000 | [diff] [blame] | 87 | sorted = [] |
Ned Deily | bfbf3f1 | 2012-10-18 18:16:44 -0700 | [diff] [blame] | 88 | for suff in suffixes: |
David Scherer | 7aced17 | 2000-08-15 01:13:23 +0000 | [diff] [blame] | 89 | i = -len(suff) |
| 90 | for name in allnames[:]: |
| 91 | normed_name = os.path.normcase(name) |
| 92 | if normed_name[i:] == suff: |
| 93 | mod_name = name[:i] |
Guido van Rossum | 811c4e0 | 2006-08-22 15:45:46 +0000 | [diff] [blame] | 94 | if mod_name not in modules: |
David Scherer | 7aced17 | 2000-08-15 01:13:23 +0000 | [diff] [blame] | 95 | modules[mod_name] = None |
| 96 | sorted.append((normed_name, name)) |
| 97 | allnames.remove(name) |
| 98 | sorted.sort() |
| 99 | return sorted |
| 100 | |
Terry Jan Reedy | bfbaa6b | 2016-08-31 00:50:55 -0400 | [diff] [blame] | 101 | |
Terry Jan Reedy | 19c1a87 | 2015-07-20 17:44:59 -0400 | [diff] [blame] | 102 | def _path_browser(parent): # htest # |
Cheryl Sabella | 20d48a4 | 2017-11-22 19:05:25 -0500 | [diff] [blame] | 103 | PathBrowser(parent, _htest=True) |
Terry Jan Reedy | a2fc99e | 2014-05-25 18:44:05 -0400 | [diff] [blame] | 104 | parent.mainloop() |
David Scherer | 7aced17 | 2000-08-15 01:13:23 +0000 | [diff] [blame] | 105 | |
| 106 | if __name__ == "__main__": |
Terry Jan Reedy | db4e5c5 | 2013-05-27 21:32:03 -0400 | [diff] [blame] | 107 | from unittest import main |
| 108 | main('idlelib.idle_test.test_pathbrowser', verbosity=2, exit=False) |
Terry Jan Reedy | 1b392ff | 2014-05-24 18:48:18 -0400 | [diff] [blame] | 109 | |
| 110 | from idlelib.idle_test.htest import run |
| 111 | run(_path_browser) |