使用Spark，如何在将所有内容加载到数据帧中时拾取文件名？

from pyspark.sql import SparkSession
from pyspark.sql.functions import input_file_name

spark = SparkSession.builder.getOrCreate()

df = spark.read.text("文件路径")

df_with_filename = df.withColumn("文件名", input_file_name())