Merge pull request #569 from balancemymoney/bug/278/institution-cell-…

…background Fix dragging a section (institution) in the accounts view has no background #278
balance-io · Feb 13, 2018 · dddc2e4 · dddc2e4
2 parents 981d624 + db7347e
commit dddc2e4
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/Balance/macOS/View Controllers/Tabs/View Controllers/Table Cells/AccountsTabTableCells.swift b/Balance/macOS/View Controllers/Tabs/View Controllers/Table Cells/AccountsTabTableCells.swift
@@ -98,6 +98,16 @@ class AccountsTabGroupCell: View {
 
         self.setAccessibilityLabel("Section: " + updatedModel.name)
         self.needsDisplay = true
+
+        updateBackgroundColors()
+    }
+
+    func updateBackgroundColors() {
+        if let color = model?.source.color {
+            self.layerBackgroundColor = color
+            amountField.backgroundColor = color
+            nameField.backgroundColor = color
+        }
     }
 }
 
@@ -136,7 +146,6 @@ class AccountsTabAccountCell: View {
 
     init() {
         super.init(frame: NSZeroRect)
-        self.layerBackgroundColor = CurrentTheme.defaults.cell.backgroundColor
 
         self.addSubview(topContainer)
         topContainer.snp.makeConstraints { make in