玩蛇网提供最新Python编程技术信息以及Python资源下载!

Python实现自动生成目录树的方法示例源码

python 培训

python语言来完成目录树的建立,这样的python代码要如何编写呢?本文就是关于Python实现自动生成目录树的方法示例源码,给大家做个参考吧。

Python实现自动生成目录树的方法示例源码如下:需要用到python os模块

# 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.iplaypython.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')   
        f.writelines(self.list)   
        print "ok"   
        f.close()   

if __name__ == '__main__':   
    d = dir()   
    d.getDirList("c:/windows") # input directory  
    d.writeList("c:/1.txt") # write to file 

玩蛇网原创,转载请注明文章出处和来源网址:http://www.iplaypython.com/code/base/b2548.html



微信公众号搜索"玩蛇网Python之家"加关注,每日最新的Python资讯、图文视频教程可以让你一手全掌握。强烈推荐关注!

微信扫描下图可直接关注

玩蛇网Python新手QQ群,欢迎加入: ① 240764603 玩蛇网Python新手群
文章发布日期:2016-03-28 15:08 玩蛇网 www.iplaypython.com

评论列表(网友评论仅供网友表达个人看法,并不表明本站同意其观点或证实其描述)
相关文章推荐
别人正在看
特别推荐
去顶部去底部