不断否定自己，但要坚持最初的意愿。
小白已经实现了"将光盘内容索引存储为硬盘上的*.cdc文本文件"，并获得了命令行工具样的程序，可以通过命令行调用python pycdc.pyw -e test.txt快速指定文件名。
类似grep一样，现在要实现搜索的功能，打开所有符合要求的文件，读取每一行，如果有指定关键词在行内就打印输出到屏幕……
结合已有的经验，可以非常简单地实现!
# -*- coding: utf-8 -*-
import os
def cdcGrep(cdcpath, keyword):
filelist = os.listdir(cdcpath) # 搜索目录中的文件
for cdc in filelist: # 循环文件列表
if ".cdc" in cdc: # 过滤器其他文件，只关注.cdc
print('找到目标文件:{}'.format(cdc))
cdcfile = open(cdcpath + cdc) # 拼接文件路径，并打开文件
for line in cdcfile.readlines(): # 读取文件每一行，并循环
if keyword in line: # 判断是否有关键词在行中
print(line)
cdcGrep('F:\\back\\', 'images')
以上代码可以扫描出指定目录"f:\back\"下的名称中含有".cdc"中带有关键词的"images"行。
运行结果如下：
找到目标文件:test1.cdc E:\iso\CentOS-6.8-x86_64-bin-DVD1\images;['pxeboot'];['efiboot.img', 'efidisk.img', 'install.img', 'TRANS.TBL'] E:\iso\CentOS-6.8-x86_64-bin-DVD1\images\pxeboot;[];['initrd.img', 'TRANS.TBL', 'vmlinuz'] 找到目标文件:test2.cdc E:\iso\CentOS-6.8-x86_64-bin-DVD1;['EFI', 'images', 'isolinux', '[BOOT]'];['.discinfo', '.treeinfo', 'CentOS_BuildTag', 'EULA', 'GPL', 'RELEASE-NOTES-en-US.html', 'RPM-GPG-KEY-CentOS-6', 'RPM-GPG-KEY-CentOS-Debug-6', 'RPM-GPG-KEY-CentOS-Security-6', 'RPM-GPG-KEY-CentOS-Testing-6', 'TRANS.TBL'] E:\iso\CentOS-6.8-x86_64-bin-DVD1\images;['pxeboot'];['efiboot.img', 'efidisk.img', 'install.img', 'TRANS.TBL']
在上面的grep 实现例子中，没有考虑子目录的处理方式，因为如果直接open目录进行读操作，会出现错误。接下来将改进这段代码，以便考虑到子目录这种特殊情况。
# -*- coding: utf-8 -*-
import os
def cdcGrep(cdcpath, keyword):
expDict = {}
filelist = os.listdir(cdcpath) # 搜索目录中的文件
cdcpath = cdcpath + "\\"
for cdc in filelist: # 循环文件列表
if os.path.isdir(cdcpath+cdc):
print(cdcpath+cdc)
cdcGrep(cdcpath+cdc, keyword) # 若是子目录，则递归调用完成查找
else:
if cdc.endswith('.cdc'):
print(cdc)
cdcfile = open(cdcpath + cdc) # 拼合文件路径，并打开文件
for line in cdcfile.readlines(): # 读取文件每一行，并循坏
if keyword in line: # 判断是否有关键词在行中
print(line)
cdcGrep('F:\\back\\', 'images')
运行结果如下：
搜索子目录F:\back\\1 找到目标文件:test1.cdc E:\iso\CentOS-6.8-x86_64-bin-DVD1\images;['pxeboot'];['efiboot.img', 'efidisk.img', 'install.img', 'TRANS.TBL'] E:\iso\CentOS-6.8-x86_64-bin-DVD1\images\pxeboot;[];['initrd.img', 'TRANS.TBL', 'vmlinuz'] 搜索子目录F:\back\\2 找到目标文件:test2.cdc E:\iso\CentOS-6.8-x86_64-bin-DVD1;['EFI', 'images', 'isolinux', '[BOOT]'];['.discinfo', '.treeinfo', 'CentOS_BuildTag', 'EULA', 'GPL', 'RELEASE-NOTES-en-US.html', 'RPM-GPG-KEY-CentOS-6', 'RPM-GPG-KEY-CentOS-Debug-6', 'RPM-GPG-KEY-CentOS-Security-6', 'RPM-GPG-KEY-CentOS-Testing-6', 'TRANS.TBL'] E:\iso\CentOS-6.8-x86_64-bin-DVD1\images\pxeboot;[];['initrd.img', 'TRANS.TBL', 'vmlinuz'] 找到目标文件:test3.cdc E:\iso\CentOS-6.8-x86_64-bin-DVD1;['EFI', 'images', 'isolinux', '[BOOT]'];['.discinfo', '.treeinfo', 'CentOS_BuildTag', 'EULA', 'GPL', 'RELEASE-NOTES-en-US.html', 'RPM-GPG-KEY-CentOS-6', 'RPM-GPG-KEY-CentOS-Debug-6', 'RPM-GPG-KEY-CentOS-Security-6', 'RPM-GPG-KEY-CentOS-Testing-6', 'TRANS.TBL'] E:\iso\CentOS-6.8-x86_64-bin-DVD1\images;['pxeboot'];['efiboot.img', 'efidisk.img', 'install.img', 'TRANS.TBL']
这边对子目录处理，使用了递归调用cdcGrep()来实现。
总结：本节实现了搜索的功能，学习了函数的递归调用。
原文地址：http://blog.51cto.com/9473774/2089630
时间： 2024-10-11 06:01:46