服务器之家

服务器之家 > 正文

使用python生成目录树

时间:2021-01-25 00:33     来源/作者:yjp2006

这是一个使用Python生成文件、目录树的程序,其中遇到一个问题是:如何确定某个目录下的某一文件是最后一个遍历的文件。因为最后一个遍历的文件前应添加"└─",非最后一个文件前添加"├─"。看了Python的API文档没有找到相关的系统函数。现在做法是:先统计出某个目录下的文件个数,在遍历目录时,当个数相等时,就可以确定该目录遍历结束。

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
# 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
          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

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持服务器之家。

原文链接:https://blog.csdn.net/yjp2006/article/details/5688335

标签:

相关文章

热门资讯

2020微信伤感网名听哭了 让对方看到心疼的伤感网名大全
2020微信伤感网名听哭了 让对方看到心疼的伤感网名大全 2019-12-26
Intellij idea2020永久破解,亲测可用!!!
Intellij idea2020永久破解,亲测可用!!! 2020-07-29
背刺什么意思 网络词语背刺是什么梗
背刺什么意思 网络词语背刺是什么梗 2020-05-22
苹果12mini价格表官网报价 iPhone12mini全版本价格汇总
苹果12mini价格表官网报价 iPhone12mini全版本价格汇总 2020-11-13
歪歪漫画vip账号共享2020_yy漫画免费账号密码共享
歪歪漫画vip账号共享2020_yy漫画免费账号密码共享 2020-04-07
返回顶部