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.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')

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://www.iplaypy.com/code/base/b2548.html

相关文章 Recommend