将工作日添加到UIDatePicker

import UIKit

class ViewController: UIViewController {
    @IBOutlet weak var datePicker: UIDatePicker!
    
    var workdays = [Date]()
    
    override func viewDidLoad() {
        super.viewDidLoad()
        
        // 设置日期模式为日期和时间
        datePicker.datePickerMode = .dateAndTime
        
        // 获取当前日期
        let currentDate = Date()
        
        // 创建一个日历对象
        let calendar = Calendar.current
        
        // 获取一周内的所有日期
        if let weekDates = calendar.dateInterval(of: .weekOfYear, for: currentDate) {
            let startDate = weekDates.start
            let endDate = weekDates.end
            
            // 遍历日期范围内的每一天
            calendar.enumerateDates(startingAfter: startDate, matching: DateComponents(hour: 0), matchingPolicy: .nextTime) { (date, _, stop) in
                guard let date = date else { return }
                
                // 判断日期是否为工作日（非周末）
                if !calendar.isDateInWeekend(date) {
                    self.workdays.append(date)
                }
                
                // 判断是否已经遍历到范围内的最后一天
                if date >= endDate {
                    stop = true
                }
            }
        }
        
        // 将工作日选项数组作为UIDatePicker的数据源
        datePicker.dataSource = self
        datePicker.delegate = self
        datePicker.reloadAllComponents()
    }
}

extension ViewController: UIPickerViewDataSource, UIPickerViewDelegate {
    func numberOfComponents(in pickerView: UIPickerView) -> Int {
        return 1
    }
    
    func pickerView(_ pickerView: UIPickerView, numberOfRowsInComponent component: Int) -> Int {
        return workdays.count
    }
    
    func pickerView(_ pickerView: UIPickerView, titleForRow row: Int, forComponent component: Int) -> String? {
        let dateFormatter = DateFormatter()
        dateFormatter.dateFormat = "yyyy-MM-dd HH:mm"
        return dateFormatter.string(from: workdays[row])
    }
}