import json import os import shutil from log import logger, log_info from data_dict import ext_name_list, know_zip_ext_name, know_ext_name, handle_zip_ext_name, clear_list # 定义通用方法 # 目标目录， 获取此目录下的全部文件，未经整理 def get_all_files(_path): file_list = [] for item in os.walk(_path): if len(item[2]) > 0: for file_name in item[2]: file_list.append(os.path.join(item[0], file_name)) return file_list # 目标文件， 重命名文件 def file_rename(_org, _target): os.rename(_org, _target) # 用于整理解压文件的对象 class FilesUnzip: def __init__(self, root_path): self.root_path = root_path # 目标根目录，返回根目录下全部文件夹的列表 def get_root_folder_list(self): return sorted(list(os.walk(self.root_path))[0][1]) # 目标文件， 获取文件扩展名， 返回 文件路径+文件名， 扩展名 @staticmethod def get_ext_name(_file): base_name = os.path.splitext(_file)[0] ext_name = os.path.splitext(_file)[1].lower() return base_name, ext_name # 目标文件， 将文件分类整理返回字典格式文件列表 def get_cate_files(self, _path): new_file_list = dict(handle_zip=[], zip=[], others=[], unknown=[]) for _f in get_all_files(_path): new = self.change_to_know_name(_f) _, ext_new = self.get_ext_name(new) if ext_new in handle_zip_ext_name: if self.del_small_zip_file(new): continue new_file_list['handle_zip'].append(new) new_file_list['zip'].append(new) elif ext_new in know_zip_ext_name: new_file_list['zip'].append(new) elif ext_new in know_ext_name: new_file_list['others'].append(new) else: new_file_list['unknown'].append(new) return new_file_list # 目标文件， 将错误的扩展名换成正确的，执行重命名操作，返回新的文件名列表 def change_to_know_name(self, _file): base_file_name, ext_file_name = self.get_ext_name(_file) if ext_file_name in know_zip_ext_name: return _file for unknown_ext, know_ext in ext_name_list.items(): if ext_file_name == unknown_ext: new_file_name = base_file_name + know_ext log_info(f'修改未知扩展名： {_file} -> {new_file_name}') file_rename(_file, new_file_name) return new_file_name return _file # 删除文件 @staticmethod def del_all_files(path_list): for path in path_list: result = os.system(f'del "{path}"') log_info(f"删除文件{'成功' if result == '0' else '失败'}： {path}") # 删除小文件 @staticmethod def del_small_zip_file(_path): if os.path.getsize(_path) < 1024000: log_info(f"无效文件小于1MB，将被删除 {_path}") result = os.system(f'del "{_path}"') log_info(f"删除文件{'成功' if result == '0' else '失败'}： {_path}") return True return False # 获取可删除文件列表 # def get_del_files(self, _path): # del_list = [] # for _f in self.get_all_files(_path): # if os.path.basename(_f) in clear_list['file_name']: # del_list.append(_f) # if os.path.splitext(_f)[1] in clear_list['ext_name']: # del_list.append(_f) # return del_list # 根据clear_list清除无效文件 # 用于整理文件夹的对象 class FilesCollection: def __init__(self, path): self.path = path # 清理无效文件 def clear_files(self): for file in get_all_files(self.path): for file_name in clear_list['file_name']: if os.path.basename(file).lower() == file_name.lower(): os.system(f'del "{file}"') # logger.info(f"删除文件成功： {path}") log_info(f"删除文件成功： {file}") break for ext_name in clear_list['ext_name']: if os.path.splitext(file)[1].lower() == ext_name.lower(): os.system(f'del "{file}"') # logger.info(f"删除文件成功： {path}") log_info(f"删除文件成功： {file}") # 用来整理文件夹，返回原路径和去掉无效后的路径 def get_move_files(self): todos = [[self.path, self.path]] move_files = [] tree = list(os.walk(self.path)) # print(json.dumps(list(tree))) # print(todos) while len(todos) > 0: # print('todos') # print(todos) temp = [] for todo in todos: for leaf in tree: if todo[0] == leaf[0]: # print('todo[0] - leaf[0]') # print(todo[0]) # print(leaf[0]) if len(leaf[1]) == 0: # 文件夹为0个， 文件有0个或多个 # print(1) for file in leaf[2]: move_files.append([os.path.join(leaf[0], file), os.path.join(todo[1], file)]) elif len(leaf[1]) == 1 and len(leaf[2]) == 0: # 1个文件夹 和 0个文件 # print(2) temp.append([os.path.join(leaf[0], leaf[1][0]), todo[1]]) elif len(leaf[1]) == 1 and len(leaf[2]) > 0: # 1个文件夹 和 多个文件 # print(3) temp.append([os.path.join(leaf[0], leaf[1][0]), os.path.join(todo[1], leaf[1][0])]) for file in leaf[2]: move_files.append([os.path.join(leaf[0], file), os.path.join(todo[1], file)]) elif len(leaf[1]) > 1: # 有多个文件夹 # print(4) for folder in leaf[1]: temp.append([os.path.join(leaf[0], folder), os.path.join(todo[1], folder)]) for file in leaf[2]: move_files.append([os.path.join(leaf[0], file), os.path.join(todo[1], file)]) else: print('看看啥情况') # print('temp') # print(temp) todos = temp return move_files # 移动文件 def move_files(self): for move in self.get_move_files(): # print(move) if not os.path.exists(os.path.dirname(move[1])): log_info(f'创建文件夹： {os.path.dirname(move[1])}') os.makedirs(os.path.dirname(move[1])) log_info(f'移动文件：{move[0]} -> {move[1]}') shutil.move(move[0], move[1]) # 获取一个空文件夹, 如果是空文件夹则返回路径，否则返回False @staticmethod def is_empty(_path): tree = list(os.walk(_path)) empty_list = [] for leaf in tree: if not leaf[1] and not leaf[2]: return leaf[0] return False # 清除全部空文件夹 def remove_empty(self): while _empty := self.is_empty(self.path): # print(_empty) log_info(f'移除空文件夹：{_empty}') os.system(f"attrib -r {_empty}") os.removedirs(_empty) # 如果根目录是空的话就移除根目录并返回True def remove_empty_root_folder(self): if _empty := self.is_empty(self.path): log_info(f'移除空的根目录：{_empty}') os.system(f"attrib -r {_empty}") os.removedirs(_empty) return True return False # 重命名根路径的文件夹， 如果只有一个文件就把文件重命名 def rename_root_folder(self, _root, name, _org, _target): if len(file := get_all_files(_org)) == 1: ext = os.path.splitext(file[0])[1] log_info(f'重命名：{file[0]} -> {os.path.join(_root, name + ext)}') file_rename(file[0], os.path.join(_root, name + ext)) self.remove_empty() else: log_info(f'重命名：{_org} -> {_target}') file_rename(_org, _target) if __name__ == '__main__': root = r"F:\Temp\sjry\hj" files = FilesUnzip(root) # print(files.get_root_folder_list()) # print(files.get_all_files(r"F:\Temp\sjry\hj\35316")) # files.file_rename('F:\\Temp\\sjry\\hj\\35316\\22\\新建 文本文档.123', 'F:\\Temp\\sjry\\hj\\35316\\22\\新建 文本文档.222') # print(files.get_ext_name('F:\\Temp\\sjry\\hj\\35316\\22\\新建 文本文档.222')) # print(files.change_to_know_name('F:\\Temp\\sjry\\hj\\35316\\22\\新建 文本文档.win')) # for f in files.get_root_folder_list(): # all_file = files.get_all_files(os.path.join(root, f)) # print(files.clear_files(all_file)) # print(files.collection_files(r"F:\Temp\sjry\hj\35316")) # print(json.dumps(files.get_folder_dict(r'F:\Temp\test\12345'))) col_obj = FilesCollection(r'F:\Temp\leshe\leshe_20230810\35480') print(col_obj.remove_empty_root_folder())