Python 文件类型计数器

简介

在这个项目中，你将学习如何编写一个Python脚本，用于统计给定目录中具有特定文件类型的文件数量。在Python中处理文件和目录时，这是一项常见任务，并且在各种应用中都很有用，例如数据分析、文件管理和系统管理。

👀 预览

$ python3 file_type.py
请输入文件类型：txt
3

🎯 任务

在这个项目中，你将学习：

• 如何实现 count_file_type 函数来统计具有特定文件类型的文件数量
• 如何处理用户输入并显示文件类型计数
• 如何增强脚本以处理多种文件类型和目录访问错误

🏆 成果

完成这个项目后，你将能够：

• 编写一个Python脚本，用于统计目录中具有特定文件类型的文件数量
• 处理用户输入并显示文件类型计数
• 增强脚本以处理边界情况并提供更多功能

Skills Graph

实现 count_file_type 函数

在这一步中，你将在 file_type.py 文件中实现 count_file_type 函数。该函数将统计给定目录中具有特定文件类型的文件数量。

1. 在文本编辑器中打开 file_type.py 文件。

2. 找到 count_file_type 函数定义。

3. 实现函数逻辑，以统计给定目录中具有指定文件类型的文件数量。

   • 使用 os.listdir() 函数获取目录中所有文件的列表。
   • 遍历文件列表，并使用 endswith() 方法检查每个文件是否以指定的文件类型结尾。
   • 对于每个与指定文件类型匹配的文件，增加 count 变量的值。
   • 返回最终的 count 值。

以下是完整的 count_file_type 函数：

def count_file_type(file_path, file_type):
    """
    统计给定目录中具有特定文件类型的文件数量。

    参数：
    - file_path：一个字符串，指定要统计文件的目录路径。
    - file_type：一个字符串，指定要统计的文件类型（文件扩展名）。

    返回：
    - count：一个整数，表示目录中具有指定文件类型的文件数量。
    """
    count = 0
    for file in os.listdir(file_path):
        if file.endswith("." + file_type):
            count += 1
    return count

运行脚本并获取用户输入

在这一步中，你将运行 file_type.py 脚本，并提示用户输入他们想要统计的文件类型。

1. 打开终端或命令提示符，导航到包含 file_type.py 文件的目录。

2. 使用以下命令运行脚本：

   python3 file_type.py

3. 脚本将提示你输入文件类型：

   请输入文件类型：

4. 输入文件类型（例如，“txt”）并按回车键。

   $ python3 file_type.py
   请输入文件类型：txt
   3

5. 然后，脚本将打印出 /home/labex/project 目录中具有指定文件类型的文件数量。

总结

恭喜你！你已经完成了这个项目。你可以在 LabEx 中练习更多实验来提升你的技能。