active_nutrition 0.5.0

data/lib/active-nutrition.rb

@@ -0,0 +1 @@
+require File.join(File.dirname(__FILE__), 'active_nutrition')

data/lib/active_nutrition.rb

@@ -0,0 +1,99 @@
+# encoding: UTF-8
+
+$:.push(File.dirname(__FILE__))
+$:.push(File.dirname(File.dirname(__FILE__)))
+
+$KCODE = 'UTF-8' unless RUBY_VERSION >= '1.9.0'
+
+require 'open-uri'
+require 'fileutils'
+require 'yaml'
+
+require 'rake'
+require 'json'
+require 'zip/zip'
+require 'active_record'
+require 'composite_primary_keys'
+
+require 'active_nutrition/models'
+require 'active_nutrition/objects'
+require 'active_nutrition/utilities'
+require 'active_nutrition/migrations/00000000001_migrations_table'
+
+require 'active_nutrition/railtie' if defined?(Rails)
+
+include ActiveNutrition::Models
+include ActiveNutrition::Objects
+include ActiveNutrition::Utilities
+
+module ActiveNutrition
+  DEFAULT_SEARCH_LIMIT = 10
+
+  class << self
+    def migrate
+      current_migrations = Migration.find(:all).map(&:sequence_no)
+    rescue ActiveRecord::StatementInvalid => e
+      ActiveNutrition::Migrations::MigrationsTable.up
+      Migration.create(:sequence_no => 1)
+      current_migrations = Migration.find(:all).map(&:sequence_no)
+    ensure
+      next_migrations = migrations.reject { |file| (current_migrations || []).include?(File.basename(file)[0..11].to_i) }.sort
+      next_migrations.each do |file|
+        require file
+        migration_class = ActiveNutrition::Migrations.const_get(File.basename(file).split("_")[1..-1].map(&:capitalize).join("").chomp(".rb"))
+        migration_class.up
+        Migration.create(:sequence_no => File.basename(file)[0..11].to_i)
+      end
+    end
+
+    def update
+      raise "Not yet supported."
+    end
+
+    def rebuild
+      upd = Update.new(:type => :full)
+      puts "Downloading data..."
+      upd.download
+      puts "Extracting..."
+      upd.unzip
+      puts "Clearing tables..."
+      upd.reset_db
+      execute_update(upd)
+      puts "Done."
+    end
+
+    def search(terms = "", options = {})
+      options[:conditions] ||= ["Long_Desc LIKE ? OR Long_Desc LIKE ?", "%#{terms.gsub(" ", "%")}%", "%#{terms.split(" ").reverse.join("%")}%"]
+      options[:limit] ||= DEFAULT_SEARCH_LIMIT
+      Food.wrap(FoodDes.find(:all, options))
+    end
+
+    def get(ndb_no)
+      Food.wrap(FoodDes.find(ndb_no))
+    end
+
+    def convert(value, options = {})
+      UnitConverter.convert(value, options)
+    end
+
+    def supported_conversion?(options)
+      UnitConverter.supported_conversion?(options)
+    end
+
+    protected
+
+    def migrations
+      @migrations ||= Dir.glob(File.join(File.dirname(__FILE__), "active_nutrition/migrations/**/**.rb"))
+    end
+
+    def execute_update(updater)
+      last_model = nil
+      updater.execute do |model, record_count, record_total|
+        puts "" unless model == last_model
+        print "\rProcessing #{model}, #{record_count} / #{record_total} (#{((record_count.to_f / record_total.to_f) * 100).round}%) records imported"
+        STDOUT.flush
+        last_model = model
+      end
+    end
+  end
+end

data/lib/active_nutrition/migrations/00000000001_migrations_table.rb

@@ -0,0 +1,18 @@
+# encoding: UTF-8
+
+module ActiveNutrition
+  module Migrations
+    class MigrationsTable < ActiveRecord::Migration
+      def self.up
+        create_table :active_nutrition_migrations do |t|
+          t.integer :sequence_no
+          t.timestamps
+        end
+      end
+
+      def self.down
+        drop_table :active_nutrition_migrations
+      end
+    end
+  end
+end

data/lib/active_nutrition/migrations/00000000002_sr_24.rb

@@ -0,0 +1,207 @@
+# encoding: UTF-8
+
+module ActiveNutrition
+  module Migrations
+    class Sr24 < ActiveRecord::Migration
+      def self.up
+        create_table "abbrev", :primary_key => "NDB_No", :force => true do |t|
+          t.string  "Shrt_Desc",                  :limit => 60
+          t.float   "Water"
+          t.integer "Energ_Kcal"
+          t.float   "Protein"
+          t.float   "Lipid_Tot"
+          t.float   "Ash"
+          t.float   "Carbohydrt"
+          t.float   "Fiber_TD"
+          t.float   "Sugar_Tot",                  :default => 0.0
+          t.integer "Calcium"
+          t.float   "Iron"
+          t.float   "Magnesium",                  :default => 0.0
+          t.integer "Phosphorus"
+          t.integer "Potassium"
+          t.integer "Sodium"
+          t.float   "Zinc",                       :default => 0.0
+          t.float   "Copper",                     :default => 0.0
+          t.float   "Manganese",                  :default => 0.0
+          t.float   "Selenium",                   :default => 0.0
+          t.float   "Vit_C"
+          t.float   "Thiamin",                    :default => 0.0
+          t.float   "Riboflavin",                 :default => 0.0
+          t.float   "Niacin",                     :default => 0.0
+          t.float   "Panto_Acid",                 :default => 0.0
+          t.float   "Vit_B6",                     :default => 0.0
+          t.float   "Folate_Tot",                 :default => 0.0
+          t.float   "Folic_Acid",                 :default => 0.0
+          t.float   "Food_Folate",                :default => 0.0
+          t.float   "Folate_DFE",                 :default => 0.0
+          t.float   "Choline_Tot",                :default => 0.0
+          t.float   "Vit_B12",                    :default => 0.0
+          t.integer "Vit_A_IU"
+          t.float   "Vit_A_RAE",                  :default => 0.0
+          t.float   "Retinol",                    :default => 0.0
+          t.float   "Alpha_Carot",                :default => 0.0
+          t.float   "Beta_Carot",                 :default => 0.0
+          t.float   "Beta_Crypt",                 :default => 0.0
+          t.float   "Lycopene",                   :default => 0.0
+          t.float   "Lut+Zea",                    :default => 0.0
+          t.float   "Vit_E",                      :default => 0.0
+          t.float   "Vit_D_mcg",                  :default => 0.0
+          t.float   "ViVit_D_IU",                 :default => 0.0
+          t.float   "Vit_K",                      :default => 0.0
+          t.float   "FA_Sat"
+          t.float   "FA_Mono",                    :default => 0.0
+          t.float   "FA_Poly",                    :default => 0.0
+          t.integer "Cholestrl"
+          t.float   "GmWt_1"
+          t.string  "GmWt_Desc1",                 :limit => 120
+          t.float   "GmWt_2",                     :default => 0.0
+          t.string  "GmWt_Desc2",                 :limit => 120
+          t.integer "Refuse_Pct"
+        end
+
+        add_index "abbrev", ["Folic_Acid"], :name => "Abbrev_Folic_Acid_Index"
+        add_index "abbrev", ["Panto_Acid"], :name => "Abbrev_Panto_Acid_Index"
+
+        create_table "data_src", :id => false, :primary_key => "DataSrc_ID", :force => true do |t|
+          t.integer "DataSrc_ID"
+          t.string "Authors"
+          t.string "Title"
+          t.string "Year",        :limit => 4
+          t.string "Journal",     :limit => 135
+          t.string "Vol_city",    :limit => 16
+          t.string "Issue_State", :limit => 5
+          t.string "Start_Page",  :limit => 5
+          t.string "End_Page",    :limit => 5
+        end
+
+        add_index "data_src", ["DataSrc_ID"], :name => "DataSrc_ID_Index"
+
+        create_table "datsrcln", :id => false, :primary_key => "NDB_No", :force => true do |t|
+          t.integer "NDB_No",                  :null => false
+          t.integer "Nutr_No",                 :null => false
+          t.string  "DataSrc_ID", :limit => 6, :null => false
+        end
+
+        add_index "datsrcln", ["NDB_No", "Nutr_No", "DataSrc_ID"], :name => "Datsrcln_NDB_No_Nutr_No_DataSrc_ID_Index", :unique => true
+
+        #create_table "deriv_cd", :primary_key => "Deriv_CD", :force => true do |t|
+        create_table "deriv_cd", :id => false, :primary_key => "Deriv_CD", :force => true do |t|
+          t.integer "Deriv_CD"
+          t.string "Deriv_Desc", :limit => 120
+        end
+
+        add_index "deriv_cd", ["Deriv_CD"], :name => "Deriv_CD_Deriv_CD_Index"
+
+        create_table "fd_group", :primary_key => "FdGrp_CD", :force => true do |t|
+          t.string "FdGrp_Desc", :limit => 60
+        end
+
+        create_table "food_des", :primary_key => "NDB_No", :force => true do |t|
+          t.string  "FdGrp_Cd",    :limit => 4
+          t.string  "Long_Desc",   :limit => 200
+          t.string  "Shrt_Desc",   :limit => 60
+          t.string  "ComName",     :limit => 100
+          t.string  "ManufacName", :limit => 65
+          t.string  "Survey",      :limit => 1
+          t.string  "Ref_Desc",    :limit => 135
+          t.integer "Refuse"
+          t.string  "SciName",     :limit => 65
+          t.float   "N_Factor"
+          t.float   "Pro_Factor"
+          t.float   "Fat_Factor"
+          t.float   "CHO_Factor"
+        end
+
+        create_table "footnote", :id => false, :primary_key => "Footnt_No", :force => true do |t|
+          t.integer "NDB_No",      :null => false
+          t.integer "Footnt_No"
+          t.string  "Footnot_Typ", :limit => 1
+          t.integer "Nutr_No"
+          t.string  "Footnot_Txt", :limit => 200
+        end
+
+        create_table "langdesc", :id => false, :primary_key => "Factor_Code", :force => true do |t|
+          t.string "Factor_Code"
+          t.string "Description", :limit => 250
+        end
+
+        add_index "langdesc", ["Factor_Code"], :name => "LangDesc_Factor_Code_Index"
+
+        create_table "langual", :id => false, :primary_key => "NDB_No", :force => true do |t|
+          t.integer "NDB_No", :null => false
+          t.string  "Factor_Code", :limit => 6, :null => false
+        end
+
+        add_index "langual", ["NDB_No", "Factor_Code"], :name => "Langual_NDB_No_Factor_Code_Index", :unique => true
+
+        create_table "nut_data", :id => false, :primary_key => "NDB_No", :force => true do |t|
+          t.integer "NDB_No",        :null => false
+          t.integer "Nutr_No",       :null => false
+          t.float   "Nutr_Val"
+          t.integer "Num_Data_Pts"
+          t.float   "Std_Error"
+          t.string  "Src_Cd",        :limit => 2
+          t.string  "Deriv_Cd",      :limit => 4
+          t.integer "Ref_NDB_No"
+          t.string  "Add_Nutr_Mark", :limit => 1
+          t.integer "Num_Studies"
+          t.float   "Min"
+          t.float   "Max"
+          t.float   "DF"
+          t.float   "Low_EB"
+          t.float   "Up_EB"
+          t.string  "Stat_Cmt",      :limit => 10
+          t.date    "AddMod_Date"
+          t.string  "CC"
+        end
+
+        add_index "nut_data", ["NDB_No", "Nutr_No"], :name => "Nut_Data_NDB_No_Nutr_No_Index", :unique => true
+        add_index "nut_data", ["Num_Data_Pts"], :name => "Nut_Data_Num_Data_Pts_Index"
+        add_index "nut_data", ["Num_Studies"], :name => "Num_Studies_Index"
+
+        create_table "nutr_def", :primary_key => "Nutr_No", :force => true do |t|
+          t.string "Units",    :limit => 7
+          t.string "Tagname",  :limit => 20
+          t.string "NutrDesc", :limit => 60
+          t.string "Num_Dec",  :limit => 1
+          t.float  "SR_Order"
+        end
+
+        add_index "nutr_def", ["Num_Dec"], :name => "Num_Dec_Index"
+
+        create_table "src_cd", :primary_key => "Src_Cd", :force => true do |t|
+          t.string "SrcCd_Desc", :limit => 60
+        end
+
+        create_table "weight", :id => false, :primary_key => "NDB_No", :force => true do |t|
+          t.integer "NDB_No",       :null => false
+          t.integer "Seq",          :null => false
+          t.float   "Amount"
+          t.string  "Msre_Desc",    :limit => 80
+          t.float   "Gm_Wgt"
+          t.integer "Num_Data_Pts"
+          t.float   "Std_Dev"
+        end
+
+        add_index "weight", ["NDB_No", "Seq"], :name => "Weight_NDB_No_Seq_Index", :unique => true
+        add_index "weight", ["Num_Data_Pts"], :name => "Weight_Num_Data_Pts_Index"
+      end
+
+      def self.down
+        drop_table "abbrev"
+        drop_table "data_src"
+        drop_table "datsrcln"
+        drop_table "deriv_cd"
+        drop_table "fd_group"
+        drop_table "food_des"
+        drop_table "footnote"
+        drop_table "langdesc"
+        drop_table "langual"
+        drop_table "nut_data"
+        drop_table "nutr_def"
+        drop_table "src_cd"
+        drop_table "weight"
+      end
+    end
+  end
+end

data/lib/active_nutrition/models.rb

@@ -0,0 +1,19 @@
+# encoding: UTF-8
+
+module ActiveNutrition
+  module Models
+    autoload :Migration, 'active_nutrition/models/migration'
+    autoload :DataSrc,   'active_nutrition/models/data_src'
+    autoload :Datsrcln,  'active_nutrition/models/datsrcln'
+    autoload :DerivCd,   'active_nutrition/models/deriv_cd'
+    autoload :FdGroup,   'active_nutrition/models/fd_group'
+    autoload :FoodDes,   'active_nutrition/models/food_des'
+    autoload :Footnote,  'active_nutrition/models/footnote'
+    autoload :Langdesc,  'active_nutrition/models/langdesc'
+    autoload :Langual,   'active_nutrition/models/langual'
+    autoload :NutData,   'active_nutrition/models/nut_data'
+    autoload :NutrDef,   'active_nutrition/models/nutr_def'
+    autoload :SrcCd,     'active_nutrition/models/src_cd'
+    autoload :Weight,    'active_nutrition/models/weight'
+  end
+end

data/lib/active_nutrition/models/data_src.rb

@@ -0,0 +1,10 @@
+# encoding: UTF-8
+
+module ActiveNutrition
+  module Models
+    class DataSrc < ActiveRecord::Base
+      set_table_name "data_src"
+      set_primary_key :DataSrc_ID
+    end
+  end
+end

data/lib/active_nutrition/models/datsrcln.rb

@@ -0,0 +1,14 @@
+# encoding: UTF-8
+
+module ActiveNutrition
+  module Models
+    class Datsrcln < ActiveRecord::Base
+      set_table_name "datsrcln"
+      set_primary_key :NDB_No
+
+      validates :NDB_No, :uniqueness => { :scope => [:Nutr_No, :DataSrc_ID] }
+      validates :Nutr_No, :uniqueness => { :scope => [:NDB_No, :DataSrc_ID] }
+      validates :DataSrc_ID, :uniqueness => { :scope => [:NDB_No, :Nutr_No] }
+    end
+  end
+end

data/lib/active_nutrition/models/deriv_cd.rb

@@ -0,0 +1,10 @@
+# encoding: UTF-8
+
+module ActiveNutrition
+  module Models
+    class DerivCd < ActiveRecord::Base
+      set_table_name "deriv_cd"
+      set_primary_key :Deriv_CD
+    end
+  end
+end

data/lib/active_nutrition/models/fd_group.rb

@@ -0,0 +1,10 @@
+# encoding: UTF-8
+
+module ActiveNutrition
+  module Models
+    class FdGroup < ActiveRecord::Base
+      set_table_name "fd_group"
+      set_primary_key :FdGrp_CD
+    end
+  end
+end

data/lib/active_nutrition/models/food_des.rb

@@ -0,0 +1,19 @@
+# encoding: UTF-8
+
+module ActiveNutrition
+  module Models
+    class FoodDes < ActiveRecord::Base
+      set_table_name "food_des"
+      set_primary_key :NDB_No
+
+      alias_attribute :description, :Long_Desc
+      alias_attribute :common_name, :ComName
+      alias_attribute :manufacturer, :ManufacName
+
+      has_many :nutrition_facts, :class_name => "ActiveNutrition::Models::NutData", :foreign_key => "NDB_No", :primary_key => "NDB_No"
+      has_many :weights, :class_name => "ActiveNutrition::Models::Weight", :foreign_key => "NDB_No", :primary_key => "NDB_No"
+
+      belongs_to :food_group, :class_name => "ActiveNutrition::Models::FdGroup", :foreign_key => "FdGrp_Cd", :primary_key => "FdGrp_CD"
+    end
+  end
+end

data/lib/active_nutrition/models/footnote.rb

@@ -0,0 +1,10 @@
+# encoding: UTF-8
+
+module ActiveNutrition
+  module Models
+    class Footnote < ActiveRecord::Base
+      set_table_name "footnote"
+      set_primary_key :Footnt_No
+    end
+  end
+end

data/lib/active_nutrition/models/langdesc.rb

@@ -0,0 +1,10 @@
+# encoding: UTF-8
+
+module ActiveNutrition
+  module Models
+    class Langdesc < ActiveRecord::Base
+      set_table_name "langdesc"
+      set_primary_key :Factor_Code
+    end
+  end
+end

data/lib/active_nutrition/models/langual.rb

@@ -0,0 +1,13 @@
+# encoding: UTF-8
+
+module ActiveNutrition
+  module Models
+    class Langual < ActiveRecord::Base
+      set_table_name "langual"
+      set_primary_key :NDB_No
+
+      validates :NDB_No, :uniqueness => { :scope => [:Factor_Code] }
+      validates :Factor_Code, :uniqueness => { :scope => [:NDB_No] }
+    end
+  end
+end

data/lib/active_nutrition/models/migration.rb

@@ -0,0 +1,9 @@
+# encoding: UTF-8
+
+module ActiveNutrition
+  module Models
+    class Migration < ActiveRecord::Base
+      set_table_name "active_nutrition_migrations"
+    end
+  end
+end

data/lib/active_nutrition/models/nut_data.rb

@@ -0,0 +1,22 @@
+# encoding: UTF-8
+
+module ActiveNutrition
+  module Models
+    class NutData < ActiveRecord::Base
+      set_table_name "nut_data"
+      set_primary_key :NDB_No
+
+      alias_attribute :value, :Nutr_Val
+
+      belongs_to :definition, :class_name => "ActiveNutrition::Models::NutrDef", :foreign_key => "Nutr_No", :primary_key => "Nutr_No"
+      validates :NDB_No, :uniqueness => { :scope => [:Nutr_No] }
+      validates :Nutr_No, :uniqueness => { :scope => [:NDB_No] }
+
+      # the value of Nutr_Val is the number of grams in 100 grams total,
+      # which it makes it easy to calculate the amount in every 1 gram
+      def amount_per_gram
+        return self.Nutr_Val.to_f / 100.0
+      end
+    end
+  end
+end

data/lib/active_nutrition/models/nutr_def.rb

@@ -0,0 +1,14 @@
+# encoding: UTF-8
+
+module ActiveNutrition
+  module Models
+    class NutrDef < ActiveRecord::Base
+      set_table_name "nutr_def"
+      set_primary_key :Nutr_No
+
+      alias_attribute :units, :Units
+      alias_attribute :name, :NutrDesc
+      alias_attribute :tag_name, :Tagname
+    end
+  end
+end

data/lib/active_nutrition/models/src_cd.rb

@@ -0,0 +1,10 @@
+# encoding: UTF-8
+
+module ActiveNutrition
+  module Models
+    class SrcCd < ActiveRecord::Base
+      set_table_name "src_cd"
+      set_primary_key :Src_Cd
+    end
+  end
+end

data/lib/active_nutrition/models/weight.rb

@@ -0,0 +1,17 @@
+# encoding: UTF-8
+
+module ActiveNutrition
+  module Models
+    class Weight < ActiveRecord::Base
+      set_table_name "weight"
+      set_primary_key :NDB_No
+
+      validates :NDB_No, :uniqueness => { :scope => [:Seq] }
+      validates :Seq, :uniqueness => { :scope => [:NDB_No] }
+
+      alias_attribute :amount, :Amount
+      alias_attribute :grams, :Gm_Wgt
+      alias_attribute :measurement, :Msre_Desc
+    end
+  end
+end

data/lib/active_nutrition/objects.rb

@@ -0,0 +1,13 @@
+# encoding: UTF-8
+
+module ActiveNutrition
+  module Objects
+    autoload :Base,           'active_nutrition/objects/base'
+    autoload :Food,           'active_nutrition/objects/food'
+    autoload :Weight,         'active_nutrition/objects/weight'
+    autoload :Weights,        'active_nutrition/objects/weights'
+    autoload :FoodGroup,      'active_nutrition/objects/food_group'
+    autoload :NutritionFact,  'active_nutrition/objects/nutrition_fact'
+    autoload :NutritionFacts, 'active_nutrition/objects/nutrition_facts'
+  end
+end

data/lib/active_nutrition/objects/base.rb

@@ -0,0 +1,25 @@
+# encoding: UTF-8
+
+module ActiveNutrition
+  module Objects
+    class Base
+      attr_reader :attributes, :base_model
+
+      def initialize(base_model)
+        @base_model = base_model
+      end
+
+      def attributes
+        base_model.attributes
+      end
+
+      def self.wrap(obj)
+        if obj.is_a?(Array)
+          obj.map { |o| new(o) }
+        else
+          new(obj)
+        end
+      end
+    end
+  end
+end

data/lib/active_nutrition/objects/food.rb

@@ -0,0 +1,41 @@
+# encoding: UTF-8
+
+module ActiveNutrition
+  module Objects
+    class Food < Base
+      METHOD_MAP = { :name => :Long_Desc,
+                     :ndb_number => :NDB_No,
+                     :common_name => :ComName,
+                     :manufacturers_name => :ManufacName,
+                     :survey => :Survey,
+                     :scientific_name => :SciName,
+                     :fat_factor => :Fat_Factor,
+                     :protein_factor => :Pro_Factor }
+
+      METHOD_MAP.each_pair do |method, attr_name|
+        define_method(method) { attributes[attr_name.to_s] }
+      end
+
+      attr_reader :food_group, :nutrition_facts
+
+      def food_group
+        @food_group ||= FoodGroup.new(base_model.food_group)
+      end
+
+      def nutrition_facts
+        @nutrition_facts ||= NutritionFacts.new(NutritionFact.wrap(base_model.nutrition_facts))
+      end
+
+      def weights
+        @weights ||= Weights.new(Weight.wrap(base_model.weights))
+      end
+
+      def factors
+        [:fat_factor, :protein_factor].inject({}) do |ret, factor|
+          ret[factor] = attributes[METHOD_MAP[factor].to_s]
+          ret
+        end
+      end
+    end
+  end
+end

data/lib/active_nutrition/objects/food_group.rb

@@ -0,0 +1,14 @@
+# encoding: UTF-8
+
+module ActiveNutrition
+  module Objects
+    class FoodGroup < Base
+      METHOD_MAP = { :name => :FdGrp_Desc,
+                     :code => :FdGrp_CD }
+
+      METHOD_MAP.each_pair do |method, attr_name|
+        define_method(method) { attributes[attr_name.to_s] }
+      end
+    end
+  end
+end

data/lib/active_nutrition/objects/nutrition_fact.rb

@@ -0,0 +1,24 @@
+# encoding: UTF-8
+
+module ActiveNutrition
+  module Objects
+    class NutritionFact < Base
+      METHOD_MAP = { :nutrition_number => :Nutr_No,
+                     :value => :Nutr_Val,
+                     :num_data_points => :Num_Data_Pts,
+                     :standard_error => :Std_Error,
+                     :units => :Units,
+                     :description => :NutrDesc,
+                     :tag_name => :Tagname,
+                     :sr_order => :SR_Order }
+
+      METHOD_MAP.each_pair do |method, attr_name|
+        define_method(method) { self.attributes[attr_name.to_s] }
+      end
+
+      def attributes
+        @attributes ||= base_model.attributes.merge(base_model.definition.attributes)
+      end
+    end
+  end
+end

data/lib/active_nutrition/objects/nutrition_facts.rb

@@ -0,0 +1,15 @@
+# encoding: UTF-8
+
+module ActiveNutrition
+  module Objects
+    class NutritionFacts < Array
+      def to_hash(options = {})
+        options[:by] ||= :description
+        inject({}) do |ret, fact|
+          ret[fact.send(options[:by])] = fact.value
+          ret
+        end
+      end
+    end
+  end
+end

data/lib/active_nutrition/objects/weight.rb

@@ -0,0 +1,15 @@
+# encoding: UTF-8
+
+module ActiveNutrition
+  module Objects
+    class Weight < Base
+      METHOD_MAP = { :amount => :Amount,
+                     :grams => :Gm_Wgt,
+                     :measurement => :Msre_Desc }
+
+      METHOD_MAP.each_pair do |method, attr_name|
+        define_method(method) { self.attributes[attr_name.to_s] }
+      end
+    end
+  end
+end

data/lib/active_nutrition/objects/weights.rb

@@ -0,0 +1,15 @@
+# encoding: UTF-8
+
+module ActiveNutrition
+  module Objects
+    class Weights < Array
+      def to_hash(options = {})
+        options[:by] ||= :measurement
+        inject({}) do |ret, weight|
+          ret[weight.send(options[:by])] = weight.grams
+          ret
+        end
+      end
+    end
+  end
+end

data/lib/active_nutrition/railtie.rb

@@ -0,0 +1,29 @@
+# encoding: UTF-8
+
+include Rake::DSL
+
+module ActiveNutrition
+  class Railtie < Rails::Railtie
+    rake_tasks do
+      namespace :active_nutrition do
+        task :migrate => :environment do
+          ActiveNutrition.migrate
+        end
+
+        task :update => :environment do
+          raise "Not yet supported."
+        end
+
+        task :rebuild => :environment do
+          ActiveNutrition.rebuild
+        end
+      end
+
+      namespace :an do
+        task :migrate => "active_nutrition:migrate"
+        task :update => "active_nutrition:update"
+        task :rebuild => "active_nutrition:rebuild"
+      end
+    end
+  end
+end

data/lib/active_nutrition/utilities.rb

@@ -0,0 +1,9 @@
+# encoding: UTF-8
+
+module ActiveNutrition
+  module Utilities
+    autoload :DataFile,      'active_nutrition/utilities/data_file'
+    autoload :Update,        'active_nutrition/utilities/update'
+    autoload :UnitConverter, 'active_nutrition/utilities/unit_converter'
+  end
+end

data/lib/active_nutrition/utilities/data_file.rb

@@ -0,0 +1,68 @@
+# encoding: UTF-8
+
+module ActiveNutrition
+  module Utilities
+    class DataFile
+      attr_reader :file, :handle
+
+      FIELD_SEP = '^'
+      FIELD_ENC = '~'
+      NULL_FIELDS = ['^^', '~~', "\r\n", '']
+
+      def initialize(file)
+        @file = file
+        @handle = File.open(file, "r")
+      end
+
+      def each
+        @handle.each do |line|
+          yield self.parse_line(line)
+        end
+        self.clean_up
+      end
+
+      protected
+
+      def clean_up
+        @handle.close
+      end
+
+      def parse_line(line)
+        line.force_encoding("ISO-8859-1")
+        #line.byte_size = 8
+        line.chomp!("\r\n")
+        fields = line.split('^')
+        p_fields = []
+
+        fields.each do |field|
+          # strip the '~' if present
+          if field.index('"')
+            field.gsub!(/[\"]/, "'")
+          end
+
+          if field.index(FIELD_ENC) && field.size > 2
+            p_fields << field[1..-2]
+          else
+            # else if the field matches a null field
+            if NULL_FIELDS.include?(field)
+              p_fields << ''
+            elsif field.include?('.')
+              # else the field must be a number
+              p_fields << field.to_f
+            elsif field.include?('/')
+              # check to see if its a date, add as a string for now
+              p_fields << field
+            else
+              p_fields << field.to_i  
+            end
+          end
+        end
+        p_fields
+      rescue => e
+        puts line
+        puts e.message
+        puts e.backtrace
+      end
+    end
+  end
+end

data/lib/active_nutrition/utilities/unit_converter.rb

@@ -0,0 +1,32 @@
+# encoding: UTF-8
+
+module ActiveNutrition
+  module Utilities
+    class UnitConverter
+      CONVERSIONS = {
+        :oz =>     { :lb => 16.0 },  # i.e. to convert from pounds to ounces, multiply by 16 (because there are 16 ounces per pound)
+        :lb =>     { :oz => 1.0/16.0 },
+        :tsp =>    { :tbsp => 3.0,      :cup  => 48.0,      :pint => 96.0,    :quart => 192.0,  :gallon => 768.0 },
+        :tbsp =>   { :tsp => 1.0/3.0,   :cup  => 16.0,      :pint => 32.0,    :quart => 64.0,   :gallon => 256.0 },
+        :cup =>    { :tsp => 1.0/48.0,  :tbsp => 1.0/16.0,  :pint => 2.0,     :quart => 4.0,    :gallon => 16.0 },
+        :pint =>   { :tsp => 1.0/96.0,  :tbsp => 1.0/32.0,  :cup => 1.0/2.0,  :quart => 2.0,    :gallon => 8.0 },
+        :quart =>  { :tsp => 1.0/192.0, :tbsp => 1.0/64.0,  :cup => 1.0/4.0,  :pint => 1.0/2.0, :gallon => 4.0 },
+        :gallon => { :tsp => 1.0/768.0, :tbsp => 1.0/256.0, :cup => 1.0/16.0, :pint => 1.0/8.0, :quart => 1.0/4.0 }
+      }
+
+      class << self
+        def convert(value, options = {})
+          unless supported_conversion?(options)
+            raise "Conversion not supported - are you trying to convert weight to volume?"
+          end
+
+          value * CONVERSIONS[options[:to]][options[:from]]
+        end
+
+        def supported_conversion?(options = {})
+          CONVERSIONS.include?(options[:to]) && CONVERSIONS[options[:to]].include?(options[:from])
+        end
+      end
+    end
+  end
+end

data/lib/active_nutrition/utilities/update.rb

@@ -0,0 +1,100 @@
+# encoding: UTF-8
+
+module ActiveNutrition
+  module Utilities
+    class Update
+      BASE_URL = "http://www.ars.usda.gov/"
+      BASE_PATH = "SP2UserFiles/Place/12354500/Data/SR{{release}}/dnload/"
+      FULL_FILE = "sr{{release}}.zip"
+      UPDATE_FILE = "sr{{release}}upd.zip"
+      CURRENT_RELEASE = ActiveNutrition::USDA_VERSION
+      DATA_DIR = File.join(File.dirname(File.dirname(File.dirname(File.dirname(__FILE__)))), "data")
+      IMPORT_CHUNK_SIZE = 1000
+
+      def initialize(options = {})
+        @release = (options[:release] || CURRENT_RELEASE).to_s
+        @usda_map = YAML::load_file(File.join(File.dirname(__FILE__), "usda_map.yml"))
+        @type = options[:type] || :update
+        @path, @file = self.url_for(@type)
+        @zip_file = File.join(DATA_DIR, @file)
+        @zip_dir = @zip_file.chomp(File.extname(@zip_file))
+      end
+
+      def execute
+        @usda_map.each_pair do |model_const, data|
+          if ActiveNutrition::Models.const_defined?(model_const.to_sym)
+            model = ActiveNutrition::Models.const_get(model_const.to_sym)
+            cur_file = File.join(@zip_dir, data["file_name"])
+            parser = DataFile.new(cur_file)
+            record_total = `wc -l "#{cur_file}"`.to_i
+            record_count = 0
+
+            parser.each do |values|
+              new_record = model.new
+
+              data["attribute_order"].each_with_index do |field, index|
+                # update each attribute by explicitly calling the attr= method, just in case
+                # there are custom methods in the model class
+                new_record.send(:"#{field}=", values[index])
+              end
+
+              new_record.save!
+
+              if record_count % IMPORT_CHUNK_SIZE == 0 || (record_count + 1) == record_total || record_count == 0
+                yield [model.to_s, record_count, record_total] if block_given?
+              end
+
+              record_count += 1
+            end
+          end
+        end
+      end
+
+      def reset_db
+        @usda_map.each_key do |model_const|
+          if ActiveNutrition::Models.const_defined?(model_const.to_sym)
+            ActiveNutrition::Models.const_get(model_const.to_sym).delete_all
+          end
+        end
+      end
+
+      def clean
+        File.unlink(@zip_file) if File.exist?(@zip_file)
+        FileUtils.rm_rf(@zip_dir) if File.exist?(@zip_dir)
+      end
+
+      def download
+        unless File.file?(File.join(DATA_DIR, @file))
+          FileUtils.mkdir_p(DATA_DIR)
+          File.open(File.join(DATA_DIR, @file), "wb") do |f|
+            f.sync = true
+            f.write(open("#{BASE_URL}#{@path}#{@file}").read)
+          end
+        end
+      end
+
+      def unzip
+        Zip::ZipFile.open(@zip_file) do |zip_file|
+          zip_file.each do |f|
+            next unless f.file?
+            f_path = File.join(@zip_dir, f.name)
+            FileUtils.mkdir_p(File.dirname(f_path))
+            zip_file.extract(f, f_path) unless File.exist?(f_path)
+          end
+        end
+      end
+
+      protected
+
+      def url_for(type = :full)
+        file = case type
+          when :full then FULL_FILE
+          when :update then UPDATE_FILE
+          else UPDATE_FILE
+        end
+
+        [BASE_PATH.gsub("{{release}}", @release), file.gsub("{{release}}", @release)]
+      end
+    end
+  end
+end

data/lib/active_nutrition/utilities/usda_map.yml

@@ -0,0 +1,119 @@
+FoodDes:
+  file_name: FOOD_DES.txt
+  attribute_order:
+    - NDB_No
+    - FdGrp_Cd
+    - Long_Desc
+    - Shrt_Desc
+    - ComName
+    - ManufacName
+    - Survey
+    - Ref_Desc
+    - Refuse
+    - SciName
+    - N_Factor
+    - Pro_Factor
+    - Fat_Factor
+    - CHO_Factor
+
+NutData:
+  file_name: NUT_DATA.txt
+  attribute_order:
+    - NDB_No
+    - Nutr_No
+    - Nutr_Val
+    - Num_Data_Pts
+    - Std_Error
+    - Src_Cd
+    - Deriv_Cd
+    - Ref_NDB_No
+    - Add_Nutr_Mark
+    - Num_Studies
+    - Min
+    - Max
+    - DF
+    - Low_EB
+    - Up_EB
+    - Stat_Cmt
+    - AddMod_Date
+    - CC
+
+Weight:
+  file_name: WEIGHT.txt
+  attribute_order:
+    - NDB_No
+    - Seq
+    - Amount
+    - Msre_Desc
+    - Gm_Wgt
+    - Num_Data_Pts
+    - Std_Dev
+
+Footnote:
+  file_name: FOOTNOTE.txt
+  attribute_order:
+    - NDB_No
+    - Footnt_No
+    - Footnot_Typ
+    - Nutr_No
+    - Footnot_Txt
+
+FdGroup:
+  file_name: FD_GROUP.txt
+  attribute_order:
+    - FdGrp_CD
+    - FdGrp_Desc
+
+Langual:
+  file_name: LANGUAL.txt
+  attribute_order:
+    - NDB_No
+    - Factor_Code
+
+Langdesc:
+  file_name: LANGDESC.txt
+  attribute_order:
+    - Factor_Code
+    - Description
+
+NutrDef:
+  file_name: NUTR_DEF.txt
+  attribute_order:
+    - Nutr_No
+    - Units
+    - Tagname
+    - NutrDesc
+    - Num_Dec
+    - SR_Order
+
+SrcCd:
+  file_name: SRC_CD.txt
+  attribute_order:
+    - Src_Cd
+    - SrcCd_Desc
+
+DerivCd:
+  file_name: DERIV_CD.txt
+  attribute_order:
+    - Deriv_CD
+    - Deriv_Desc
+
+DataSrc:
+  file_name: DATA_SRC.txt
+  attribute_order:
+    - DataSrc_ID
+    - Authors
+    - Title
+    - Year
+    - Journal
+    - Vol_city
+    - Issue_State
+    - Start_Page
+    - End_Page
+
+Datsrcln:
+  file_name: DATSRCLN.txt
+  attribute_order:
+    - NDB_No
+    - Nutr_No
+    - DataSrc_ID

data/lib/active_nutrition/version.rb

@@ -0,0 +1,6 @@
+# encoding: UTF-8
+
+module ActiveNutrition
+  VERSION = "0.5.0"
+  USDA_VERSION = "24"
+end

metadata

@@ -0,0 +1,127 @@
+--- !ruby/object:Gem::Specification
+name: active_nutrition
+version: !ruby/object:Gem::Version
+  version: 0.5.0
+  prerelease: 
+platform: ruby
+authors:
+- Cameron Dutro
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-07-03 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: activerecord
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '3.0'
+- !ruby/object:Gem::Dependency
+  name: rubyzip
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.9.4
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.9.4
+- !ruby/object:Gem::Dependency
+  name: composite_primary_keys
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 4.1.2
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 4.1.2
+description: Updates, maintains, and provides systematic access to USDA nutrition
+  data.  See lib/examples to get started.
+email:
+- camertron@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/active-nutrition.rb
+- lib/active_nutrition/migrations/00000000001_migrations_table.rb
+- lib/active_nutrition/migrations/00000000002_sr_24.rb
+- lib/active_nutrition/models/data_src.rb
+- lib/active_nutrition/models/datsrcln.rb
+- lib/active_nutrition/models/deriv_cd.rb
+- lib/active_nutrition/models/fd_group.rb
+- lib/active_nutrition/models/food_des.rb
+- lib/active_nutrition/models/footnote.rb
+- lib/active_nutrition/models/langdesc.rb
+- lib/active_nutrition/models/langual.rb
+- lib/active_nutrition/models/migration.rb
+- lib/active_nutrition/models/nut_data.rb
+- lib/active_nutrition/models/nutr_def.rb
+- lib/active_nutrition/models/src_cd.rb
+- lib/active_nutrition/models/weight.rb
+- lib/active_nutrition/models.rb
+- lib/active_nutrition/objects/base.rb
+- lib/active_nutrition/objects/food.rb
+- lib/active_nutrition/objects/food_group.rb
+- lib/active_nutrition/objects/nutrition_fact.rb
+- lib/active_nutrition/objects/nutrition_facts.rb
+- lib/active_nutrition/objects/weight.rb
+- lib/active_nutrition/objects/weights.rb
+- lib/active_nutrition/objects.rb
+- lib/active_nutrition/railtie.rb
+- lib/active_nutrition/utilities/data_file.rb
+- lib/active_nutrition/utilities/unit_converter.rb
+- lib/active_nutrition/utilities/update.rb
+- lib/active_nutrition/utilities/usda_map.yml
+- lib/active_nutrition/utilities.rb
+- lib/active_nutrition/version.rb
+- lib/active_nutrition.rb
+homepage: http://github.com/camertron
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: active_nutrition
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: USDA nutrition data
+test_files: []