Python提取Linux内核源代码的目录结构
今天用Python提取了Linux内核源代码的目录树结构,没有怎么写过脚本程序,我居然折腾了2个小时,先是如何枚举出给定目录下的所有文件和文件夹,os.walk可以实现列举,但是os.walk是只给出目录名和文件名,而没有绝对路径。使用os.path.listdir可以达到这个目的,然后是创建目录,由于当目录存在是会提示创建失败的错误,所以我先想删除所有目录,然后再创建,但是发现还是有问题,最好还是使用判断如果不存在才创建目录,存在时就不创建,贴下代码:
1 # @This script can be used to iterate the given directory,and create the
2 # empty directory structure without file in it,e.g,I want to have you directory
3 # as the linux kernel source, but i don't want the files, then this script comes.
4 # @This script is running under python 3.1
5 # @author:zhangchao
6 # @Time:2011年7月25日18:43:26
7 ###########################################################################
8
9
10 import os
11 import re
12
13 #listmydirs is created to recursivly list all the entrys in the specified path.
14 #In fact, we have os.walk to handle this problem
15
16 #
17 #level:目录的层数,不要也可以,主要是为了显示目录在那一层
18 #srcpath:内核源代码所在的路路径
19 #destpath:将要生成的内核源代码的目录结构所在路径
20 #
21
22 def createkerneldirs(level,srcpath,destpath):
23 for entrys in os.listdir(srcpath): #学习listdir函数的用法
24 tmpsrcpath=srcpath+os.sep+entrys
25 tmpdestpath = tmpsrcpath.replace(srcpath,destpath)#将源路径中的E:\linux-2.6替换为E:\tmp,学习字符串替换函数的用法
26
27 print('in level:'+str(level))
28 print(tmpsrcpath)
29 print(tmpdestpath)
30
31 if os.path.isdir(tmpsrcpath):
32 listmydirs(level+1,tmpsrcpath,tmpdestpath)
33 if os.path.exists(tmpdestpath)==False: #如果文件不存在才创建文件
34 os.makedirs(tmpdestpath)
35
36 if __name__=='__main__':
37 #将E:\linux-2.6的内核源代码目录结构拷贝到E:\tmp目录下
38 createkerneldirs(1,r'E:\linux-2.6',r'E:\tmp')