python知识点100篇系列（5） -根据后缀名整理文件夹

需求来了：

平常用浏览器在互联网下载的文件，一般都在一个“下载”文件夹内，里面的文件什么格式的都有，看着就很乱；所以看能不能给整理一下，这个活python可以干；

解决方案：

思路一、根据文件后缀名，归类文件，相同后缀名，放到同一个文件夹下；

主要用到os模块，shutil模块；
os用了获取文件路径信息
shutil用来移动文件；

主要步骤是这样的：

• 给定一个文件夹
• 获取文件夹下所有文件，和获取文件的后缀名
• 根据后缀名创建文件夹（如果没有）
• 将文件移动到后缀名的文件夹内；

代码如下：

# 引入模块
import os
import shutil
def extFile():
    # 定义一个文件夹路径
    src_dir = "你的文件夹路径"
    for d_file in os.listdir(src_dir):
        #获取文件后缀名称
        ext = os.path.splitext(d_file)[1]
        ext = ext[1:]
        # 文件夹不存在就创建
        if not os.path.isdir(f"{src_dir}/{ext}"):
            os.mkdir(f"{src_dir}/{ext}")
        # 拼接文件路径
        source_path = f"{src_dir}/{d_file}"
        target_path = f"{src_dir}/{ext}/{d_file}"
        # 移动文件
        shutil.move(source_path, target_path)

    print("整理完成！\n")

思路二、根据后缀名，指定分类，多个不同后缀名的文件可以放到一个文件夹下；

用到的python模块还是只有

os模块和shutil模块

主要步骤有所不同：

• 首先定义一个后缀名的分类

 {
        "图片": [".jpeg", ".jpg", ".tiff", ".gif", ".bmp", ".png", ".psd"],
        "视频": [".avi", ".flv", ".wmv", ".mov", ".mp4",".mkv"],
        "文档": [".oxps", ".epub", ".pages", ".docx", ".doc",".xls", ".xlsx", ".ppt",
                    "pptx",".csv",".pdf"],
        "压缩文件": [".a", ".ar", ".cpio", ".iso", ".tar", ".gz", ".rz", ".7z",
                    ".dmg", ".rar", ".xar", ".zip"]
    }

• 然后扫描文件夹，获取文件后缀

 src_dir = "你的文件夹路径"
    for entry in os.scandir(src_dir):
        if entry.is_dir():
        # 如果是文件夹则跳过
            continue
    file_path = f"{src_dir}/{entry.name}"
    file_format = file_path.suffix.lower()

• 根据文件后缀名获取分类名称

dir_name = '未知分类'
    for name ,type_list in DIRECTORIES.items():
        if file_format in type_list:
            dir_name = name

-- 最后根据分类名称创建文件夹（如果没有）

• 将文件移动到分类名称的文件夹内；

总结：

两种思路都可行，第一种更简单一点