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)