opencv-uninstaller.zip

import os import shutil class OpencvUninstallManager(object): def __init__(self): self.uninstall_dir = '/usr/local' self.store_dir = os.path.join(os.path.expanduser('~'), 'opencv_uninstall') def get_all_subdirs(self, dir): ''' 获取一个文件夹下所有文件夹绝对路径，包括子目录 :param dir: :return: ''' dir_result = [] for root, dirs, files in os.walk(dir): for dir in dirs: dir_result.append(os.path.join(root, dir)) return dir_result def get_all_files_subdir2(self, dir, filter_list=[]): ''' 获取文件夹下所有的文件，默认是所有类型文件，如果只想获取txt,filter_list=[".txt"],一定要加点 :param dir: :param filter_list: :return: ''' file_list = [] if len(filter_list) == 0: for root, dirs, files in os.walk(dir): for file in files: file_list.append(os.path.join(root, file)) return file_list else: for root, dirs, files in os.walk(dir): for file in files: suffix = os.path.splitext(file)[-1] if suffix in filter_list: file_list.append(os.path.join(root, file)) return file_list def start_uninstall(self): if not os.path.exists(self.store_dir): os.makedirs(self.store_dir) print('start uninstall directories') dirs = self.get_all_subdirs(self.uninstall_dir) for dir in dirs: if dir.lower().find('opencv') > 0: new_dir = dir.replace(self.uninstall_dir, self.store_dir) if not os.path.exists(new_dir): shutil.move(dir, new_dir) print('start uninstall opencv files') files = self.get_all_files_subdir2(self.uninstall_dir) for file in files: if file.lower().find('opencv') > 0: new_file = file.replace(self.uninstall_dir, self.store_dir) if not os.path.exists(new_file): temp_dir = os.path.dirname(new_file) if not os.path.exists(temp_dir): os.makedirs(temp_dir) shutil.move(file, new_file) def start_resume(self): print('start resume all files') files = self.get_all_files_subdir2(self.store_dir) for file in files: new_file = file.replace(self.store_dir, self.uninstall_dir) if not os.path.exists(new_file): temp_dir = os.path.dirname(new_file) if not os.path.exists(temp_dir): os.makedirs(temp_dir) shutil.copyfile(file, new_file) if __name__ == '__main__': mm = OpencvUninstallManager() mm.start_uninstall()