如何在pyspark sql中通过循环日期来拉取数据？

from pyspark.sql import SparkSession
from pyspark.sql.functions import col
from pyspark.sql.types import DateType
from datetime import datetime, timedelta

spark = SparkSession.builder.appName("DateLoop").getOrCreate()

start_date = datetime(2022, 1, 1)
end_date = datetime(2022, 1, 31)

result_df = spark.createDataFrame([], schema)

current_date = start_date
while current_date <= end_date:
    # 将日期转换为字符串格式
    current_date_str = current_date.strftime("%Y-%m-%d")
    
    # 构建查询条件
    condition = col("date_column") == current_date_str
    
    # 执行数据拉取操作
    temp_df = spark.sql("SELECT * FROM table_name WHERE {}".format(condition))
    
    # 将当前日期的数据添加到结果DataFrame中
    result_df = result_df.union(temp_df)
    
    # 增加一天，继续下一次循环
    current_date += timedelta(days=1)

result_df.show()