本文實例為大家分享了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__)時，是針對運(yùn)行時對所給程序腳本的路徑來獲取父目錄的，即截取你輸入的腳本路徑的所在目錄名稱，如上圖示例，輸入絕對路徑時返回絕對路徑，輸入相對路徑時返回相對路徑，如果只輸入了腳本名稱，則返回空。
（ 注： 當(dāng)從命令行中進(jìn)入python環(huán)境時時，參數(shù)__file__不能使用）
（2）當(dāng)直接使用os.path.dirname(“/home/test_MK/test.py”)時，直接返回“/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()用來將路徑拼接
dir_list.append(os.path.join(path, obj))#保存時保存完整路徑才能對其進(jìn)行后續(xù)操作
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()時，參數(shù)必須是一個相對路徑或者絕對路徑，不能光是一個文件名或者目錄名稱，這也是上面示例代碼中使用os.path.join()的原因，否則函數(shù)將判斷不出正確結(jié)果）
運(yùn)行示例
批量處理目錄下同類文件
以處理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))
#跳過文件夾以及非流量包文件，將后綴名改為自己需要的文件類型即可實現(xiàn)自己的過濾
if os.path.isfile(child) and (".pcap" in str(allDir) or (".cap" in str(allDir))):
Path.append(child)
except:
pass
return Path
2、處理函數(shù)，打印一個pcap文件中所有數(shù)據(jù)包的五元組信息{src_ip，src_port,dst_ip,dst_port}
def print_pack_f(file_path):
'''
name:print_pack_f
function:打印一個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) #單個pcap文件處理，可將本函數(shù)替換成自定義的功能，便可實現(xiàn)批量處理
if __name__ == '__main__':
opts,args = getopt.getopt(sys.argv[1:], "hi:") #從命令行獲取參數(shù)
if not opts: #若沒有帶參數(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() #幫助信息，只是簡單的一個輸出函數(shù)，輸出內(nèi)容自定義
sys.exit()
main(input_path)
結(jié)果展示
測試目錄如下
運(yùn)行結(jié)果（python test.py -i ./test）
完畢
以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。
更多文章、技術(shù)交流、商務(wù)合作、聯(lián)系博主
微信掃碼或搜索：z360901061
微信掃一掃加我為好友
QQ號聯(lián)系： 360901061
您的支持是博主寫作最大的動力，如果您喜歡我的文章，感覺我的文章對您有幫助，請用微信掃描下面二維碼支持博主2元、5元、10元、20元等您想捐的金額吧，狠狠點擊下面給點支持吧，站長非常感激您！手機(jī)微信長按不能支付解決辦法：請將微信支付二維碼保存到相冊，切換到微信，然后點擊微信右上角掃一掃功能，選擇支付二維碼完成支付。
【本文對您有幫助就好】元
