查看原文
其他

Python 三种遍历目录的方法,轻松帮你找出隐藏文件

极客猴 极客猴 2022-09-11

题图:by Instagram from Instagram

阅读文本大概需要 6 分钟。

无论在 Windows 系统中还是 Linux 系统中,都存在着隐藏文件以及隐藏文件夹。隐藏文件夹一般是系统关键性目录,例如 Windows 系统中的 C 盘中的 Boot 文件夹Windows 文件夹等。在 Linux 系统中,一些系统配置文件或软件配置文件会被隐藏起来,如:系统环境变量配置文件 .profile。有些时候,我们必须通过文件路径来遍历整个目录,然后找到隐藏文件。本文的主要内容是给大家分享 Python 三种遍历文件的方法。

01简单暴力法-递归

假设在 E 盘中,有个名为“Python”的文件夹;该文件夹中也有两个文件夹,分别是“A”和“B”;另外,在“A”文件夹中还有一个 “results.txt” 的文本文件。因此,“Python”文件夹的文件结构如下:

Python
|--A
|  |--results.txt
|--B

我们可以从上述看出,一个文件夹其实是一个树型的数据结构。遍历树的最简单、最暴力的办法就是递归。因此,遍历“Python”的文件夹的代码可以这么写。

# -*- coding: UTF-8 -*-
import os

# 递归遍历目录
def traversal_files(path):
    for dir in os.listdir(path):
        dir = os.path.join(path, dir)
        print(dir)
        # 判断当前目录是否为文件夹
        if os.path.isdir(dir):
            traversal_files(dir)

if __name__ == '__main__':
    path = '.'
    traversal_files(path)

运行脚本程序后,发现“Python”中的文件和文件夹都被打印出来。

这种遍历方法能否找出隐藏文件?答案是肯定能。让我们来验证一番。首先,我们在“Python”的文件夹中放入一个“config.txt”的隐藏文件。

然后再运行程序,结果发现“config.txt”被打印出来。

这种办法虽然写起来代码简洁,但是在文件夹的子目录层级过深的情况下,效率会比较低。

02优雅的 os.walk()

既然递归太暴力,那么使用 os.walk() 会让程序显得优雅。os.walk() 方法是一个简单易用的文件、目录遍历器,可以帮助我们高效的处理文件、目录方面的事情。这个方法适用于在 Linix 和 Windows。

os.walk() 一般用法是传入两个参数。第一个参数是 path, 即所要遍历的目录的地址。它返回的是一个三元组(root, dirs, files)。

  • root 所指的是当前正在遍历的这个文件夹的本身的地址

  • dirs 是一个 list ,内容是该文件夹中所有的目录的名字(不包括子目录)

  • files 同样是 list , 内容是该文件夹中所有的文件(不包括子目录)

第二个参数是topdown,它是一个可选参数。当它的值为 True 时,则优先遍历 path 目录,否则优先遍历 top 的子目录(默认为开启)。

因此,使用 os.walk 遍历文件夹,找出隐藏文件的代码如下所示。

# -*- coding: UTF-8 -*-
import os

def traversal_files(path):
    for root, dirs, files in os.walk(path, topdown=False):
        for name in files:
            print(os.path.join(root, name))
        for name in dirs:
            print(os.path.join(root, name))

if __name__ == '__main__':
    path = '.'
    traversal_files(path)


03高效的 os.scandir()

在 Python 3.5版本中,新添加了 os.scandir()方法,它是一个目录迭代方法。os.scandir() 的运行效率要比 os.walk 高。在 PEP 471 中,Python 官方也推荐我们使用 os.scandir() 来遍历目录。

按照前面的例子,遍历“Python”的文件夹中的隐藏文件的代码如下:

# -*- coding: UTF-8 -*-
import os

dirs = []
files = []

def traversal_files(path):
    for item in os.scandir(path):
        if item.is_dir():
            dirs.append(item.path)
        elif item.is_file():
            files.append(item.path)

    print('dirs:', dirs)
    print('files:', files)

if __name__ == '__main__':
    path = '.'
    traversal_files(path)

运行结果,同样也是能找出隐藏文件。


如果你觉得文章还不错,请大家点赞分享下。你的肯定是我最大的鼓励和支持。


推荐阅读:

Python 居然有这么奇葩的库--The Fuck

详解 Scrapy 中间键的用法

10张图带你了解后台服务架构演变

高并发的那些事



不积跬步,无以至千里



您可能也对以下帖子感兴趣

文章有问题?点此查看未经处理的缓存