1

我正在尝试在 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])
4

1 回答 1

1

关于性能,这是一个常见问题解答。关于你的算法:当你到达subdirA piter点时subdirA,在下一次迭代中,当你到达时a.py piter仍然指向subdirA

正如你所说,使用os.walk.

于 2010-03-31T09:36:58.910 回答