Python 文件类型计数器：自动化文件管理

介绍

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

👀 预览

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

🎯 任务

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

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

🏆 成果

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

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

实现 count_file_type 函数

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

在文本编辑器中打开 file_type.py 文件。
找到 count_file_type 函数定义。
实现函数逻辑，以统计给定目录中具有指定文件类型的文件数量。
- 使用 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 脚本，并提示用户输入他们想要统计的文件类型。

打开终端或命令提示符，导航到包含 file_type.py 文件的目录。
使用以下命令运行脚本：
```
python3 file_type.py
```
脚本将提示你输入文件类型：
```
请输入文件类型：
```
输入文件类型（例如，“txt”）并按回车键。
```
$ python3 file_type.py
请输入文件类型：txt
3
```
然后，脚本将打印出 /home/labex/project 目录中具有指定文件类型的文件数量。

总结

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