Kld/master

lib/far_mar.rb

@@ -1,5 +1,6 @@
 require "csv"
 require "time"
+require "./lib/far_mar/parent"
 require "./lib/far_mar/market"
 require "./lib/far_mar/vendor"
 require "./lib/far_mar/product"

lib/far_mar/market.rb

@@ -1,4 +1,80 @@
+require "csv"
+require "pry"
+
 module FarMar
-  class Market
+  attr_accessor :id, :name, :address, :city, :county, :state, :zip
+  class Market < FarMar::FarMarParent
+    def self.filepath
+      "./support/markets.csv"
+    end
+
+    def self.create_hash(market_array)
+      market_hash = {}
+      market_hash[:id] = market_array[0].to_i
+      market_hash[:name] = market_array[1]
+      market_hash[:address] = market_array[2]
+      market_hash[:city] = market_array[3]
+      market_hash[:county] = market_array[4]
+      market_hash[:state] = market_array[5]
+      market_hash[:zip] = market_array[6].to_i
+      return market_hash
+    end
+
+    def find_vendors(market_id)
+      vendors_array = []
+      FarMar::Vendor.all.each do |vendor|
+        if market_id == vendor.market_id
+          vendors_array.push(vendor)
+        end
+      end
+        return vendors_array
+    end
+
+    def find_products(market_id)
+      products_array = []
+      vendors = find_vendors(market_id)
+      vendors.each do |vendor|
+        products = vendor.find_products(vendor.id)
+        products_array.push(products)
+      end
+      return products_array.flatten
+    end
+
+    def name
+      return @name
+    end
+
+    def self.search(search_term)
+      #need to sanitize input, needs to be a string only
+      results = []
+      market_list = self.all.find_all do |market|
+        market_name = market.name.downcase
+        market_name.include?(search_term.downcase)
+      end
+      vendor_list = FarMar::Vendor.all.find_all do |vendor|
+        vendor_name = vendor.name.downcase
+        vendor_name.include?(search_term.downcase)
+      end
+      vendor_list.map! do |vendor|
+        FarMar::Vendor.find_market(FarMar::Vendor.market_id)
+      end
+      results.push(market_list)
+      results.push(vendor_list)
+      return results.flatten
+    end
+
+    # def preferred_vendor
+    #   results = []
+    #   # this is an array of vendor objects
+    #   vendors = find_vendors(self.id)
+    #   # this calculates the revenue per vendor
+    #   vendors.each do |vendor|
+    #     moniez = vendor.revenue(vendor.id)
+    #     results.push([vendor, moniez])
+    #   end
+    #   # need to return highest revenue vendor
+    #   results.
+    # end
+
   end
 end

lib/far_mar/parent.rb

@@ -0,0 +1,32 @@
+require "csv"
+
+module FarMar
+  class FarMarParent
+    def initialize(init_hash)
+      # This block sets each key to an instance variable and assigns the value
+      init_hash.each do |k,v|
+        instance_variable_set("@#{k}", v) unless v.nil?
+      end
+    end
+
+    def self.all()
+      if !instance_variable_defined?("@all")
+        instance_variable_set("@all", CSV.read(self.filepath))
+        @all.map! do |line|
+          self.new(self.create_hash(line))
+        end
+      end
+      return @all
+    end
+
+
+    def self.find(id)
+      self.all.find {|object| object.id == id}
+    end
+
+    def id
+      return @id
+    end
+
+  end
+end

lib/far_mar/product.rb

@@ -1,4 +1,53 @@
 module FarMar
-  class Product
+  class Product < FarMar::FarMarParent
+    attr_accessor :id, :name, :vendor_id
+
+    def self.filepath
+      "./support/products.csv"
+    end
+
+      def self.create_hash(product_array)
+        product_hash = {}
+        product_hash[:id] = product_array[0].to_i
+        product_hash[:name] = product_array[1]
+        product_hash[:vendor_id] = product_array[2].to_i
+        return product_hash
+      end
+
+      def vendor_id
+        return @vendor_id
+      end
+
+      def name
+        return @name
+      end
+
+      def find_vendor(vendor_id)
+        FarMar::Vendor.find(vendor_id)
+      end
+
+      def find_sales(product_id)
+        sales_array = []
+        FarMar::Sale.all.each do |sale|
+          if product_id == sale.product_id
+            sales_array.push(sale)
+          end
+        end
+          return sales_array
+      end
+
+      def number_of_sales(product_id)
+        sales = find_sales(product_id)
+        total = 0
+        sales.each {|sale| total += 1}
+        return total
+      end
+
+      def self.by_vendor(vendor_id)
+        all_products = self.all
+        #all_products is an array of Vendor objects
+        matches = all_products.find_all {|product| product.vendor_id == vendor_id}
+        return matches
+      end
   end
 end

lib/far_mar/sale.rb

@@ -1,4 +1,41 @@
 module FarMar
-  class Sale
+  class Sale < FarMar::FarMarParent
+    attr_accessor :id, :amount, :purchase_time, :vendor_id, :product_id
+      def self.filepath
+        "./support/sales.csv"
+      end
+
+      def self.create_hash(sale_array)
+        sale_hash = {}
+        sale_hash[:id] = sale_array[0].to_i
+        sale_hash[:amount] = sale_array[1].to_i
+        sale_hash[:purchase_time] = DateTime.parse(sale_array[2])
+        sale_hash[:vendor_id] = sale_array[3].to_i
+        sale_hash[:product_id] = sale_array[4].to_i
+        return sale_hash
+      end
+
+      def amount
+        return @amount
+      end
+
+      def find_vendor(vendor_id)
+        FarMar::Vendor.find(vendor_id)
+      end
+
+      def find_product(product_id)
+        FarMar::Product.find(product_id)
+      end
+
+      def self.between(beginning_time, end_time)
+        # add input validation
+        matching_sales_array = []
+        self.all.find_all do |sale|
+          sale.purchase_time >= beginning_time && sale.purchase_time <= end_time
+          matching_sales_array.push(sale)
+        end
+        return matching_sales_array
+      end
+
   end
 end

lib/far_mar/vendor.rb

@@ -1,4 +1,67 @@
 module FarMar
-  class Vendor
+  class Vendor < FarMar::FarMarParent
+    attr_accessor :id, :name, :num_employees, :market_id
+      def self.filepath
+        "./support/vendors.csv"
+      end
+
+      def self.create_hash(vendor_array)
+        vendor_hash = {}
+        vendor_hash[:id] = vendor_array[0].to_i
+        vendor_hash[:name] = vendor_array[1]
+        vendor_hash[:num_employees] = vendor_array[2].to_i
+        vendor_hash[:market_id] = vendor_array[3].to_i
+        return vendor_hash
+      end
+
+      def market_id
+        return @market_id
+      end
+
+      def name
+        return @name
+      end
+
+      def find_market(market_id)
+        return FarMar::Market.find(market_id)
+      end
+
+      def find_products(vendor_id)
+        products_array = []
+        FarMar::Product.all.each do |product|
+          if vendor_id == product.vendor_id
+            products_array.push(product)
+          end
+        end
+          return products_array
+      end
+
+      def find_sales(vendor_id)
+        sales_array = []
+        FarMar::Sale.all.each do |sale|
+          if vendor_id == sale.vendor_id
+            sales_array.push(sale)
+          end
+        end
+          return sales_array
+      end
+
+      def revenue(vendor_id)
+        sales = find_sales(vendor_id)
+        #sales is an array of Sale objects
+        sum = 0
+        sales.each do |sale|
+          sum += sale.amount
+        end
+        return sum
+      end
+
+      def self.by_market(market_id)
+        all_vendors = self.all
+        #all_vendors is an array of Vendor objects
+        matches = all_vendors.find_all {|vendor| vendor.market_id == market_id}
+        return matches
+      end
+
   end
 end