Picker working and saving the selected category

hardyrz · Jun 30, 2017 · 7a74d75 · 7a74d75
1 parent 894863d
commit 7a74d75
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/iExpense/Base.lproj/Main.storyboard b/iExpense/Base.lproj/Main.storyboard
@@ -166,7 +166,7 @@
                                                     <constraint firstAttribute="width" constant="60" id="nXY-sE-srY"/>
                                                 </constraints>
                                             </imageView>
-                                            <textField opaque="NO" clipsSubviews="YES" contentMode="scaleToFill" contentHorizontalAlignment="left" contentVerticalAlignment="center" borderStyle="roundedRect" textAlignment="natural" minimumFontSize="17" translatesAutoresizingMaskIntoConstraints="NO" id="VOl-kA-Y2Q">
+                                            <textField opaque="NO" clipsSubviews="YES" contentMode="scaleToFill" contentHorizontalAlignment="left" contentVerticalAlignment="center" borderStyle="roundedRect" placeholder="Category" textAlignment="natural" minimumFontSize="17" translatesAutoresizingMaskIntoConstraints="NO" id="VOl-kA-Y2Q">
                                                 <rect key="frame" x="60" y="0.0" width="283" height="60"/>
                                                 <nil key="textColor"/>
                                                 <fontDescription key="fontDescription" type="system" pointSize="14"/>

diff --git a/iExpense/NewExpenseViewController.swift b/iExpense/NewExpenseViewController.swift
@@ -16,6 +16,7 @@ class NewExpenseViewController: UIViewController, UIPickerViewDelegate, UIPicker
     var realm : Realm!
     var categoriesList : Results<CategoryDTO>!
     var categoryPicker = UIPickerView()
+    var selectedCategory : CategoryDTO?
 
     override func viewDidLoad() {
         super.viewDidLoad()
@@ -42,10 +43,12 @@ class NewExpenseViewController: UIViewController, UIPickerViewDelegate, UIPicker
     }
 
     @IBAction func saveAction(_ sender: Any) {
-        if (Int(self.valueInput.text!) != nil) {
+        if (Int(self.valueInput.text!) != nil && self.selectedCategory != nil) {
             let myExpense = ExpenseDTO()
             myExpense.value = Int(self.valueInput.text!)!
             myExpense.desc = self.descriptionInput.text!
+            myExpense.category = self.selectedCategory
+
             try! realm.write {
                 realm.add(myExpense)
             }
@@ -69,6 +72,7 @@ class NewExpenseViewController: UIViewController, UIPickerViewDelegate, UIPicker
 
     func pickerView(_ pickerView: UIPickerView, didSelectRow row: Int, inComponent component: Int) {
         self.categoryInput.text = categoriesList.elements[row].name
+        self.selectedCategory = categoriesList.elements[row]
         self.view.endEditing(false)
     }
 }