Fast_Food_Nutrition 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d6d91d9e922a4445184cbe3f025eff17e794d745875c071624e27d31bb437a33
-  data.tar.gz: b0c8ffd6947c2f4206cd86f829c5c9b7fc9f30008d8b008e5885d6ec96b56e4b
+  metadata.gz: 9f360021fb48ae1d58ca22ca8d7d7c684c014094558645982ffde43dea5e59d4
+  data.tar.gz: 95abd536119b466e7f68df58ed760fec36d64080c3431ec2a803138ebcbf7764
 SHA512:
-  metadata.gz: e1fb70c3dcfcf2fffcde6c73daf0edbe23e425435d765f0803c03224d832121a15d83bd1a2227097c5166e9403dc9f211351329626a8b5757eaff3c66ec670fc
-  data.tar.gz: df126e92d5d929f68b4d6c7d73fd6d265f417da81e746e9ce78e6c07873df50ef314ec187c74281a5cd3e9ada3264793c4ae762b3c1af00b3971153c4e86e7ce
+  metadata.gz: b584351e44e9e48b9bf9295aff2f77a07af9dcd5ac242ce6520875c2b41c206a8a1e1f4427daa44cbb2cebab1ac39664ac4b8f2276f7cdc4c40a79a49cdc1005
+  data.tar.gz: 2a1c975fa1ecdf6c02c022d4907e3b01db386a0ea7db29b15a2b3015000305267168e0bd363085a47f1ef87e4a2c77c19e077cc535f1b282823af3ece47c0308

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    Fast_Food_Nutrition (0.1.0)
+    Fast_Food_Nutrition (0.2.0)
       nokogiri (~> 1.8, >= 1.8.2)
       open_uri_redirections (~> 0.2.1)
 

data/lib/Fast_Food_Nutrition/category.rb

@@ -0,0 +1,22 @@
+class Category
+  attr_accessor :name
+  @@all = []
+
+  def initialize(name)
+    @name = name
+    @@all << self
+  end
+
+  def self.all
+    @@all
+  end
+
+  def items=(item_array)
+    @items = item_array
+  end
+
+  def items
+    @items
+  end
+
+end

data/lib/Fast_Food_Nutrition/cli.rb

@@ -8,38 +8,39 @@ class Welcome
   def list_restaurants
     selection = 0
     puts "\nPlease select a Restaurant by number:"
-    restaurants = Scraper.new.scrape_site_for_restaurants("http://www.nutrition-charts.com/")
-    restaurants.each_with_index {|restaurant, i| puts "#{i + 1}: #{restaurant.flatten[0]}"}
-    selection = gets.strip.to_i until selection > 0 && selection <= restaurants.length
-    puts "\nYou selected #{restaurants[selection - 1].flatten[0]}"
-    select_category(Scraper.new.scrape_restaurant_categories(restaurants[selection - 1]),"http://www.nutrition-charts.com/#{restaurants[selection - 1].flatten[1]}", restaurants[selection - 1].flatten[0])
+    Scraper.new.scrape_site_for_restaurants("http://www.nutrition-charts.com/") if Restaurant.all.size == 0
+    Restaurant.all.each_with_index {|restaurant, i| puts "#{i + 1}: #{restaurant.name}"}
+    selection = gets.strip.to_i until selection > 0 && selection <= Restaurant.all.size
+    puts "\nYou selected #{Restaurant.all[selection - 1].name}"
+    select_category(Restaurant.all[selection - 1])
   end
 
-  def select_category(categories, item_site, restaurant)
+  def select_category(restaurant)
+    restaurant.categories = Scraper.new.scrape_restaurant_categories(restaurant) if !restaurant.categories
     selection = 0
     puts "\nPlease select a Category of Menu Items:"
-    categories.each_with_index {|category, i| puts "#{i + 1}: #{category}"}
-    selection = gets.strip.to_i until selection > 0 && selection <= categories.length
-    puts "\nYou selected #{categories[selection - 1]}"
-    select_item(Scraper.new.scrape_category_items(categories[selection - 1], selection, item_site, restaurant), item_site, restaurant)
+    restaurant.categories.each_with_index {|category, i| puts "#{i + 1}: #{category.name}"}
+    selection = gets.strip.to_i until selection > 0 && selection <= restaurant.categories.size
+    puts "\nYou selected #{restaurant.categories[selection - 1].name}"
+    select_item(restaurant, selection - 1)
   end
 
-  def select_item(menu_item_list,item_site, restaurant)
-    i = 1
+  def select_item(restaurant, cat_picked)
+    restaurant.categories[cat_picked].items = Scraper.new.scrape_category_items(restaurant, cat_picked) if !restaurant.categories[cat_picked].items
     selection = 0
     puts "\nPlease select an item:"
-    menu_item_list.each do |item|
-      puts "#{i}: #{item}"
-      i += 1
+    restaurant.categories[cat_picked].items.each_with_index do |item, i|
+      puts "#{i + 1}: #{item.name}"
     end
-    selection = gets.strip.to_i until selection > 0 && selection <= menu_item_list.length
-    puts "\nYou selected #{menu_item_list[selection - 1]}"
-    get_nutrition(Scraper.new.scrape_nutrition_info(menu_item_list[selection - 1], item_site, restaurant), menu_item_list[selection - 1])
+    selection = gets.strip.to_i until selection > 0 && selection <= restaurant.categories[cat_picked].items.size
+    puts "\nYou selected #{restaurant.categories[cat_picked].items[selection - 1].name}"
+    get_nutrition(restaurant, cat_picked, selection - 1)
   end
 
-  def get_nutrition(nutrition_list, item_name)
-    puts "\nHere is the nutrition information for #{item_name}:"
-    nutrition_list.each do |item|
+  def get_nutrition(restaurant, cat_picked, item_picked)
+    restaurant.categories[cat_picked].items[item_picked].nutrition = Scraper.new.scrape_nutrition_info(restaurant, cat_picked, item_picked) if !restaurant.categories[cat_picked].items[item_picked].nutrition
+    puts "\nHere is the nutrition information for #{restaurant.categories[cat_picked].items[item_picked].name}:"
+    restaurant.categories[cat_picked].items[item_picked].nutrition.each do |item|
       puts "  #{item[0]}: #{item[1]}"
     end
     continue?

data/lib/Fast_Food_Nutrition/items.rb

@@ -0,0 +1,21 @@
+class Item
+  attr_accessor :name
+  @@all = []
+
+  def initialize(name)
+    @name = name
+  end
+
+  def self.all
+    @@all
+  end
+
+  def nutrition=(nutrition_array)
+    @nutrition = nutrition_array
+  end
+
+  def nutrition
+    @nutrition
+  end
+
+end

data/lib/Fast_Food_Nutrition/restaurant.rb

@@ -0,0 +1,32 @@
+class Restaurant
+  attr_accessor :name, :url
+  @@all = []
+  def initialize(name, url)
+    @name = name
+    @url = url
+    @@all << self
+  end
+
+  def self.all
+    @@all
+  end
+
+  def categories=(category_array)
+    @categories = category_array
+  end
+
+  def categories
+    @categories
+  end
+
+  def self.find_restaurant_by_name(name)
+    restaurant = ""
+    self.all.each do |location|
+      if name == location.name
+        restaurant = location
+      end
+    end
+    restaurant
+  end
+
+end

data/lib/Fast_Food_Nutrition/scraper.rb

@@ -1,85 +1,101 @@
-require 'nokogiri'
-require 'open-uri'
 require 'pry'
+require 'Nokogiri'
+require 'open-uri'
 
 class Scraper
   def scrape_site_for_restaurants(url)
-    restaurants = []
     site = Nokogiri::HTML(open(url))
     site = site.css("div.entry-content table tbody tr")
     site.each do |restaurant|
-      restaurants << {restaurant.css("td")[1].css("a").text.strip => restaurant.css("td")[1].css("a").attribute("href").value}
-      restaurants << {restaurant.css("td")[3].css("a").text.strip => restaurant.css("td")[3].css("a").attribute("href").value}
+      Restaurant.new(restaurant.css("td")[1].css("a").text.strip, "http://www.nutrition-charts.com/#{restaurant.css("td")[1].css("a").attribute("href").value}")
+      Restaurant.new(restaurant.css("td")[3].css("a").text.strip, "http://www.nutrition-charts.com/#{restaurant.css("td")[3].css("a").attribute("href").value}")
     end
-    restaurants
+    # Restaurant.all.each do |location|
+    #   puts "#{location.name}"
+    #   location.categories = scrape_restaurant_categories(location)
+    #   #binding.pry
+    # end
+    # binding.pry
   end
 
-  def scrape_restaurant_categories(category)
+  def scrape_restaurant_categories(location)
     category_list = []
-    url = "http://www.nutrition-charts.com/#{category.flatten[1]}"
+    url = "#{location.url}"
     site = Nokogiri::HTML(open(url))
-    case category.flatten[0]
+    case location.name
     when "Burger King"
       site = site.css("div table tbody td h3")
       site.each do |item|
-        category_list << item.text
+        category = Category.new(item.text)
+        category_list << category
       end
       category_list
     when "Wendys"
       site = site.css("div table tbody td h3")
       site.each do |item|
-        category_list << item.text
+        category = Category.new(item.text)
+        category_list << category
       end
       category_list
     when "Buffalo Wild Wings"
       site = site.css("div table tbody td h3")
       site.each do |item|
-        category_list << item.text
+        category = Category.new(item.text)
+        category_list << category
       end
       category_list
     when "Pizza Hut"
       site = site.css("div table tbody td strong")
       site.each do |item|
-        category_list << item.text
+        category = Category.new(item.text)
+        category_list << category
       end
       category_list
     when "Applebees"
       site = site.css("div table tbody")
       site = site.css("tr.rowheader")
       site.each do |item|
-        category_list << item.css("td")[0].text
+        category = Category.new(item.css("td")[0].text)
+        category_list << category
       end
       category_list
-    when "Baja Fresh"
+    when "Baja Fresh" #Still having issues with this one
       site = site.css("div table tbody tr")
       site.each do |item|
-        item.to_s.include?("<th>") ? category_list << item.css("th")[0].text : ""
+        item.to_s.include?("<th>") ? category = Category.new(item.css("th")[0].text) : ""
+        #category = Category.new(item.css("th")[0].text)
+        if category
+          category_list << category
+        end
       end
       category_list
     else
       site = site.css("div table tbody")
-      category_list << site[0].css("tr")[0].css("th")[0].text
+      category = Category.new(site[0].css("tr")[0].css("th")[0].text)
+      category_list << category
       site = site.css("tr.rowheader")
       site.each do |item|
-        category_list << item.css("th")[0].text
+        category = Category.new(item.css("th")[0].text)
+        category_list << category
       end
       category_list
     end
   end
 
-  def scrape_category_items(item_name, index, item_site, restaurant)
+  def scrape_category_items(restaurant, category)
       item_list = []
-      site = Nokogiri::HTML(open(item_site))
-      case restaurant
+      url = "#{restaurant.url}"
+      site = Nokogiri::HTML(open(url))
+      case restaurant.name
       when "Burger King"
         site = site.css("div table tbody tr")
         site.each do |item|
           if item.to_s.include?("<h3>")
-            if item.css("h3")[0].text == item_name
+            if item.css("h3")[0].text == restaurant.categories[category].name
               next_item = item.next_element
               begin
                 if next_item.to_s.include?("<td>") && !next_item.to_s.include?("<span") && !next_item.to_s.include?("header")
-                  item_list << next_item.css("td")[0].text
+                  item_list << Item.new(next_item.css("td")[0].text)
                 end
                 next_item = next_item.next_element
                 next_item == nil ? next_item = "<h3>" : next_item
@@ -92,11 +108,11 @@ class Scraper
         site = site.css("div table tbody tr")
         site.each do |item|
           if item.to_s.include?("<h3>")
-            if item.css("h3")[0].text == item_name
+            if item.css("h3")[0].text == restaurant.categories[category].name
               next_item = item.next_element
               begin
                 if next_item.to_s.include?("<td>") && !next_item.to_s.include?("<span") && !next_item.to_s.include?("header")
-                  item_list << next_item.css("td")[0].text
+                  item_list << Item.new(next_item.css("td")[0].text)
                 end
                 next_item = next_item.next_element
                 next_item == nil ? next_item = "<h3>" : next_item
@@ -109,11 +125,11 @@ class Scraper
         site = site.css("div table tbody tr")
         site.each do |item|
           if item.to_s.include?("<h3>")
-            if item.css("h3")[0].text == item_name
+            if item.css("h3")[0].text == restaurant.categories[category].name
               next_item = item.next_element
               begin
                 if next_item.to_s.include?("<td>") && !next_item.to_s.include?("<span") && !next_item.to_s.include?("header") && next_item.css("td")[0].text != " "
-                  item_list << next_item.css("td")[0].text
+                  item_list << Item.new(next_item.css("td")[0].text)
                 end
                 next_item = next_item.next_element
                 next_item == nil ? next_item = "<h3>" : next_item
@@ -126,10 +142,10 @@ class Scraper
         site = site.css("div table tbody tr")
         site.each do |item|
           if item.to_s.include?("<strong>")
-            if item.css("strong")[0].text == item_name
+            if item.css("strong")[0].text == restaurant.categories[category].name
               next_item = item.next_element
               begin
-                item_list << next_item.css("td")[0].text if !next_item.to_s.include?("<span")
+                item_list << Item.new(next_item.css("td")[0].text) if !next_item.to_s.include?("<span")
                 next_item = next_item.next_element
                 next_item == nil ? next_item = "<strong>" : next_item
               end while !next_item.to_s.include?("<strong>")
@@ -141,10 +157,10 @@ class Scraper
        site = site.css("div table tbody tr")
        site.each do |item|
           if item.to_s.include?("rowheader")
-            if item.css("td")[0].text == item_name
+            if item.css("td")[0].text == restaurant.categories[category].name
               next_item = item.next_element
               begin
-                item_list << next_item.css("td")[0].text if !next_item.to_s.include?("<span")
+                item_list << Item.new(next_item.css("td")[0].text) if !next_item.to_s.include?("<span")
                 next_item = next_item.next_element
                 next_item == nil ? next_item = "<rowheader>" : next_item
               end while !next_item.to_s.include?("rowheader")
@@ -156,10 +172,10 @@ class Scraper
       site = site.css("div table tbody tr")
       site.each do |item|
         if item.to_s.include?("<th>")
-          if item.css("th")[0].text == item_name
+          if item.css("th")[0].text == restaurant.categories[category].name
             next_item = item.next_element
             begin
-              item_list << next_item.css("td")[0].text if !next_item.to_s.include?("<span")
+              item_list << Item.new(next_item.css("td")[0].text) if !next_item.to_s.include?("<span")
               next_item = next_item.next_element
               next_item == nil ? next_item = "<th>" : next_item
             end while !next_item.to_s.include?("<th>")
@@ -170,18 +186,18 @@ class Scraper
     end
   end
 
-  def scrape_nutrition_info(item_name, item_site, restaurant)
+  def scrape_nutrition_info(restaurant, cat_picked, item_picked)
     index = 1
     nutrition_list = []
-    site = Nokogiri::HTML(open(item_site))
-    case restaurant
+    site = Nokogiri::HTML(open(restaurant.url))
+    case restaurant.name
     when "Burger King"
       nutrition_items = site.css("div table thead tr th")
       nutrition_items.each {|desc| nutrition_list << [desc.text.strip, ""]}
       site = site.css("div table tbody tr")
       site.each do |item|
         if item.to_s.include?("<td>")
-          if item.css("td")[0].text == item_name
+          if item.css("td")[0].text == restaurant.categories[cat_picked].items[item_picked].name
             i = 0
             while i < nutrition_list.length
               nutrition_list[i][1] = item.css("td")[i].text
@@ -198,7 +214,7 @@ class Scraper
       site = site.css("div table tbody tr")
       site.each do |item|
         if item.to_s.include?("<td>")
-          if item.css("td")[0].text == item_name
+          if item.css("td")[0].text == restaurant.categories[cat_picked].items[item_picked].name
             i = 0
             while i < nutrition_list.length
               nutrition_list[i][1] = item.css("td")[i].text
@@ -215,7 +231,7 @@ class Scraper
       site = site.css("div table tbody tr")
       site.each do |item|
         if item.to_s.include?("<td>")
-          if item.css("td")[0].text == item_name
+          if item.css("td")[0].text == restaurant.categories[cat_picked].items[item_picked].name
             i = 0
             while i < nutrition_list.length
               nutrition_list[i][1] = item.css("td")[i].text

data/lib/Fast_Food_Nutrition/version.rb

@@ -1,3 +1,3 @@
 module FastFoodNutrition
-  VERSION = "0.1.0"
+  VERSION = "0.2.0"
 end

data/lib/Fast_Food_Nutrition.rb

@@ -1,3 +1,6 @@
 require_relative "./Fast_Food_Nutrition/version"
 require_relative "./Fast_Food_Nutrition/cli"
 require_relative "./Fast_Food_Nutrition/scraper"
+require_relative "./Fast_Food_Nutrition/restaurant"
+require_relative "./Fast_Food_Nutrition/category"
+require_relative "./Fast_Food_Nutrition/items"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: Fast_Food_Nutrition
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Daniel Hernandez
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2018-06-16 00:00:00.000000000 Z
+date: 2018-06-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: open_uri_redirections
@@ -108,7 +108,10 @@ files:
 - bin/ff_nutrition
 - bin/setup
 - lib/Fast_Food_Nutrition.rb
+- lib/Fast_Food_Nutrition/category.rb
 - lib/Fast_Food_Nutrition/cli.rb
+- lib/Fast_Food_Nutrition/items.rb
+- lib/Fast_Food_Nutrition/restaurant.rb
 - lib/Fast_Food_Nutrition/scraper.rb
 - lib/Fast_Food_Nutrition/version.rb
 - spec.md