复制和删除一个文件夹下,符合指定条件的文件,例如把selenium的所有测试用例提取出来到一个文件夹。
import os
class CasesSuit: def add_cases(self, case_dir): ''' 加载 ../test_cases 文件夹下测试用例到 ../data/tmp ''' tmp_save_case_folder = '../data/tmp' if not os.path.exists(tmp_case_folder): os.mkdir(tmp_case_folder) list_dir = os.listdir(case_dir) for i in list_dir: case_file = os.path.join(case_dir, i) if os.path.isfile(case_file):
##筛选符合条件的测试用例文件名 if 'test' in os.path.split(case_file)[1] and '.py' in os.path.split(case_file)[1]: shutil.copy(case_file,tmp_case_folder) if os.path.isdir(case_file): self.add_cases(case_file) def del_cases(self, tmp_case_dir): ''' 删除 ../data/tmp 临时文件夹下所有测试用例 ''' filedir = os.listdir(tmp_case_dir) for filename in filedir: file = os.path.join(tmp_case_dir, filename) if os.path.isdir(file): self.del_cases(file) else: os.remove(file)