python如何遍歷文件目錄以及批量處理同類文件

這篇文章主要為大家展示了“python如何遍歷文件目錄以及批量處理同類文件”，內容簡而易懂，條理清晰，希望能夠幫助大家解決疑惑，下面讓小編帶領大家一起研究并學習一下“python如何遍歷文件目錄以及批量處理同類文件”這篇文章吧。

具體內容如下

目錄操作

1、獲取當前目錄

import os
curr_path=os.path.dirname(__file__) #返回當前文件所在的目錄,即當前運行的腳本所在父目錄
print curr_path

運行示例

（1）使用os.path.dirname(__file__)時，是針對運行時對所給程序腳本的路徑來獲取父目錄的，即截取你輸入的腳本路徑的所在目錄名稱，如上圖示例，輸入絕對路徑時返回絕對路徑，輸入相對路徑時返回相對路徑，如果只輸入了腳本名稱，則返回空。

（注：當從命令行中進入python環境時時，參數__file__不能使用）

（2）當直接使用os.path.dirname(“/home/test_MK/test.py”)時，直接返回“/home/test_MK”

2、獲取目錄文件列表

file_list=os.listdir("/home/test_MK/test")) 
print file_list

運行示例

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()用來將路徑拼接
 dir_list.append(os.path.join(path, obj))#保存時保存完整路徑才能對其進行后續操作
 print "dir:",obj
 else:
 file_list.append(os.path.join(path, obj))
 print "file:",obj
print "目錄列表：",dir_list
print "文件列表：",file_list
#如果項判斷是否是文件時用os.isfile()

（注：使用os.isdir()與os.isfile()時，參數必須是一個相對路徑或者絕對路徑，不能光是一個文件名或者目錄名稱，這也是上面示例代碼中使用os.path.join()的原因，否則函數將判斷不出正確結果）

運行示例

批量處理目錄下同類文件

以處理pcap文件為例

1、獲取某一文件夾下所有pcap包路徑，過濾掉其它文件

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))
  #跳過文件夾以及非流量包文件，將后綴名改為自己需要的文件類型即可實現自己的過濾
  if os.path.isfile(child) and (".pcap" in str(allDir) or (".cap" in str(allDir))): 
  Path.append(child)
 except:
 pass
 return Path

2、處理函數，打印一個pcap文件中所有數據包的五元組信息{src_ip，src_port,dst_ip,dst_port}

def print_pack_f(file_path):
 '''
 name:print_pack_f
 function:打印一個pcap文件中所有數據包的五元組信息
 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): #解包，網絡層
 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、調用示例

def main(dir_path):
 all_file_path=getPathFile(dir_path) #獲取目錄下所有pcap文件路徑
 for file in all_file_path: #遍歷處理
 print_pack_f(file) #單個pcap文件處理，可將本函數替換成自定義的功能，便可實現批量處理

if __name__ == '__main__':
 opts,args = getopt.getopt(sys.argv[1:], "hi:") #從命令行獲取參數
 if not opts: #若沒有帶參數
 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() #幫助信息，只是簡單的一個輸出函數，輸出內容自定義
 sys.exit()
 main(input_path)

結果展示

測試目錄如下

運行結果（python test.py -i ./test）

以上是“python如何遍歷文件目錄以及批量處理同類文件”這篇文章的所有內容，感謝各位的閱讀！相信大家都有了一定的了解，希望分享的內容對大家有所幫助，如果還想學習更多知識，歡迎關注億速云行業資訊頻道！