Fixed sub-categories.

driki · Mar 19, 2012 · 2d66d11 · 2d66d11
1 parent acdd05d
commit 2d66d11
Showing 1 changed file with 17 additions and 3 deletions.
diff --git a/app/models/project.rb b/app/models/project.rb
@@ -135,20 +135,34 @@ def bulk_load
         # look to see if there is a category before creating one
         category = categories.find_by_name(category_name)
         if category.nil?
-          category = categories.build(:name => category_name, 
+          if expense === "true"
+            category = categories.build(:name => category_name, 
                                   :expense_budget => category_amount,
                                   :tag_list => category_tags)
+          else
+            category = categories.build(:name => category_name, 
+                                  :revenue_budget => category_amount,
+                                  :tag_list => category_tags)
+          end
           category.save
           categories << category
         end
 
         # look to see if there is a sub-category before creating one
         sub_category = categories.find_by_name(sub_category_name)
         if sub_category.nil?
-          sub_category = categories.build(:name => sub_category_name, 
-                                      :expense_budget => sub_category_amount,
+          if expense === "true"
+            sub_category = categories.build(:name => sub_category_name, 
+                                  :revenue_budget => sub_category_amount,
+                                  :tag_list => sub_category_tags,
+                                  :parent_id => category.id)
+          else
+            sub_category = categories.build(:name => sub_category_name, 
+                                      :revenue_budget => sub_category_amount,
                                       :tag_list => sub_category_tags,
                                       :parent_id => category.id)
+          end
+
           sub_category.save
           categories << sub_category
         end