From 7c22229c2b5d6b0216a13b88d17e3c16d8af8e85 Mon Sep 17 00:00:00 2001 From: Kevin G Christiano Date: Mon, 20 Apr 2020 14:42:42 -0400 Subject: [PATCH] path based on delegate existence --- .../Atoms/TextFields/DigitEntryField.swift | 16 ++++++---------- 1 file changed, 6 insertions(+), 10 deletions(-) diff --git a/MVMCoreUI/Atomic/Atoms/TextFields/DigitEntryField.swift b/MVMCoreUI/Atomic/Atoms/TextFields/DigitEntryField.swift index 977447ff..01305a95 100644 --- a/MVMCoreUI/Atomic/Atoms/TextFields/DigitEntryField.swift +++ b/MVMCoreUI/Atomic/Atoms/TextFields/DigitEntryField.swift @@ -334,17 +334,13 @@ import UIKit setAsSecureTextEntry(entryType == .secure || entryType == .password) } - for digitBox in digitBoxes { - - if let observingTextFieldelegate = delegateObject?.observingTextFieldDelegate { - digitBox.digitField.inputAccessoryView = UIToolbar.getToolbarWithDoneButton(delegate: observingTextFieldelegate, - action: #selector(observingTextFieldelegate.dismissFieldInput)) - } else { - digitBox.digitField.inputAccessoryView = UIToolbar.getToolbarWithDoneButton(delegate: self, - action: #selector(dismissFieldInput)) - } - } + let observingDelegate = delegateObject?.observingTextFieldDelegate ?? self + digitBoxes.forEach { + $0.digitField.inputAccessoryView = UIToolbar.getToolbarWithDoneButton(delegate: observingDelegate, + action: #selector(observingDelegate.dismissFieldInput)) + } + super.set(with: model, delegateObject, additionalData) }