From 33b2c7b846962bc92b7ea32049ab2e0502cd4d80 Mon Sep 17 00:00:00 2001
From: Harald Eilertsen <haraldei@anduin.net>
Date: Thu, 13 Nov 2014 17:54:55 +0100
Subject: Make Sale class.

---
 lib/sale.rb     | 25 +++++++++++++++++++++++++
 parse_report.rb | 26 +++++++++++++++-----------
 2 files changed, 40 insertions(+), 11 deletions(-)
 create mode 100644 lib/sale.rb

diff --git a/lib/sale.rb b/lib/sale.rb
new file mode 100644
index 0000000..f02a471
--- /dev/null
+++ b/lib/sale.rb
@@ -0,0 +1,25 @@
+module SalesReporter
+  class Sale
+    attr_reader :date, :amount, :quantity
+
+    def initialize(date_, amount, quantity)
+      @date = sanitize_date(date_)
+      raise TypeError('sanitize failed') unless @date.is_a? Date
+      @amount = amount
+      @quantity = quantity
+    end
+
+    private
+
+    def sanitize_date(d)
+      case d.class.to_s
+      when "String"
+        return Date.parse(d)
+      when "Date"
+        return d
+      else
+        raise TypeError.new('Expected date argument to be of type Date or String')
+      end
+    end
+  end
+end
diff --git a/parse_report.rb b/parse_report.rb
index aac426b..efaa231 100644
--- a/parse_report.rb
+++ b/parse_report.rb
@@ -1,4 +1,5 @@
-require "csv"
+require 'sale'
+require 'csv'
 
 csv_opts = {
   :col_sep => ";",
@@ -7,19 +8,22 @@ csv_opts = {
   :converters => :all,
 }
 
+sales = []
 sales_by_date = {}
 
-rows = CSV.foreach(ARGV[0], csv_opts)
-
-rows.each do |row|
+CSV.foreach(ARGV[0], csv_opts) do |row|
   if row.count > 0
-    d = Date.parse(row["Date"])
-    if sales_by_date[d]
-      sales_by_date[d][:count] += row["quantity"]
-      sales_by_date[d][:total] += row["revenue EUR"]
-    else
-      sales_by_date[d] = {:count => row["quantity"], :total => row["revenue EUR"]}
-    end
+    sales << SalesReporter::Sale.new(row["Date"], row["revenue EUR"], row["quantity"])
+  end
+end
+
+sales.each do |s|
+  d = s.date
+  if sales_by_date[d]
+    sales_by_date[d][:count] += s.quantity
+    sales_by_date[d][:total] += s.amount
+  else
+    sales_by_date[d] = {:count => s.quantity, :total => s.amount}
   end
 end
 
-- 
cgit v1.2.3