Lib/idlelib/PathBrowser.py

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