Final Pull Request

lib/far_mar.rb

@@ -1,5 +1,6 @@
 require 'csv'
 require 'time'
+require 'pry'
 require './lib/far_mar/market'
 require './lib/far_mar/vendor'
 require './lib/far_mar/sale'

lib/far_mar/market.rb

@@ -1,13 +1,121 @@
-# require './support/markets.csv'
+# CSV.read('./support/markets.csv')
 
 module FarMar
 
   class Market
+    attr_reader :market_id, :name, :address, :city, :county, :state, :zip
 
-  	def initialize
-
+  	def initialize(market_id, name, address, city, county, state, zip)
+  		@market_id = market_id.to_i
+  		@name = name
+  		@address = address
+  		@city = city
+  		@county = county
+  		@state = state
+  		@zip = zip
   	end
 
-  end
+  	def self.all
+      @@market_array ||= []
+      if @@market_array == []
+  		  CSV.read('./support/markets.csv').each do |row|
+          market = FarMar::Market.new(row[0], row[1], row[2], row[3], row[4], row[5], row[6])
+          @@market_array.push(market)
+        end
+      end
+      return @@market_array
+  	end
+
+    def self.find(id)
+      FarMar::Market.all.find {|mar| mar.market_id == id }
+    end
+
+    def vendors
+      FarMar::Vendor.all.find_all {|ven| ven.market == market_id }
+    end
+
+    def products
+      ven_array = self.vendors
+      prod_array = []
+      ven_array.each do |ven|
+        prod_array.push(ven.products)
+        end
+      return prod_array.flatten
+    end
 
+    def self.search(search_term)
+      search_term = search_term.downcase
+      match_markets = []
+      FarMar::Market.all.each do |market|
+        match_markets.push(market) if market.name.downcase.match(/#{search_term}/)
+      end
+      match_vendors = []
+      FarMar::Vendor.all.each do |vendor|
+        match_vendors.push(vendor) if vendor.name.downcase.match(/#{search_term}/)
+       end
+       match_vendors.map! {|vendor| vendor.market}
+       match_vendors.each do |market_id|
+         market = FarMar::Market.find(market_id)
+         match_markets.push(market)
+       end
+       return match_markets
+    end
+
+    def preferred_vendor(date = nil)
+      if date == nil
+        ven = self.vendors
+        preferred = ven.max_by{|ven| ven.revenue}
+      else 
+        date = DateTime.strptime(date, "%Y-%m-%d").to_date
+        max_revenue = 0
+        pref_vend = nil
+        total = 0
+        self.vendors.each do |vendor_inst|
+          sales_array = []
+          vendor_inst.sales.each do |sale_inst|
+            if sale_inst.purchase_time.to_date == date
+            sales_array.push(sale_inst.amount)
+            end
+          end
+          if sales_array.length > 0
+            total = sales_array.inject(0, :+)
+            if total > max_revenue
+              max_revenue = total
+              pref_vend = vendor_inst
+            end
+          end
+        end
+        return pref_vend
+      end
+    end
+
+    def worst_vendor(date = nil)
+      if date == nil        
+        ven = self.vendors
+        worst = ven.min_by{|ven| ven.revenue}
+      else
+        date = DateTime.strptime(date, "%Y-%m-%d").to_date
+        min_revenue = Float::INFINITY        
+        worst_ven = nil
+        total = 0
+        self.vendors.each do |vendor_inst|
+          sales_array = []
+          vendor_inst.sales.each do |sale_inst|
+            if sale_inst.purchase_time.to_date == date
+             sales_array.push(sale_inst.amount)
+            end
+          end
+          if sales_array.length > 0
+            total = sales_array.inject(0, :+)
+            if total < min_revenue
+              min_revenue = total
+              worst_ven = vendor_inst
+            end
+         end
+        end
+        return worst_ven
+      end
+    end
+
+  end
 end

lib/far_mar/product.rb

@@ -1,7 +1,48 @@
+# CSV.read("./support/products.csv")
+
 module FarMar
 
 	class Product
+		attr_reader :product_id, :name, :vendor_id
+
+		def initialize(product_id, name, vendor_id)
+			@product_id = product_id.to_i
+			@name = name.to_s
+			@vendor_id = vendor_id.to_i
+		end
+
+		def self.all
+			@@product_array ||= []
+			if @@product_array == []
+				CSV.read('./support/products.csv').each do |row|
+        	product = FarMar::Product.new(row[0], row[1], row[2])
+        	@@product_array.push(product)
+				end
+			end
+			return @@product_array
+		end
+
+		def self.find(id)
+			FarMar::Product.all.find {|pro| pro.product_id == id}
+		end
+
+		def vendor
+			FarMar::Vendor.all.find{|ven| ven.vendor_id == @vendor_id}
+		end
+
+		def sales
+			sales_hash = FarMar::Sale.sales_by_product
+			return sales_hash[@product_id]
+		end
+
+		def number_of_sales
+			self.sales.length
+		end
+
+		def self.by_vendor(vendor_id)
+			FarMar::Product.all.find_all{|prod| prod.vendor_id == vendor_id}
+		end
 
 	end
 
-end
+end

lib/far_mar/sale.rb

@@ -1,8 +1,73 @@
-# require "./support/sales.csv"
+# CSV.read("./support/sales.csv")
 
 module FarMar
 
 	class Sale
+		attr_reader :sale_id, :amount, :purchase_time, :vendor_id, :product_id
+
+		def initialize(sale_id, amount, purchase_time, vendor_id, product_id)
+			@sale_id = sale_id.to_i
+			@amount = amount.to_i
+			purchase_time = purchase_time.to_s
+			@purchase_time = DateTime.strptime(purchase_time, "%Y-%m-%d %H:%M:%S %z")
+			@vendor_id = vendor_id.to_i
+			@product_id = product_id.to_i
+		end
+
+		# def self.all
+		# 		@@sales_array ||= []
+		# 		if @@sales_array = []
+		# 		CSV.read('./support/sales.csv').each do |row|
+  #       	sale = FarMar::Sale.new(row[0], row[1], row[2], row[3], row[4])
+  #       	@@sales_array.push(sale)
+  #     	end
+  #     end
+		# 	return @@sales_array
+		# end
+
+		def self.sales_init
+			@@sales_by_id = Hash.new
+			@@sales_by_product = Hash.new {|hash, key| hash[key] = []}
+			@@sales_by_vendor = Hash.new {|hash, key| hash[key] = []}
+			CSV.read('./support/sales.csv').each do |row|
+				new_sale = FarMar::Sale.new(row[0], row[1], row[2], row[3], row[4])
+				@@sales_by_id[new_sale.sale_id] = new_sale
+				@@sales_by_product[new_sale.product_id].push(new_sale)
+				@@sales_by_vendor[new_sale.vendor_id].push(new_sale)
+			end
+			@@sales_all = sales_by_product.values.flatten
+		end
+
+		def self.sales_by_product
+			return @@sales_by_product
+		end
+
+		def self.sales_by_vendor
+			return @@sales_by_vendor
+		end
+
+		def self.all
+			return @@sales_all
+		end
+
+		def self.find(id)
+			#FarMar::Sale.all.find {|sale| sale.sale_id == id}
+			return @@sales_by_id[id]
+		end
+
+		def vendor
+			FarMar::Vendor.all.find {|ven| ven.vendor_id == @vendor_id}
+		end
+
+		def product
+			FarMar::Product.all.find {|pro| pro.product_id == @product_id}
+		end
+
+		def self.between(beginning_time, ending_time)
+			FarMar::Sale.all.find_all do |sale|
+				sale.purchase_time.between?(beginning_time, ending_time)
+			end
+		end
 
 	end
 

lib/far_mar/vendor.rb

@@ -1,8 +1,86 @@
-# require './suport/vendors.csv'
+# CSV.read('./support/vendors.csv')
 
 module FarMar
 
 	class Vendor
+		attr_reader :vendor_id, :name, :employees, :market
+
+		def initialize(vendor_id, name, employees, market)
+			@vendor_id = vendor_id.to_i
+			@name = name
+			@employees = employees
+			@market = market.to_i
+		end
+
+		def self.all
+			@@vendor_array ||= []
+			  if @@vendor_array == []
+			  	CSV.read('./support/vendors.csv').each do |row|
+        		vendor = FarMar::Vendor.new(row[0], row[1], row[2], row[3])
+        		@@vendor_array.push(vendor)
+      		end
+      	end
+      return @@vendor_array
+		end
+
+		def self.find(vendor_id)
+ 			FarMar::Vendor.all.find {|ven| ven.vendor_id == vendor_id}
+		end
+
+		def products
+			FarMar::Product.all.find_all {|pro| pro.vendor_id == @vendor_id}
+		end
+
+		def sales
+			sales_hash = FarMar::Sale.sales_by_vendor
+			return sales_hash[@vendor_id]
+		end
+
+		def revenue
+			sales_array = self.sales
+			total = 0
+			sales_array.each do |sale|
+			#	binding.pry
+				total += sale.amount
+			end
+			return total
+		end
+
+		def self.by_market(market_id)
+			FarMar::Vendor.all.find_all {|ven| ven.market == market_id }
+		end
+
+		# Couldn't figure out how to rewrite this to get it to run in a decent amount of time
+		# def self.most_revenue(n)
+		# 	top_vendors = @@vendor_array.sort_by{|ven| ven.revenue}
+		# 	return top_vendors[0..n]
+		# end
+
+		def self.most_items(n)
+			most_items = 0
+			vendors_by_items = []
+			vendor_array = @@vendor_array.dup
+			n.times do
+				vendor = vendor_array.max_by{|ven| ven.products.length}
+				vendors_by_items.push(vendor)
+				vendor_array.delete(vendor)
+			end
+			return vendors_by_items
+		end
+
+		# This method is currently taking too long to run; I can't even get the tests to finish on it. Not sure if I'll have the time to retool it
+		# def self.revenue(date)
+		# 	date = DateTime.strptime(date, "%Y-%m-%d").to_date
+		# 	revenue = 0
+		# 	FarMar::Vendor.all.each do |vendor_inst|
+		# 		vendor_inst.sales.each do |sale_inst|
+  #         if sale_inst.purchase_time.to_date == date
+  #           revenue += sale_inst.amount
+  #         end
+  #       end
+		# 	end
+		# 	return revenue
+		# end
 
 	end