我正在尝试在 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)
这是我运行应用程序时得到的:
我还认为我的函数效率低下,我应该使用 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])