Rlh/master

lib/far_mar.rb

@@ -2,6 +2,7 @@
 require "time"
 require "rspec"
 require "csv"
+require "date"
 require "./lib/far_mar/market.rb"
 require "./lib/far_mar/vendor.rb"
 require "./lib/far_mar/product.rb"

lib/far_mar/market.rb

@@ -1,15 +1,157 @@
-CSV.read("./support/markets.csv")
 
 module FarMar
   class Market
-    CSV.read("./support/markets.csv")
-    # @id =
-    # @name =
-    # @address =
-    # @city =
-    # @county =
-    # @state =
-    # @zip =
+
+    attr_reader :id, :name
+
+    def initialize(id, name, address, city, county, state, zip)
+      @id = id
+      @name = name
+      @address = address
+      @city = city
+      @county = county
+      @state = state
+      @zip = zip
+    end
+
+    def self.all
+      @@all_markets ||= []
+
+      if @@all_markets == []
+        CSV.read("./support/markets.csv").each do |line|
+          y = FarMar::Market.new(line[0].to_i, line[1], line[2], line[3], line[4], line[5], line[6])
+          @@all_markets.push(y)
+        end
+      end
+
+      return @@all_markets
+    end
+
+    def self.find(id)
+      FarMar::Market.all.find do |market_instance|
+        market_instance.id == id
+      end
+    end
+
+    def vendors
+      FarMar::Vendor.all.find_all do |vendor_instance|
+        vendor_instance.market_id == @id
+      end
+    end
+
+    def products
+      # empty array to fill with collection of products
+      all_products = []
+
+      # get a collection of all vendors
+      all_vendors = vendors
+
+      # to each vendor instance, run the products method to get a list of products,
+      # then push all of those proudcts into an array
+      all_vendors.each do |vendor_instance|
+        all_products.push(vendor_instance.products)
+      end
+
+      # need to flatten all products so it doesn't return an array of arrays
+      return all_products.flatten!
+    end
+
+    def self.search(search_term)
+      search_term_instances = []
+
+      markets = FarMar::Market.all
+
+      markets.each do |mkt_instance|
+        if mkt_instance.name.match(/#{search_term}/i)
+          search_term_instances.push(mkt_instance)
+        end
+
+        mkt_vendors = mkt_instance.vendors
+
+        mkt_vendors.each do |vendor_instance|
+          if vendor_instance.name.match(/#{search_term}/i)
+            search_term_instances.push(mkt_instance)
+          end
+        end
+      end
+      return search_term_instances
+    end
+
+    def prefered_vendor
+      highest_revenue = 0
+      ven_highest_revenue = nil
+
+      vendors.each do |vendor_instance|
+        ven_revenue = vendor_instance.revenue
+        if ven_revenue > highest_revenue
+          highest_revenue = ven_revenue
+          ven_highest_revenue = vendor_instance
+        end
+      end
+      return ven_highest_revenue
+    end
+
+    def prefered_vendor_on(date)
+      date = DateTime.strptime(date, "%Y-%m-%d").to_date
+      highest_revenue = 0
+      ven_highest_revenue = nil
+
+      vendors.each do |vendor_instance|
+        vendor_instance.sales.each do |sales_instance|
+          sales_on_date = []
+          revenue = 0
+          if sales_instance.purchase_time.to_date == date
+            sales_on_date.push(sales_instance.amount)
+          end
+          if sales_on_date.length > 0
+            sales_on_date.each {|amount| revenue += amount}
+            if revenue > highest_revenue
+              highest_revenue = revenue
+              ven_highest_revenue = vendor_instance
+            end
+          end
+        end
+      end
+      return ven_highest_revenue
+    end
+
+    def worst_vendor
+      lowest_revenue = Float::INFINITY
+      ven_lowest_revenue = nil
+
+      vendors.each do |vendor_instance|
+        ven_revenue = vendor_instance.revenue
+        if ven_revenue < lowest_revenue
+          lowest_revenue = ven_revenue
+          ven_lowest_revenue = vendor_instance
+        end
+      end
+      return ven_lowest_revenue
+    end
+
+    def worst_vendor_on(date)
+      date = Date.parse(date)
+      lowest_revenue = Float::INFINITY
+      ven_lowest_revenue = nil
+
+      vendors.each do |vendor_instance|
+        sales_on_date = []
+        revenue = 0
+        vendor_instance.sales.each do |sales_instance|
+          if sales_instance.purchase_time.to_date == date
+            sales_on_date.push(sales_instance.amount)
+          end
+        end
+        if sales_on_date.length > 0
+          sales_on_date.each {|amount| revenue += amount}
+          if revenue < lowest_revenue
+            lowest_revenue = revenue
+            ven_lowest_revenue = vendor_instance
+          end
+        end
+      end
+      return ven_lowest_revenue
+    end
 
   end
 end

lib/far_mar/product.rb

@@ -2,5 +2,63 @@
 
 module FarMar
   class Product
+    attr_reader :id, :name, :vendor_id
+
+    def initialize(id, name, vendor_id)
+      @id = id
+      @name = name
+      @vendor_id = vendor_id
+    end
+
+    def self.all
+      @@all_products ||= []
+
+      if @@all_products == []
+        CSV.read("./support/products.csv").each do |line|
+          y = FarMar::Product.new(line[0].to_i, line[1], line[2].to_i)
+          @@all_products.push(y)
+        end
+      end
+
+      return @@all_products
+    end
+
+    def self.find(id)
+      FarMar::Product.all.find do |product_instance|
+        product_instance.id == id
+      end
+    end
+
+    def self.by_vendor(vendor_id)
+      FarMar::Product.all.find_all do |product_instance|
+        product_instance.vendor_id == vendor_id
+      end
+    end
+
+    def vendor
+      FarMar::Vendor.all.find do |vendor_instance|
+        vendor_instance.id == @vendor_id
+      end
+    end
+
+    def sales
+      FarMar::Sale.all.find_all do |sales_instance|
+        sales_instance.product_id == @id
+      end
+    end
+
+    def number_of_sales
+      self.sales.length
+    end
+
+    # def self.most_revenue(n)
+    #   most_revenue = []
+    #
+    #   all.each do |product_instance|
+    #       product_instance
+    #   end
+    #
+    #
+    # end
   end
 end

lib/far_mar/sale.rb

@@ -1,6 +1,54 @@
-CSV.read("./support/sales.csv")
 
 module FarMar
   class Sale
+    attr_reader :id, :amount, :purchase_time, :vendor_id, :amount, :product_id
+
+    def initialize(id, amount, purchase_time, vendor_id, product_id)
+      @id = id
+      @amount = amount
+      @purchase_time = DateTime.strptime(purchase_time, "%Y-%m-%d %H:%M:%S %z")
+      @vendor_id = vendor_id
+      @product_id = product_id      
+    end
+
+    def self.all
+      @@all_sales ||= []
+
+        if @@all_sales == []
+          CSV.read("./support/sales.csv").each do |line|
+            y = FarMar::Sale.new(line[0].to_i, line[1].to_i, line[2], line[3].to_i, line[4].to_i)
+            @@all_sales.push(y)
+          end
+        end
+
+      return @@all_sales
+    end
+
+    def self.find(id)
+      FarMar::Sale.all.find do |sale_instance|
+        sale_instance.id == id
+      end
+    end
+
+    def self.between(beginning_time, end_time)
+      beginning_time = DateTime.strptime(beginning_time, "%Y-%m-%d %H:%M:%S %z")
+      end_time = DateTime.strptime(end_time, "%Y-%m-%d %H:%M:%S %z")
+
+      FarMar::Sale.all.find_all do |sales_instance|
+        sales_instance.purchase_time >= beginning_time && sales_instance.purchase_time <= end_time
+      end
+    end
+
+    def vendor
+      FarMar::Vendor.all.find do |vendor_instance|
+        vendor_instance.id == @vendor_id
+      end
+    end
+
+    def product
+      FarMar::Product.all.find do |product_instance|
+        product_instance.id == @product_id
+      end
+    end
   end
 end