平时工作没有养成分类的习惯,整个桌面杂乱无章都是文档和资料,几乎快占满整个屏幕了。所以必须要整理一下了,今天我们来看下用python如何批量将不同后缀的文件移动到同一文件夹。
演示效果:
使用前
使用后
代码:
# # -*- coding:utf-8 -*-import osimport globimport shutilimport tkinterimport tkinter.filedialogimport filecmpfrom datetime import datetimedef start():"""tkinter弹出窗口获取文件路径path。"""root = tkinter.Tk()root.withdraw()dirname = tkinter.filedialog.askdirectory(parent=root, initialdir="/", title='请选择文件夹')return dirnamedef check_repeat_file(path):"""去掉重复文件"""dir_path = pathfile_lst = []file_count = 0for i in glob.glob(dir_path + '/**/*', recursive=True):if os.path.isfile(i):file_lst.append(i)file_count += 1print("\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 namereturn "未知分类"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: # 文件大小为0print(file+'是空文件,即将执行删除操作!')os.remove(os.fspath(path+'/'+file)) # 删除这个文件returnexcept 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 osimport shutilimport tkinterimport tkinter.filedialogfrom datetime import datetimedef 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)