diff options
author | Harald Eilertsen <haraldei@anduin.net> | 2014-11-13 17:54:55 +0100 |
---|---|---|
committer | Harald Eilertsen <haraldei@anduin.net> | 2014-11-13 17:54:55 +0100 |
commit | 33b2c7b846962bc92b7ea32049ab2e0502cd4d80 (patch) | |
tree | 6d7764253fc499cb1db63393bd6cb2395ae0f06b | |
parent | 91387893047f9b02cb0e9741f4843161e2e78179 (diff) | |
download | imusician-reports-33b2c7b846962bc92b7ea32049ab2e0502cd4d80.tar.gz imusician-reports-33b2c7b846962bc92b7ea32049ab2e0502cd4d80.tar.bz2 imusician-reports-33b2c7b846962bc92b7ea32049ab2e0502cd4d80.zip |
Make Sale class.
-rw-r--r-- | lib/sale.rb | 25 | ||||
-rw-r--r-- | parse_report.rb | 26 |
2 files changed, 40 insertions, 11 deletions
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 |