使用python根据后缀名整理文件夹

405 阅读2分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第16天,点击查看活动详情

需求来了:

平常用浏览器在互联网下载的文件,一般都在一个“下载”文件夹内,里面的文件什么格式的都有,看着就很乱;所以看能不能给整理一下,这个活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

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

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

总结:

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