Python删除某一目录下指定类型的文件

适合场景：在做自动化前，需要对环境进行初始化。如，上一次自动化产生的截图文件需要删除，以避免对本次自动化产生影响。
值得注意的是，删除文件时，不能放在setUp里面做初始化。否则在本次自动化执行下一个.py时会再初始化一次，执行删除上一个.py产生的新的文件，导致截图文件丢失。思路是：放在当次自动化第一个.py执行用例前做初始化。

import os

def preDelPic(path):#如果需要删除更多目录下的文件，可以传两个元组进来，一个元组存放路径(字符串类型)，一个元组存放指定删除的文件格式(字符串类型)。
  for root, dirs, files in os.walk(path):#os.walk()返回元组，包含三个元素:每次遍历的路径名、路径下子目录列表、目录下文件列表
    for name in files:
      if name.endswith(".png"): #指定要删除的文件格式，这里是png，可以换成其他格式
        os.remove(os.path.join(root, name))
        print ("Delete File: " + os.path.join(root, name))
        
if __name__ == "__main__":
  func_path = os.path.dirname(__file__)  # 获取当前脚本所在目录的绝对路径
  base_dir = os.path.dirname(func_path) 
  base_dir = str(base_dir)  # 以字符串方式来处理
  base_dir = base_dir.replace('\\', '/')  
  base = base_dir.split("testCase")[0] 
  filepath = base + "/test_report/screenshot/"
  preDelPic(filepath)