根据图片名批量创建文件夹

比眉伴天荒 2023-07-07 04:58 7阅读 0赞

源数据是这样的  
![源数据][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM4Njg5Mzk1_size_16_color_FFFFFF_t_70]  
想要的效果是这样的  
![文件夹][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM4Njg5Mzk1_size_16_color_FFFFFF_t_70 1]  
**注意：你文件后缀名是什么你就在 suffix\_data 这个数组中添加一下**

# -*- coding: UTF-8 -*- 
    import os
    import time
    import shutil
    
    root_path = input('请输入要生成文件的地址：')
    root_path = root_path.strip().rstrip('\\')
    
    def append_data(root_path):
        data = []
        try:
            for filename in os.listdir(root_path):
                if(filename.endswith(suffix)):
                    filename = filename.rstrip(suffix)
                    data.append(filename)
            makedirs(root_path, data)
        except FileNotFoundError:
            print('地址输入错误，请检查地址'.center(20,'-'))
            time.sleep(1.2)
    
    def makedirs(root_path, data):
        for folder in data:
            folder_path = root_path + '\\' + folder
            if not os.path.exists(folder_path):
                os.makedirs(folder_path)
                for suffix_append in suffix_data:
                    filename_all = root_path + '\\' + folder + suffix_append
                    if(os.path.isfile(filename_all)):
                        shutil.move(filename_all,folder_path)
            else:
                for suffix_append in suffix_data:
                    filename_all = root_path + '\\' + folder + suffix_append
                    if(os.path.isfile(filename_all)):
                        shutil.move(filename_all,folder_path)
        print('移动完毕'.center(20,'-'))
        time.sleep(1.2)
    
    if __name__ == "__main__":
        suffix = '.jpg'
        suffix_data = ['.jpg', '.png', '.JPG', '.psd', '.afphoto']
        append_data(root_path)

如果只是想让图片移动到文件夹，可以使用如下代码：

# -*- coding: UTF-8 -*-
    import os
    import shutil
    
    root_path = input('请输入要生成文件的地址：')
    root_path = root_path.strip().rstrip('\\')
    
    def makedirs():
        for dirpath, dirnames, files in os.walk(root_path):
            for file in files:
                folder_path = os.path.join(root_path, file.split('.')[0])
                file_path = os.path.join(dirpath, file)
                if not os.path.exists(folder_path):
                    os.makedirs(folder_path)
                    shutil.move(file_path, folder_path)
        print('移动完毕'.center(20,'-'))
    
    if __name__ == "__main__":
        makedirs()

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM4Njg5Mzk1_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20200220230711935.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM4Njg5Mzk1,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM4Njg5Mzk1_size_16_color_FFFFFF_t_70 1]: https://img-blog.csdnimg.cn/20200311180936778.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM4Njg5Mzk1,size_16,color_FFFFFF,t_70