python3怎么自定義函數讀取目錄下指定后綴文件？

import os

def test(text):

path = "d:\\data"

dirs = os.listdir(path)

for i in dirs:

if os.path.splitext(i)[1] == '.txt':

# 處理文件的代碼

return text