timestream 0.0.2 → 0.0.3

data/bin/timestream

@@ -1,42 +1,8 @@
 #!/usr/bin/env ruby
 
 require 'rubygems'
-require 'spreadsheet'
+require 'timestream/employee_activity_report'
 
-# TODO Add selector for spreadsheets
-spreadsheet = Spreadsheet.open ARGV[0] 
-worksheet = spreadsheet.worksheet "EAR - Current Week"
+employee_activity_report = Timestream::EmployeeActivityReport.new ARGV[0]
 
-DATE_ROW = 2
-DATE_COLUMN = 1
-
-saturday_date = worksheet.row(DATE_ROW)[DATE_COLUMN]
-project_totals_by_day = Hash.new { |hash, day|
-  hash[day] = Hash.new(0)
-}
-
-DAY_COLUMN_SPAN = 3
-PROJECT_NUMBER_COLUMN = 0
-
-# 0-indexed - rows are for this week's time
-(12 .. 36).each do |row_index|
-  row = worksheet.row(row_index)
-  project_number = row[PROJECT_NUMBER_COLUMN]
-  
-  # TODO handle Vacation / Holiday / Sick time
-  # no project number assigned
-  next if project_number.nil?
-  
-  # 0 - Weekend
-  # 1 - Monday
-  # 2 - Tuesday...
-  (0 .. 5).each do |day_index|
-    day_column = 9 + DAY_COLUMN_SPAN * day_index
-    hours = row[day_column] || 0
-    
-    total_by_project = project_totals_by_day[day_index]
-    total_by_project[project_number] += hours 
-  end
-end
-
-puts project_totals_by_day.inspect
+puts employee_activity_report.to_text_table

data/lib/timestream/employee_activity_report.rb

@@ -0,0 +1,136 @@
+require 'rubygems'
+require 'spreadsheet'
+require 'terminal-table/import'
+
+module Timestream
+  class EmployeeActivityReport
+    class Cell
+      attr_reader :column
+      
+      def initialize(row, column)
+        @column = column
+        @row = row
+      end
+      
+      attr_reader :row
+    end
+    
+    def cell(location)
+      worksheet.row(location.row)[location.column]
+    end
+    
+    DATE_CELL = Cell.new(2, 1)
+    
+    DAY_COLUMN_SPAN = 3
+    
+    def initialize(path)
+      @path = path
+    end
+    
+    attr_reader :path
+    
+    def row(index)
+      worksheet.row(index)
+    end
+    
+    def spreadsheet
+      @spreadsheet ||= Spreadsheet.open path
+    end
+    
+    class Timestream
+      include Enumerable
+      
+      SLOT_NAMES = [:weekend, :monday, :tuesday, :wednesday, :thursday, :friday]
+      
+      def each(&block)
+        @totals.each(&block)
+      end
+      
+      def initialize
+        @totals = Array.new(SLOT_NAMES.length, 0)
+      end
+      
+      def [](day)
+        if day.is_a? Symbol
+          day = SLOT_NAMES.index(day)
+        end
+        
+        @totals[day]
+      end
+      
+      def []=(day, total)
+        if day.is_a? Symbol
+          day = SLOT_NAMES.index(day)
+        end
+        
+        @totals[day] = total
+      end
+      
+      def to_a
+        @totals.dup
+      end
+    end
+    
+    def timestream_by_project_number
+      if @timestream_by_project_number.nil?
+        parse
+      end
+      
+      @timestream_by_project_number
+    end
+    
+    def to_text_table
+      employee_activity_report = self
+      table {
+        self.headings = ['Project Number']
+        self.headings.concat Timestream::SLOT_NAMES.collect { |name|
+          name.to_s.capitalize
+        }
+        
+        employee_activity_report.timestream_by_project_number.each do |project_number, timestream|
+          row = [project_number]
+          timestream.each do |daily_total|
+            row << "%02.02f" % daily_total
+          end
+          
+          add_row row
+        end 
+      }
+    end
+    
+    def worksheet
+      @worksheet ||= spreadsheet.worksheet "EAR - Current Week"
+    end
+    
+    private
+    
+    def parse
+      @timestream_by_project_number = Hash.new { |hash, project_number|
+        hash[project_number] = Timestream.new
+      }
+      
+      (12 .. 36).each do |row_index|
+        row = self.row(row_index)
+        project_number = row[PROJECT_NUMBER_COLUMN]
+        
+        # TODO handle Vacation / Holiday / Sick time
+        # no project number assigned
+        next if project_number.nil?
+        
+        # 0 - Weekend
+        # 1 - Monday
+        # 2 - Tuesday...
+        (0 .. 5).each do |day_index|
+          timestream = @timestream_by_project_number[project_number]
+          
+          day_column = 9 + DAY_COLUMN_SPAN * day_index
+          hours = row[day_column] || 0
+          
+          timestream[day_index] += hours
+        end
+      end
+    end
+    
+    PROJECT_NUMBER_COLUMN = 0
+  end
+end

data/lib/timestream/version.rb

@@ -1,3 +1,3 @@
 module Timestream
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

data/timestream.gemspec

@@ -15,6 +15,7 @@ Gem::Specification.new do |s|
   s.rubyforge_project = "timestream"
   
   s.add_dependency("spreadsheet")
+  s.add_dependency("terminal-table")
 
   s.files         = `git ls-files`.split("\n")
   s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: timestream
 version: !ruby/object:Gem::Version 
-  hash: 27
+  hash: 25
   prerelease: 
   segments: 
   - 0
   - 0
-  - 2
-  version: 0.0.2
+  - 3
+  version: 0.0.3
 platform: ruby
 authors: 
 - Luke Imhoff
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-05-11 00:00:00 -05:00
+date: 2011-05-12 00:00:00 -05:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -32,6 +32,20 @@ dependencies:
         version: "0"
   type: :runtime
   version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: terminal-table
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id002
 description: Converts Hamster report to EAR to JDE Timecard
 email: 
 - luke@cray.com
@@ -47,6 +61,7 @@ files:
 - Rakefile
 - bin/timestream
 - lib/timestream.rb
+- lib/timestream/employee_activity_report.rb
 - lib/timestream/version.rb
 - timestream.gemspec
 has_rdoc: true