我正在尝试在 pyGTK 中生成分层目录列表。
目前,我有以下目录树:
/root
folderA
- subdirA
- subA.py
- a.py
folderB
- b.py
我写了一个几乎可以工作的函数:
def go(root, piter=None):
for filename in os.listdir(root):
isdir = os.path.isdir(os.path.join(root, filename))
piter = self.treestore.append(piter, [filename])
if isdir == True:
go(os.path.join(root, filename), piter)
This is what i get when i run the app:
我还认为我的函数效率低下,我应该使用 os.walk(),因为它已经存在用于此目的。
我怎样才能使用 pyGTK 生成目录树,以及什么是正确/最有效的方法?
- -编辑 - -我最终使用的有效代码块是:
parents = {}
for dir, dirs, files in os.walk(root):
for subdir in dirs:
parents[os.path.join(dir, subdir)] = self.treestore.append(parents.get(dir, None), [subdir])
for item in files:
self.treestore.append(parents.get(dir, None), [item])