Added custom input view attribute.

GE-N · Jan 15, 2016 · e49b12f · e49b12f
1 parent 99404ae
commit e49b12f
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/MHTextField.h b/MHTextField.h
@@ -30,10 +30,11 @@ typedef void(^controlBarChangedTextFieldBlock)(id, ControlBarAction);
 @property (nonatomic, strong) NSDate *maximumDate;
 @property (nonatomic, strong) NSDate *minimumDate;
 
-@property (nonatomic, setter = setEmailField:) BOOL isEmailField;
-@property (nonatomic, setter = setPickerField:) BOOL isPickerField;
-@property (nonatomic, setter = setDateField:) BOOL isDateField;
-@property (nonatomic, setter = setTimeField:) BOOL isTimeField;
+@property (nonatomic, setter = setEmailField:)      BOOL isEmailField;
+@property (nonatomic, setter = setPickerField:)     BOOL isPickerField;
+@property (nonatomic, setter = setDateField:)       BOOL isDateField;
+@property (nonatomic, setter = setTimeField:)       BOOL isTimeField;
+@property (nonatomic, setter = setCustomInputView:) UIView *customInputView;
 @property (nonatomic, readonly) BOOL isValid;
 
 @property (nonatomic, strong) NSArray *pickerDatas;

diff --git a/MHTextField.m b/MHTextField.m
@@ -187,7 +187,10 @@ - (void)setBarButtonNeedsDisplayAtTag:(NSInteger)tag{
 }
 
 - (void) selectInputView:(UITextField *)textField{
-    if (_isDateField || _isTimeField){
+    if (self.customInputView != nil) {
+        [textField setInputView:self.customInputView];
+    }
+    else if (_isDateField || _isTimeField){
         UIDatePicker *datePicker = [[UIDatePicker alloc] init];
         if (_isDateField)
             datePicker.datePickerMode = UIDatePickerModeDate;