From cc3b3bfed70e71bec6da67f2cb768598f247fae7 Mon Sep 17 00:00:00 2001 From: nikAshanin Date: Mon, 10 Apr 2017 18:21:39 +0300 Subject: [PATCH] guard tableview added in tabledirector extension --- .../TableDirector/TableDirector+Extensions.swift | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/LeadKit/LeadKit/Extensions/TableDirector/TableDirector+Extensions.swift b/LeadKit/LeadKit/Extensions/TableDirector/TableDirector+Extensions.swift index c108c618..bc00d2aa 100644 --- a/LeadKit/LeadKit/Extensions/TableDirector/TableDirector+Extensions.swift +++ b/LeadKit/LeadKit/Extensions/TableDirector/TableDirector+Extensions.swift @@ -56,10 +56,14 @@ public extension TableDirector { @discardableResult public func reload(sectionAtIndex index: Int, with animation: UITableViewRowAnimation = .none) -> Self { let action = { [tableView] in - if index < tableView?.numberOfSections ?? 0 { - tableView?.reloadSections([index], with: animation) + guard let tableView = tableView else { + return + } + + if index < tableView.numberOfSections { + tableView.reloadSections([index], with: animation) } else { - tableView?.reloadData() + tableView.reloadData() } } if animation == .none {