今天用Python提取了Linux內核源代碼的目錄樹結構,沒有怎麼寫過腳本程式,我居然折騰了2個小時,先是如何枚舉出給定目錄下的所有文件和文件夾,os.walk可以實現列舉,但是os.walk是只給出目錄名和文件名,而沒有絕對路徑。使用os.path.listdir可以達到這個目的,然後是創建目錄, ...
今天用Python提取了Linux內核源代碼的目錄樹結構,沒有怎麼寫過腳本程式,我居然折騰了2個小時,先是如何枚舉出給定目錄下的所有文件和文件夾,os.walk可以實現列舉,但是os.walk是只給出目錄名和文件名,而沒有絕對路徑。使用os.path.listdir可以達到這個目的,然後是創建目錄,由於當目錄存在是會提示創建失敗的錯誤,所以我先想刪除所有目錄,然後再創建,但是發現還是有問題,最好還是使用判斷如果不存在才創建目錄,存在時就不創建,貼下代碼:
# @This script can be used to iterate the given directory,and create the # empty directory structure without file in it,e.g,I want to have you directory # as the linux kernel source, but i don't want the files, then this script comes. # @This script is running under python 3.1 # @author:zhangchao # @Time:2011年7月25日18:43:26 ########################################################################### import os import re #listmydirs is created to recursivly list all the entrys in the specified path. #In fact, we have os.walk to handle this problem # #level:目錄的層數,不要也可以,主要是為了顯示目錄在那一層 #srcpath:內核源代碼所在的路路徑 #destpath:將要生成的內核源代碼的目錄結構所在路徑 # def createkerneldirs(level,srcpath,destpath): for entrys in os.listdir(srcpath): #學習listdir函數的用法 tmpsrcpath=srcpath+os.sep+entrys tmpdestpath = tmpsrcpath.replace(srcpath,destpath)#將源路徑中的E:\linux-2.6替換為E:\tmp,學習字元串替換函數的用法 print('in level:'+str(level)) print(tmpsrcpath) print(tmpdestpath) if os.path.isdir(tmpsrcpath): listmydirs(level+1,tmpsrcpath,tmpdestpath) if os.path.exists(tmpdestpath)==False: #如果文件不存在才創建文件 os.makedirs(tmpdestpath) if __name__=='__main__': #將E:\linux-2.6的內核源代碼目錄結構拷貝到E:\tmp目錄下 createkerneldirs(1,r'E:\linux-2.6',r'E:\tmp')