python遍歷文件目錄、批量處理同類文件
本文實(shí)例為大家分享了python遍歷文件目錄、批量處理同類文件的具體代碼,供大家參考,具體內(nèi)容如下
目錄操作
1、獲取當(dāng)前目錄
import os curr_path=os.path.dirname(__file__) #返回當(dāng)前文件所在的目錄,即當(dāng)前運(yùn)行的腳本所在父目錄 print curr_path
運(yùn)行示例
(1)使用os.path.dirname(__file__)時(shí),是針對(duì)運(yùn)行時(shí)對(duì)所給程序腳本的路徑來(lái)獲取父目錄的,即截取你輸入的腳本路徑的所在目錄名稱,如上圖示例,輸入絕對(duì)路徑時(shí)返回絕對(duì)路徑,輸入相對(duì)路徑時(shí)返回相對(duì)路徑,如果只輸入了腳本名稱,則返回空。
(注:當(dāng)從命令行中進(jìn)入python環(huán)境時(shí)時(shí),參數(shù)__file__不能使用)
(2)當(dāng)直接使用os.path.dirname(“/home/test_MK/test.py”)時(shí),直接返回“/home/test_MK”
2、獲取目錄文件列表
file_list=os.listdir("/home/test_MK/test")) print file_list
運(yùn)行示例
3、獲取該目錄下文件夾或者文件列表
path="/home/test_MK/test" objects=os.listdir(path) dir_list=[] #存放目錄列表 file_list=[] #存放文件列表 for obj in objects: if os.path.isdir(os.path.join(path, obj)):#判斷是否是目錄os.path.join()用來(lái)將路徑拼接 dir_list.append(os.path.join(path, obj))#保存時(shí)保存完整路徑才能對(duì)其進(jìn)行后續(xù)操作 print "dir:",obj else: file_list.append(os.path.join(path, obj)) print "file:",obj print "目錄列表:",dir_list print "文件列表:",file_list #如果項(xiàng)判斷是否是文件時(shí)用os.isfile()
(注:使用os.isdir()與os.isfile()時(shí),參數(shù)必須是一個(gè)相對(duì)路徑或者絕對(duì)路徑,不能光是一個(gè)文件名或者目錄名稱,這也是上面示例代碼中使用os.path.join()的原因,否則函數(shù)將判斷不出正確結(jié)果)
運(yùn)行示例
批量處理目錄下同類文件
以處理pcap文件為例
1、獲取某一文件夾下所有pcap包路徑,過(guò)濾掉其它文件
def getPathFile(path): ''' name:getPathFile function:獲取所給文件夾下所有pcap文件路徑 path:所給文件夾路徑 ''' Path = [] try: pathDir = os.listdir(path) for allDir in pathDir: child = os.path.join('%s/%s' % (path, allDir)) #跳過(guò)文件夾以及非流量包文件,將后綴名改為自己需要的文件類型即可實(shí)現(xiàn)自己的過(guò)濾 if os.path.isfile(child) and (".pcap" in str(allDir) or (".cap" in str(allDir))): Path.append(child) except: pass return Path
2、處理函數(shù),打印一個(gè)pcap文件中所有數(shù)據(jù)包的五元組信息{src_ip,src_port,dst_ip,dst_port}
def print_pack_f(file_path): ''' name:print_pack_f function:打印一個(gè)pcap文件中所有數(shù)據(jù)包的五元組信息 file_path:所給pcap文件路徑 ''' file_p= open(file_path) pcap = dpkt.pcap.Reader(file_p) if not pcap: return print "\n\n*******file:%s*******\n"% file_path for (ts,buf) in pcap: try: eth = dpkt.ethernet.Ethernet(buf) #解包,物理層 if not isinstance(eth.data, dpkt.ip.IP): #解包,網(wǎng)絡(luò)層 continue ip = eth.data src_ip="%d.%d.%d.%d"%tuple(map(ord,list(ip.src))) dst_ip="%d.%d.%d.%d"%tuple(map(ord,list(ip.dst))) if (not isinstance(ip.data, dpkt.tcp.TCP)) and (not isinstance(ip.data, dpkt.udp.UDP)): #解包,傳輸層 continue transf= ip.data print "<",src_ip,":",transf.sport,"-->",dst_ip,":",transf.dport,">" except Exception,err: print "[error] %s" % err
3、調(diào)用示例
def main(dir_path): all_file_path=getPathFile(dir_path) #獲取目錄下所有pcap文件路徑 for file in all_file_path: #遍歷處理 print_pack_f(file) #單個(gè)pcap文件處理,可將本函數(shù)替換成自定義的功能,便可實(shí)現(xiàn)批量處理 if __name__ == '__main__': opts,args = getopt.getopt(sys.argv[1:], "hi:") #從命令行獲取參數(shù) if not opts: #若沒(méi)有帶參數(shù) print "\n\ *******************\n\ warn! please enter related parameters,enter -h for help!\n\n\ *******************\n" sys.exit() input_path='' for op, value in opts: if op == "-i": input_path = value elif op == "-h": usage() #幫助信息,只是簡(jiǎn)單的一個(gè)輸出函數(shù),輸出內(nèi)容自定義 sys.exit() main(input_path)
結(jié)果展示
測(cè)試目錄如下
運(yùn)行結(jié)果(python test.py -i ./test)
完畢
以上就是本文的全部?jī)?nèi)容,希望對(duì)大家的學(xué)習(xí)有所幫助,也希望大家多多支持腳本之家。
相關(guān)文章
Python實(shí)現(xiàn)一個(gè)簡(jiǎn)單的驗(yàn)證碼程序
這篇文章主要介紹了Python實(shí)現(xiàn)一個(gè)簡(jiǎn)單的驗(yàn)證碼程序,具有一定參考價(jià)值,需要的朋友可以了解下。2017-11-11Python正則表達(dá)式如何進(jìn)行字符串替換實(shí)例
Python正則表達(dá)式在使用中會(huì)經(jīng)常應(yīng)用到字符串替換的代碼。這篇文章主要介紹了Python正則表達(dá)式如何進(jìn)行字符串替換,具有一定的參考價(jià)值,感興趣的小伙伴們可以參考一下。2016-12-12Python中subprocess的簡(jiǎn)單使用示例
這篇文章主要介紹了Python中subprocess的簡(jiǎn)單使用示例,是Python進(jìn)程方面處理的相關(guān)重要知識(shí),需要的朋友可以參考下2015-07-07Python unittest單元測(cè)試框架總結(jié)
這篇文章主要介紹了Python unittest單元測(cè)試框架總結(jié),小編覺(jué)得挺不錯(cuò)的,現(xiàn)在分享給大家,也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧2018-09-09Python讀取Excel一列并計(jì)算所有對(duì)象出現(xiàn)次數(shù)的方法
這篇文章主要給大家介紹了關(guān)于Python讀取Excel一列并計(jì)算所有對(duì)象出現(xiàn)次數(shù)的相關(guān)資料,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2020-09-09用Python3創(chuàng)建httpServer的簡(jiǎn)單方法
今天小編就為大家分享一篇用Python3創(chuàng)建httpServer的簡(jiǎn)單方法,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-06-06