Python文件目录树

2018-9-10 4:02

102 字

3 分钟

为了看清文件夹里面的文件是如何分布的，或者是想生成代码的文件结构，可以使用下方的Python代码快速生成文件目录树。

# encoding: utf-8

import os

class dir(object):
    def __init__(self):
        self.SPACE = ""
        self.list = []

    def getCount(self, url):
        files = os.listdir(url)
        count = 0;
        for file in files:
            myfile = url + "\\" + file
            if os.path.isfile(myfile):
                count = count + 1
        return count

    def getDirList(self, url):
        files = os.listdir(url)
        fileNum = self.getCount(url)
        tmpNum = 0

        for file in files:
            myfile = url + "\\" + file
            size = os.path.getsize(myfile)

            if os.path.isfile(myfile):
                tmpNum = tmpNum + 1
                if (tmpNum != fileNum):
                    self.list.append(str(self.SPACE) + "├─" + file + "\n")
                else:  # www.iplaypy.com
                    self.list.append(str(self.SPACE) + "└─" + file + "\n")

            if os.path.isdir(myfile):
                self.list.append(str(self.SPACE) + "├─" + file + "\n")
                # change into sub directory
                self.SPACE = self.SPACE + "│  "
                self.getDirList(myfile)
                # if iterator of sub directory is finished, reduce "│  "
                self.SPACE = self.SPACE[:-4]
        return self.list

    def writeList(self, url):
        f = open(url, 'w',encoding='utf-8')
        f.writelines(self.list)
        print
        "ok"
        f.close()


if __name__ == '__main__':
    d = dir()
    d.getDirList("E:\\按类别分")  # input directory
    d.writeList("F:\\文件目录.txt")  # write to file

Python