如何将工作时间添加到日期时间

import datetime

current_datetime = datetime.datetime.now()
start_time = datetime.time(9, 0)  # 工作开始时间为上午9点
end_time = datetime.time(17, 0)  # 工作结束时间为下午5点

work_start_datetime = datetime.datetime.combine(current_datetime.date(), start_time)
work_end_datetime = datetime.datetime.combine(current_datetime.date(), end_time)

if current_datetime < work_start_datetime:
    # 当前时间在工作开始时间之前
    result_datetime = work_start_datetime
elif current_datetime > work_end_datetime:
    # 当前时间在工作结束时间之后，将工作时间添加到第二天的开始时间
    result_datetime = datetime.datetime.combine(current_datetime.date() + datetime.timedelta(days=1), start_time)
else:
    # 当前时间在工作时间范围内，将工作时间添加到当前日期时间
    result_datetime = current_datetime + datetime.timedelta(hours=(end_time.hour - current_datetime.hour))

print(result_datetime)