日期选择器添加nonSelectableDates属性 #178

Author: agiapp

BRPickerView/DatePickerView/BRDatePickerView.h

@@ -138,6 +138,9 @@ typedef void (^BRDateResultBlock)(NSDate * _Nullable selectDate, NSString * _Nul
 /** 设置时区，默认为当前时区 */
 @property (nullable, nonatomic, copy) NSTimeZone *timeZone;
 
+/** 指定不允许选择的日期 */
+@property (nullable, nonatomic, copy) NSArray <NSDate *> *nonSelectableDates;
+
 /// 初始化时间选择器
 /// @param pickerMode  日期选择器显示类型
 - (instancetype)initWithPickerMode:(BRDatePickerMode)pickerMode;

BRPickerView/DatePickerView/BRDatePickerView.m

@@ -737,6 +737,7 @@ - (NSString *)pickerView:(UIPickerView *)pickerView titleForRow:(NSInteger)row f
 // 4.滚动 pickerView 执行的回调方法
 - (void)pickerView:(UIPickerView *)pickerView didSelectRow:(NSInteger)row inComponent:(NSInteger)component {
     NSString *lastSelectValue = self.mSelectValue;
+    NSDate *lastSelectDate = self.mSelectDate;
     if (self.pickerMode == BRDatePickerModeYMDHMS) {
         if (component == 0) {
             self.yearIndex = row;
@@ -1082,6 +1083,19 @@ - (void)pickerView:(UIPickerView *)pickerView didSelectRow:(NSInteger)row inComp
         }
     }
 
+    // 过滤不可选择日期
+    if (self.nonSelectableDates && self.nonSelectableDates.count > 0 && ![self.mSelectValue isEqualToString:self.addCustomString]) {
+        for (NSDate *date in self.nonSelectableDates) {
+            if ([self br_compareDate:date targetDate:self.mSelectDate dateFormat:self.dateFormatter] == NSOrderedSame) {
+                // 如果当前的日期不可选择，就回滚到上次选择的日期
+                [self scrollToSelectDate:lastSelectDate animated:YES];
+                self.mSelectDate = lastSelectDate;
+                self.mSelectValue = lastSelectValue;
+                break;
+            }
+        }
+    }
+    
     // 由 【自定义字符串】 滚动到 其它时间时，回滚到上次选择的位置
     if ([lastSelectValue isEqualToString:self.addCustomString] && ![self.mSelectValue isEqualToString:self.addCustomString]) {
         [self scrollToSelectDate:self.mSelectDate animated:NO];

BRPickerViewDemo/Classes/Demo/Controller/TestViewController.m

@@ -310,7 +310,9 @@ - (void)handlerTextFieldSelect:(UITextField *)textField {
             datePickerView.maxDate = [NSDate br_setYear:2022 month:10 day:20];
             datePickerView.isAutoSelect = YES;
             datePickerView.addCustomString = @"自定义";
-            //datePickerView.keyView = self.view; // 也可将组件 datePickerView 添加到 self.view 上，默认是添加到 keyWindow 上
+            // 指定不可选择的日期
+            // datePickerView.nonSelectableDates = @[[NSDate br_setYear:2020 month:8 day:25], [NSDate br_setYear:2020 month:8 day:28]];
+            // datePickerView.keyView = self.view; // 将组件 datePickerView 添加到 self.view 上，默认是添加到 keyWindow 上
             datePickerView.resultBlock = ^(NSDate *selectDate, NSString *selectValue) {
                 self.birthdaySelectDate = selectDate;
                 self.infoModel.birthdayStr = selectValue;