《可爱的Python》读书笔记（四）

小白已经实现了"将光盘内容索引存储为硬盘上的*.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()来实现。

总结：本节实现了搜索的功能，学习了函数的递归调用。

《可爱的Python》读书笔记（四）

原文：http://blog.51cto.com/9473774/2089630