python 用代码实现文件归类处理

python 用代码实现文件归类处理

直接上代码

import os, shutil

def main():
# 1.获取文件夹，并获取文件夹里面的内容
file_path = input(r’请输入要整理的文件夹的绝对路径：’)
os.chdir(file_path)
file_names = os.listdir()

# 2.设置文件夹里面的文件格式
formats = {
    '音乐': ['.mp3', '.m4a'],
    '视频': ['.mp4', '.avi', '.mkv'],
    '图片': ['.jpg', '.png', ',jepg', '.gif'],
    'word文档': ['.txt', '.doc', '.docx'],
    'ppt文档': ['.ppt'],
    'pdf文档': ['.pdf'],
    'excel文档': ['.xlsx'],
    '程序': ['.exe','.msi'],
    '压缩': ['.zip', '.rar'],
    '脚本': ['.bat', '.vba']
}
# 3.获取文件的后缀
for file_name in file_names:
    houzhui = os.path.splitext(file_name)[-1]

    # 4.自定义文件的后缀
    for file_type, file_suffix in formats.items():
        if not os.path.isdir(file_type):
            os.mkdir(file_type)
        # 文件的移动
        if houzhui in file_suffix:
            shutil.move(file_name, f'{file_type}/{file_name}')
print('****文件整理完成****')

if name == ‘main’:
main()

祝大家python学习顺利！