FarMar with part 1 of optionals

lib/far_mar/market.rb

@@ -1,6 +1,125 @@
 module FarMar
   class Market
+    attr_reader :id, :name
 
+    def initialize(market_info)
+      @id = market_info[0].to_i
+      @name = market_info[1]
+      @address = market_info[2]
+      @city = market_info[3]
+      @county = market_info[4]
+      @state = market_info[5]
+      @zip = market_info[6]
+    end
+
+    def self.all
+      @@market_array ||=  []
+
+      if @@market_array == []
+        CSV.read("./support/markets.csv").each do |market|
+          new_market = FarMar::Market.new(market)
+          @@market_array.push(new_market)
+        end
+      end
+
+      return @@market_array
+    end
+
+    def self.find(id)
+      FarMar::Market.all.find do |market|
+        market.id == id
+      end
+    end
+
+    def vendors
+      vendors = FarMar::Vendor.all.find_all do |vendor|
+        vendor.market_id == self.id
+      end
+      return vendors
+    end
+
+    def products
+      products = []
+      vendors = self.vendors
+      vendors.each do |vendor|
+        vendor_products = vendor.products
+        vendor_products.each do |vendor_product|
+          products.push(vendor_product)
+        end
+      end
+      return products
+    end
+
+    def self.search(search_term)
+      results = []
+      name_matches = FarMar::Market.all.find_all do |market|
+        market.name.downcase.include?(search_term.downcase)
+      end
+      name_matches.each { |match| results.push(match)}
+      vendor_matches = FarMar::Vendor.all.find_all do |vendor|
+        vendor.name.downcase.include?(search_term.downcase)
+      end
+      vendor_matches.each { |match| results.push(match)}
+      return results
+    end
+
+    def preferred_vendor
+      vendors = self.vendors
+      best_vendor = vendors.max_by do |vendor|
+        vendor.revenue
+      end
+      return best_vendor
+    end
+
+    def preferred_vendor_by_date(date)
+      # expecting a string in the form yyyy-mm-dd
+      date = Date.parse(date.to_s)
+
+      vendors = self.vendors
+      vendor_hash = {}
+      vendors.each do |vendor|
+        vendor_sales = vendor.sales
+        day_sales = vendor_sales.find_all do |sale|
+          date == sale.purchase_time.to_date
+        end
+        day_revenue = 0
+        day_sales.each do |sale|
+          day_revenue = day_revenue + sale.amount
+        end
+        vendor_hash[vendor.id] = day_revenue
+      end
+      best_vendor_id = vendor_hash.key(vendor_hash.values.max)
+      return FarMar::Vendor.find(best_vendor_id)
+    end
+
+    def worst_vendor
+      vendors = self.vendors
+      worst_vendor = vendors.min_by do |vendor|
+        vendor.revenue
+      end
+      return worst_vendor
+    end
+
+    def worst_vendor_by_date(date)
+      # expecting a string in the form yyyy-mm-dd
+      date = Date.parse(date.to_s)
+
+      vendors = self.vendors
+      vendor_hash = {}
+      vendors.each do |vendor|
+        vendor_sales = vendor.sales
+        day_sales = vendor_sales.find_all do |sale|
+          date == sale.purchase_time.to_date
+        end
+        day_revenue = 0
+        day_sales.each do |sale|
+          day_revenue = day_revenue + sale.amount
+        end
+        vendor_hash[vendor.id] = day_revenue
+      end
+      worst_vendor_id = vendor_hash.key(vendor_hash.values.min)
+      return FarMar::Vendor.find(worst_vendor_id)
+    end
 
   end
 end

lib/far_mar/product.rb

@@ -1,6 +1,75 @@
 module FarMar
   class Product
+    attr_reader :id, :vendor_id
 
+    def initialize(product_info)
+      @id = product_info[0].to_i
+      @name = product_info[1]
+      @vendor_id = product_info[2].to_i
+    end
+
+    def self.all
+      @@product_array ||= []
+
+      if @@product_array == []
+        CSV.read("./support/products.csv").each do |product|
+          new_product = FarMar::Product.new(product)
+          @@product_array.push(new_product)
+        end
+      end
+
+      return @@product_array
+    end
+
+    def self.find(id)
+      FarMar::Product.all.find do |product|
+        product.id == id
+      end
+    end
+
+    def vendor
+      FarMar::Vendor.all.find do |vendor|
+        vendor.id == self.vendor_id
+      end
+    end
+
+    def sales
+      sales_hash = FarMar::Sale.sales_by_product
+      return sales_hash[@id]
+    end
+
+    def product_revenue
+      product_revenue = 0
+      sales = self.sales
+      sales.each do |sale|
+        product_revenue = product_revenue + sale.amount
+      end
+      return product_revenue
+    end
+
+    def number_of_sales
+      return self.sales.length
+    end
+
+    def self.by_vendor(vendor_id)
+      vendor_products = FarMar::Product.all.find_all do |product|
+        product.id == vendor_id
+      end
+      return vendor_products
+    end
+
+    def self.most_revenue(n)
+      sorted_products = FarMar::Product.all.sort_by do |product|
+        product_revenue = 0
+        product_sales = product.sales
+        product_sales.each do |sale|
+          product_revenue = product_revenue + sale.amount
+        end
+        product_revenue
+      end
+      sorted_products.reverse!
+      return sorted_products[0..n-1]
+    end
 
   end
 end

lib/far_mar/sale.rb

@@ -1,6 +1,73 @@
 module FarMar
   class Sale
+    attr_reader :id, :vendor_id, :product_id, :amount, :purchase_time
 
+    def initialize(sale_info)
+      @id = sale_info[0].to_i
+      @amount = sale_info[1].to_i
+      @purchase_time = DateTime.strptime(sale_info[2], "%Y-%m-%d %H:%M:%S %z")
+      @vendor_id = sale_info[3].to_i
+      @product_id = sale_info[4].to_i
+    end
+
+    def self.all
+      return sales_by_product.values.flatten
+    end
+
+    def self.sales_by_product
+      @@sales_by_product ||= Hash.new { |hash,key| hash[key] = [] }
+
+      if @@sales_by_product.empty?
+        CSV.read("./support/sales.csv").each do |sale|
+          new_sale = FarMar::Sale.new(sale)
+          @@sales_by_product[new_sale.product_id].push(new_sale)
+        end
+      end
+      return @@sales_by_product
+    end
+
+    def self.sales_by_vendor
+      @@sales_by_vendor ||= Hash.new { |hash,key| hash[key] = [] }
+
+      if @@sales_by_vendor.empty?
+        CSV.read("./support/sales.csv").each do |sale|
+          new_sale = FarMar::Sale.new(sale)
+          @@sales_by_vendor[new_sale.vendor_id].push(new_sale)
+        end
+      end
+      return @@sales_by_vendor
+    end
+
+    def self.find(id)
+      FarMar::Sale.all.find do |sale|
+        sale.id == id
+      end
+    end
+
+    def vendor
+      FarMar::Vendor.all.find do |vendor|
+        vendor.id == self.vendor_id
+      end
+    end
+
+    def product
+      FarMar::Product.all.find do |product|
+        product.id == self.product_id
+      end
+    end
+
+    def self.between(beginning_time, end_time)
+      if beginning_time.class == String
+        beginning_time = DateTime.strptime(beginning_time, "%Y-%m-%d %H:%M:%S %z")
+      end
+      if end_time.class == String
+        end_time = DateTime.strptime(end_time, "%Y-%m-%d %H:%M:%S %z")
+      end
+      sales_between = FarMar::Sale.all.find_all do |sale|
+        sale.purchase_time.between?(beginning_time,end_time)
+      end
+      return sales_between
+    end
 
   end
 end

lib/far_mar/vendor.rb

@@ -1,6 +1,105 @@
 module FarMar
   class Vendor
+    attr_reader :id, :market_id, :name
 
+    def initialize(vendor_info)
+      @id = vendor_info[0].to_i
+      @name = vendor_info[1]
+      @number_of_employees = vendor_info[2].to_i
+      @market_id = vendor_info[3].to_i
+    end
+
+    def self.all
+      @@vendor_array ||= []
+
+      if @@vendor_array == []
+        CSV.read("./support/vendors.csv").each do |vendor|
+          new_vendor = FarMar::Vendor.new(vendor)
+          @@vendor_array.push(new_vendor)
+        end
+      end
+
+      return @@vendor_array
+    end
+
+    def self.find(id)
+      FarMar::Vendor.all.find do |vendor|
+        vendor.id == id
+      end
+    end
+
+    def market
+      FarMar::Market.all.find do |market|
+        market.id == self.market_id
+      end
+    end
+
+    def products
+      products = FarMar::Product.all.find_all do |product|
+        product.vendor_id == self.id
+      end
+      return products
+    end
+
+    def sales
+      sales_hash = FarMar::Sale.sales_by_vendor
+      return sales_hash[@id]
+    end
+
+    def revenue
+      revenue = 0
+      sales_array = self.sales
+      sales_array.each do |sale|
+        revenue = revenue + sale.amount
+      end
+      return revenue
+    end
+
+    def self.by_market(market_id)
+      market_vendors = FarMar::Vendor.all.find_all do |vendor|
+        vendor.id == market_id
+      end
+      return market_vendors
+    end
+
+    def self.most_revenue(n)
+      sorted_vendors = FarMar::Vendor.all.sort_by do |vendor|
+        vendor.revenue
+      end
+      sorted_vendors.reverse!
+      return sorted_vendors[0..n-1]
+    end
+
+    def self.most_items(n)
+      sorted_vendors = FarMar::Vendor.all.sort_by do |vendor|
+        vendor.products.length
+      end
+      sorted_vendors.reverse!
+      return sorted_vendors[0..n-1]
+    end
+
+    def self.revenue_by_date(date)
+      date = Date.parse(date.to_s)
+      total_revenue = 0
+      FarMar::Sale.all.each do |sale|
+        if sale.purchase_time.to_date == date
+          total_revenue = total_revenue + sale.amount
+        end
+      end
+      return total_revenue
+    end
+
+    def vendor_revenue_by_date(date)
+      date = Date.parse(date.to_s)
+      day_sales = self.sales.find_all do |sale|
+        date == sale.purchase_time.to_date
+      end
+      day_revenue = 0
+      day_sales.each do |sale|
+        day_revenue = day_revenue + sale.amount
+      end
+      return day_revenue
+    end
 
   end
 end