Add by_nutrient_number scopes

mattbeedle · Aug 1, 2015 · 4462d66 · 4462d66
1 parent bba2b63
commit 4462d66
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/lib/usda_nutrient_database/food.rb b/lib/usda_nutrient_database/food.rb
@@ -21,5 +21,10 @@ class Food < ActiveRecord::Base
 
     belongs_to :food_group, class_name: 'UsdaNutrientDatabase::FoodGroup',
       foreign_key: :food_group_code
+
+    scope :by_nutrient_number, ->(nutrient_number) {
+      joins(:nutrients)
+        .where(nutrients: { nutrient_number: nutrient_number })
+    }
   end
 end
diff --git a/lib/usda_nutrient_database/foods_nutrient.rb b/lib/usda_nutrient_database/foods_nutrient.rb
@@ -18,5 +18,8 @@ class FoodsNutrient < ActiveRecord::Base
     scope :by_nutrient_databank_number, ->(nutrient_databank_number) {
       where(nutrient_databank_number: nutrient_databank_number)
     }
+    scope :by_nutrient_number, ->(nutrient_number) {
+      where(nutrient_number: nutrient_number)
+    }
   end
 end
diff --git a/lib/usda_nutrient_database/footnote.rb b/lib/usda_nutrient_database/footnote.rb
@@ -12,5 +12,9 @@ class Footnote < ActiveRecord::Base
       foreign_key: :nutrient_databank_number
     belongs_to :nutrient, class_name: 'UsdaNutrientDatabase::Nutrient',
       foreign_key: :nutrient_number
+
+    scope :by_nutrient_number, ->(nutrient_number) {
+      where(nutrient_number: nutrient_number)
+    }
   end
 end