python3 文件内容搜索文件名搜索目录遍历

柔光的暖阳◎ 2021-09-07 21:30 487阅读 0赞

**目录**

目录遍历

文件名搜索

文件内容搜索

关键字搜索整合

完整代码

--------------------

## 目录遍历 ##

首先我们来看如何实现目录遍历

目录遍历的过程很简单:

1.  将目标路径作为当前目录
2.  罗列出当前目录下的所有内容
3.  判断每一个目标是否是文件夹。
4.  如果是，进入该文件夹搜索（将文件夹路径改为当前路径，多级目录在此处会循环)

为什么我们要使用递归，与for对已知大小的列表的遍历相比，递归更适合与那些确定开始结束条件而过程不确定的情况。我们将初始路径设为起点，将目录下没有其他目录可以进入的情况设为终点

让我们来看如何通过递归的方式来遍历某个路径下的所有内容

def search(path):
    	content = os.listdir(path)
    	for each in content:
    		each_path = path + os.sep + each
    		if os.path.isdir(each_path):
    			search(each_path)
    		print(each_path)#测试输出

*  第1行：首先我们先定义一个search函数，参数path是接收的路径
 *  第2行：os.listdir(path) 将path路径下的所有(文件/文件夹)作为一个列表返回
 *  第3行：使用for循环遍历当前目录
 *  第4行：构建当前目录下文件的完整目录（当前目录/文件名称），os.sep为当前操作系统的分隔符
 *  第5行：判断这个目标是否是文件夹
 *  第6行：如果是，把这个文件夹作为新的路径像刚才那样再来一遍
 *  第7行：最后在循环的后面我们加一个print来看看输出了什么

来看我们的测试目录test的结构

test
    ├── a
    │   ├── a
    │   ├── b
    │   └── c
    │       └── file.md
    ├── b
    │   ├── a
    │   └── b
    │       └── target.py
    ├── c
    │   └── a
    └── d

测试，进入test目录并使用os.getcwd()返回当前路径：

>>> search(os.getcwd())

输出结果

/home/Jkob/Github/test/b/b/target.py
    /home/Jkob/Github/test/b/b
    /home/Jkob/Github/test/b/a
    /home/Jkob/Github/test/b
    /home/Jkob/Github/test/a/b
    /home/Jkob/Github/test/a/a
    /home/Jkob/Github/test/a/c/file.md
    /home/Jkob/Github/test/a/c
    /home/Jkob/Github/test/a
    /home/Jkob/Github/test/d
    /home/Jkob/Github/test/c/a
    /home/Jkob/Github/test/c

目录的遍历有什么用呢，我们可以添加条件来（搜索/操作）目录下所有符合条件的文件

## 文件名搜索 ##

我们来对刚刚的函数进行简单的修改

def search(path,keyword):
    	content= os.listdir(path)
    	for each in content:
    		each_path = path+os.sep+each
    		if filename in each:
    			print(each_path)
    		if os.path.isdir(each_path):
    			search(each_path,keyword)

与刚刚不同的只有简单几行

*  第1行：函数参数增加了要搜索的文件名
 *  第5,6行：增加了判断，我们这里使用‘in’ 而不是‘==’ 的目的是为了使用模糊匹配，只要包含搜索的关键字就符合条件  
    \-第8行：由于我们整个函数没有修改filename,所以我们可以将搜索条件传递下去

我们这次再来测试，寻找藏起来的target.py

>>>search(os.getcwd(),'target')

结果

/home/Jkob/Github/test/b/b/target.py

再试一次

>>>search(os.getcwd(),'tar')

结果

/home/Jkob/Github/test/b/b/target.py

## 文件内容搜索 ##

然而只有文件名字符合并不能满足我们的需求，我们有时也需要查找文件里面的内容是否符合关键字需求

def content_search(filepath,keyword):
    	f = open(filepath,'r')
    	for line in f:
    		if keyword in line:
    			f.close()
    			return True
    	f.close()
    	return False

在这里我们定义了一个函数content\_search(),搜索文件内容并默认返回False，如果符合关键字条件即返回True并终端该项搜索

别忘记关闭文件哦

接下来我们将这个函数整合到我们刚刚的文件搜索函数中

## 关键字搜索整合 ##

我们来把刚才的内容搜索函数添加到我们刚才的搜索函数中

def search(path,keyword):
    	content= os.listdir(path)
    	for each in content:
    		each_path = path+os.sep+each
    		if keyword in each:
    			print(each_path)
    		if os.path.isdir(each_path):
    			search(each_path,keyword)
    		elif content_search(each_path,keyword):
    			print(each_path)

我们只添加了最后两行，即当该文件包含关键字时，我们显示该文件路径。但是为什么我们不能把这个和刚才文件名判定条件放在一起呢？**因为有的文件名指代的是文件夹**,我们需要将其放在文件夹判断之后

我们来测试一下,其实目录下的file.md文件是编辑过的，内容为’target’:

search(os.getcwd(),'tar')

/home/Jkob/Github/test/b/b/target.py
    /home/Jkob/Github/test/a/c/file.md

如果我们将整个文件作为工具的话，我们可以最后一行添加

search(os.getcwd(),input('Your Keyword:'))

## 完整代码 ##

import os
    def search(path,keyword):
    	content= os.listdir(path)
    	for each in content:
    		each_path = path+os.sep+each
    		if keyword in each:
    			print(each_path)
    		if os.path.isdir(each_path):
    			search(each_path,keyword)
    		elif content_search(each_path,keyword):
    			print(each_path)
    def content_search(filepath,keyword):
    	f = open(filepath,'r')
    	for line in f:
    		if keyword in line:
    			f.close()
    			return True
    	f.close()
    	return False
    search(os.getcwd(),input('Your Keyword:'))

示例:

[Jkob@localhost test] $ python3 test.py 
    Your Keyword:tar
    /home/Jkob/Github/test/b/b/target.py
    /home/Jkob/Github/test/a/c/file.md