获取文件夹下的文件数据.txt

import os

path = str(input('请输入地址：'))
data = set([i for i in os.listdir(path)])
current_directory = os.path.dirname(os.path.abspath(__file__)) + '\diff.txt'
file_write = open(current_directory, "w")
for i in data:
    text = i + '\n'
    file_write.write(text)