python遍历文件目录、批量处理同类文件
作者:MK_夕阳 时间:2021-10-19 14:58:12
本文实例为大家分享了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)
完毕
来源:https://blog.csdn.net/qq_33515733/article/details/88887978
标签:python,遍历文件,批量处理
![](/images/zang.png)
![](/images/jiucuo.png)
猜你喜欢
在ASP中使用SQL语句之5:开始执行
2007-08-11 12:36:00
PHP中Http协议post请求参数
2023-11-16 18:38:38
asp.net中文件下载功能的实例代码
2023-07-20 15:46:51
python数据处理之如何修改索引和行列
2023-11-04 13:57:27
![](https://img.aspxhome.com/file/2023/7/64647_0s.png)
Oracle SQL性能优化系列学习三
2010-07-23 13:08:00
如何正确处理ajax 302跳转问题回博客首页
2009-02-28 14:01:00
python自动化测试用例全对偶组合与全覆盖组合比较
2021-02-03 01:59:45
![](https://img.aspxhome.com/file/2023/1/78481_0s.png)
网页设计十大诀窍
2007-10-19 13:03:00
em与px的区别以及em特点和应用
2008-11-11 12:03:00
CSS Sprites (CSS图像拼合技术)教程工具
2009-05-26 15:30:00
![](https://img.aspxhome.com/file/UploadPic/20095/26/1742060-99s.gif)
python导入模块交叉引用的方法
2021-11-01 07:03:42
低效的键盘和高效的登录框
2007-08-22 09:17:00
![](https://img.aspxhome.com/file/UploadPic/20078/22/200782291926838s.jpg)
在Python的Django框架中实现Hacker News的一些功能
2023-11-24 19:27:46
基于php常用函数总结(数组,字符串,时间,文件操作)
2023-11-24 00:23:46
利用XMLBean轻轻松松读写XML
2008-09-04 11:25:00
CSS网页布局编码小技巧整理
2009-12-30 16:50:00
网页特效文字之—银箔字
2013-08-07 00:21:39
![](https://img.aspxhome.com/file/UploadPic/20072/20072311613309s.jpg)
PHP实现网页内容html标签补全和过滤的方法小结【2种方法】
2023-09-06 22:28:26
python实现代理服务功能实例
2023-10-04 05:36:57
![](https://img.aspxhome.com/file/2023/6/62936_0s.jpg)
从SNS看社会化界面设计(一)
2009-02-23 12:21:00
![](https://img.aspxhome.com/file/UploadPic/20092/23/1-73s.jpg)