python自动整理文件，就这么简单

平时工作没有养成分类的习惯，整个桌面杂乱无章都是文档和资料，几乎快占满整个屏幕了。所以必须要整理一下了，今天我们来看下用python如何批量将不同后缀的文件移动到同一文件夹。

演示效果：

• 使用前
  

• 使用后

代码：

# # -*- coding:utf-8 -*-
import os
import glob
import shutil
import tkinter
import tkinter.filedialog
import filecmp
from datetime import datetime



def start():
    """
    tkinter弹出窗口获取文件路径path。
    """
    root = tkinter.Tk()
    root.withdraw()
    dirname = tkinter.filedialog.askdirectory(
        parent=root, initialdir="/", title='请选择文件夹')
    return dirname
def check_repeat_file(path):
    """
    去掉重复文件
    """
    dir_path = path
    file_lst = []
    file_count = 0
    for i in glob.glob(dir_path + '/**/*', recursive=True):
        if os.path.isfile(i):
            file_lst.append(i)
            file_count += 1
    print("\n检查文件计数：%d" % file_count, end='')

    for x in file_lst:
        for y in file_lst:
            if x != y and os.path.exists(x) and os.path.exists(y):
                if filecmp.cmp(x, y):
                    os.remove(y)
                    print('\n移除重复文件：%d'%y)





def JudgeFile(suffix):
    """
    定义一个函数，传入每个文件对应的后缀。判断文件是否存在于字典file_dict中；
    如果存在，返回对应的文件夹名；如果不存在，将该文件夹命名为"未知分类"；
    定义一个文件字典，不同的文件类型，属于不同的文件夹
    """
    file_dict = {
        "图片": ["jpeg", "jpg", "tiff", "gif", "bmp", "png", "bpg", "svg", "heif", "psd"],
        "视频": ["avi", "flv", "wmv", "mov", "mp4", "webm", "vob", "mng", "qt", "mpg", "mpeg", "3gp", "mkv"],
        "音频": ["aac", "aa", "aac", "dvf", "m4a", "m4b", "m4p", "mp3", "msv", "ogg", "oga", "raw", "vox", "wav", "wma"],
        "文档": ["oxps", "epub", "pages", "docx", "doc", "fdf", "ods", "odt", "pwi", "xsn", "xps", "dotx", "docm", "dox",
               "rvg", "rtf", "rtfd", "wpd", "xls", "xlsx", "xlsm", "ppt", "pptx", "csv", "pdf", "md", "xmind"],
        "压缩文件": ["a", "ar", "cpio", "iso", "tar", "gz", "rz", "7z", "dmg", "rar", "xar", "zip"],
        "文本": ["txt", "in", "out", "json", "xml", "log"],
        "程序脚本": ["py", "html5", "html", "htm", "xhtml", "cpp", "java", "css", "sql"],
        '可执行程序': ['exe', 'bat', 'lnk', 'sys', 'com', 'apk'],
        '字体文件': ['eot', 'otf', 'fon', 'font', 'ttf', 'ttc', 'woff', 'woff2', 'shx'],
        '工程图文件': ['bak', 'dwg', 'dxf', 'dwl', 'dwl2', 'stp', 'SLDPRT', 'ipj', 'ipt', 'idw']
    }
    for name, type_list in file_dict.items():
        if suffix.lower() in type_list:
            return name
    return "未知分类"



def deldir(path):
    """
    清理空文件夹和空文件
    path: 文件路径，检查此文件路径下的子文件
    """
    try:
        files = os.listdir(path)  # 获取路径下的子文件(夹)列表
        for file in files:
#             print('遍历路径：'+os.fspath(path + '/' + file))
            if os.path.isdir(os.fspath(path+'/'+file)):  # 如果是文件夹
                if not os.listdir(os.fspath(path+'/'+file)):  # 如果子文件为空
                    print(file+'是空文件夹,即将执行删除操作')
                    os.rmdir(os.fspath(path+'/'+file))  # 删除这个空文件夹
                else:
#                     print(file+'不是空文件夹')
                    deldir(os.fspath(path+'/'+file))  # 遍历子文件
                    if not os.listdir(os.fspath(path+'/'+file)):  # 如果子文件为空
                        print(file+'是空文件夹,即将执行删除操作')
                        os.rmdir(os.fspath(path+'/'+file))  # 删除这个空文件夹
            elif os.path.isfile(os.fspath(path+'/'+file)):  # 如果是文件
                if os.path.getsize(os.fspath(path+'/'+file)) == 0:  # 文件大小为0
                    print(file+'是空文件，即将执行删除操作！')
                    os.remove(os.fspath(path+'/'+file))  # 删除这个文件
        return
    except FileNotFoundError:
        print("\n文件夹路径错误")


def main():
    """
    主函数，根据文件后缀整理文件
    path: 文件路径，检查此文件路径下的子文件
    """
    try:
        path = start()
        print("路径是:\n ", path)
        check_repeat_file(path)
        if path == "":
            print("没有选择路径！")
        # 递归获取 "待处理文件路径" 下的所有文件和文件夹。
     
        for file in glob.glob(f"{path}/**/*", recursive=True):
            # 由于我们是对文件分类，这里需要挑选出文件来。
            if os.path.isfile(file):
                # 由于isfile()函数，获取的是每个文件的全路径。这里再调用basename()函数，直接获取文件名；
                file_name = os.path.basename(file)
                suffix = file_name.split(".")[-1]
                # 判断 "文件名" 是否在字典中。
                name = JudgeFile(suffix)
                # 根据每个文件分类，创建各自对应的文件夹。
                if not os.path.exists(f"{path}\\{name}"):
                    os.mkdir(f"{path}\\{name}")
                    print('path-->', name)
                # 将文件复制到各自对应的文件夹中。
                # shutil.copy(file, f"{path}\\{name}")
                # 将文件移动到各自对应的文件夹中。
                shutil.move(file, f"{path}\\{name}")
        deldir(path)
        print("\n已经整理完成。")
    except BaseException:
        print('\n出现未知错误！')
        

if __name__ == '__main__':
    main()

执行起来很简单，只要写完程序，点击程运行，等待弹出窗口，选择需要整理的文件夹即可。

如果觉得以上代码觉得复杂，可以尝试以下更为简单的程序。

# 如何实现文件自动分类？
# 同一目录下存在很多不同类型的资源条件
# 1.分类
# 2.创建分类目录
# 3.移动文件资源

import os
import shutil
import tkinter
import tkinter.filedialog
from datetime import datetime

def start():
    root = tkinter.Tk()
    root.withdraw()
    dirname = tkinter.filedialog.askdirectory(parent=root,initialdir="/",title='请选择文件夹')
    return dirname

# 源文件存在路径
src_dir=start()
# 分类资源存在路径
dest_dir=src_dir
# 判断目录是否存在
if not os.path.exists(dest_dir):
    os.mkdir(dest_dir)
# 源目录分析
files=os.listdir(src_dir)
for item in files:
    src_path=os.path.join(src_dir,item)
    # 判断状态
    if os.path.isfile(src_path):
        #如果是文件，进入代码块
        # 判断文件资源的类型
        ndir = item.split('.')[-1]
        desc_path=os.path.join(dest_dir,ndir)
        # 创建分类目录
        if not os.path.exists(desc_path):
            # 如果分类子目录不存在，创建
            os.mkdir(desc_path)
        shutil.move(src_path,desc_path)