harvest_wheelman 0.0.1 → 1.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 08f7c6b51b6bb944171bb219bb8076928933cd22
-  data.tar.gz: 9d3f26cbb92f6066e3b8b184d3720a2ea9a59a95
+  metadata.gz: c4024eb4f75db002fb29a9ef766f7dcdd596ab4a
+  data.tar.gz: 3e9193ec68b32f41ecd4cac141676b1f9a3d0eb0
 SHA512:
-  metadata.gz: 65bcb571eb81cfe6f22ee7a1a0327ee0c14a1e314c5de68d1e22eb817b6e9d79c3bf449b401082f6eec5207e0f52023428a959282d11cb6a7144da1fdb5736ba
-  data.tar.gz: 00fa3e3ba4eda7da8065a9c6c88c049c78f74f0dbfe354b487206fd158091991913995daaa09abdcf623828ff8623f2600b02909d0dc26394fad193fc487591e
+  metadata.gz: 93d61f4fcb80b44a049fe4560ddad053c32580f37d7eded0f187ca2ac3069540d38850bf84c9f6058dc67e85c4529c34f61fcb653914fee1daccc89ccb40788e
+  data.tar.gz: 7aa4e52c19d9f811e91e5514b97bd23237d292059728b7d6aea83922cb9d5901c69cfd7837d29d2b6822150f52d90d2aa5559f94be72fcc5eff472d77c7defef

data/.gitignore

@@ -15,3 +15,4 @@ spec/reports
 test/tmp
 test/version_tmp
 tmp
+vendor

data/README.md

@@ -4,43 +4,12 @@ I use this tool every 2 weeks to generate a PDF timesheet report from Harvest th
 
 ## Installation
 
-Add this line to your application's Gemfile:
-
-    gem 'harvest_wheelman'
-
-And then execute:
-
-    $ bundle
-
-Or install it yourself as:
-
     $ gem install harvest_wheelman
 
 ## Usage
 
 `harvest_wheelman`
 
-Optionally you can pass in a settings file like this:
-
-`harvest_wheelman ~/MyStuff/harvest_Wheelman.json` 
-
-## Example Settings File
-
-```json
-{
-  "harvest":{
-    "site":"---",
-    "user":"---",
-    "pass":"---"
-  },
-  "pay_period":{
-    "weeks":2,            # weeks in a pay period; used when only 1 date given (optional and defaults to 2)
-    "from":"09/03/2012",  # start of pay period (optional if 'to' exists)
-    "to":"09/16/2012"     # end of pay period (optional if 'from' exists)
-  }
-}
-```
-
 ## Contributing
 
 1. Fork it

data/bin/harvest_wheelman

@@ -1,35 +1,58 @@
 #!/usr/bin/env ruby
 require 'harvest_wheelman'
+require 'optparse'
 
-SETTINGS_FILE = File.expand_path(ENV['SETTINGS_FILE_PATH'] || ARGV[0] || 'settings.json')
+EXAMPLE = %{{
+  "harvest":{
+    "site":"---",
+    "user":"---",
+    "pass":"---"
+  }
+}}
+
+SETTINGS_HELP = %{#{EXAMPLE}
+
+Optionally include a "pay_period" block, otherwise we automatically use
+a 2 week pay period where "to" is set to the closest Sunday to today.
+
+{
+  "harvest": { ... },
+  "pay_period":{
+    "weeks":2, 
+    "from":"09/03/2012",
+    "to":"09/16/2012" 
+  }
+}}
+
+SETTINGS_FILE = File.join(File.expand_path('~'), '.harvest_wheelman')
+if !File.exists? SETTINGS_FILE
+  File.open(SETTINGS_FILE, 'w') do |f|
+    f.write EXAMPLE
+  end
+  puts "Wrote to #{SETTINGS_FILE}"
+  puts SETTINGS_HELP
+  exit(0)
+end
 
 begin
   hs = HarvestWheelman::HarvestSubmit.new(SETTINGS_FILE)
   puts %{
     Pay period is set from: #{hs.from}
                         to: #{hs.to}
-    Press enter to open firefox and generate the PDF using these dates.
+    Press enter to open chrome and generate the PDF using these dates.
     Ctrl-C to quit.
   }
   puts hs.drive_to_pdf if STDIN.gets
 rescue => ex
-  puts %{
-    Could not start, check your settings file.
-    Settings file: #{SETTINGS_FILE}
-    
-    Example:
-    {
-      "harvest":{
-        "site":"---",
-        "user":"---",
-        "pass":"---"
-      },
-      "pay_period":{
-        "weeks":2,            # weeks in a pay period (defaults to 2), used when only 1 date given
-        "from":"09/03/2012",  # start of pay period
-        "to":"09/16/2012"     # end of pay period
-      }
-    }
-  }
-  puts "There was a problem!\n#{ex.backtrace.join("\n")}\n\n#{ex.message}"
-end
+puts <<-EOF
+Could not start, check your settings file.
+Settings file path: #{SETTINGS_FILE}
+  
+Example:
+#{SETTINGS_HELP}
+
+Error: #{ex.message}
+------------------
+#{ex.backtrace.join("\n")} 
+EOF
+end

data/harvest_wheelman.gemspec

@@ -26,4 +26,4 @@ Gem::Specification.new do |spec|
   spec.add_runtime_dependency 'selenium-webdriver', '>=2.31.0'
   spec.add_runtime_dependency 'timerizer'
   spec.add_runtime_dependency 'json'
-end
+end

data/lib/harvest_submit.rb

@@ -6,7 +6,24 @@ require 'timerizer'
 # Puts +str+ into the pasteboard (Mac only)
 def pbcopy(str)
   return unless RUBY_PLATFORM =~ /darwin/
-  `echo "#{str}" | pbcopy`
+  `echo "#{str.strip}" | pbcopy`
+end
+
+class Date
+  def self.nearest_sunday
+    date = Date.today
+    wday = date.wday
+    if wday >= 4
+      wday.upto(6) do
+        date = 1.day.after(date)
+      end
+    elsif wday <= 3
+      wday.downto(1) do
+        date = 1.day.before(date)
+      end
+    end
+    return date
+  end
 end
 
 module HarvestWheelman
@@ -20,13 +37,17 @@ module HarvestWheelman
       File.open(settings_file) do |f|
         @settings = JSON.parse(f.read)
       end
+      if not @settings['pay_period']
+        @ppw = 2
+        @to = date_to_string Date.nearest_sunday
+      end
       determine_time_interval!
     end
 
     ##
     # Pay period weeks
     def ppw
-      @ppw ||= @settings['pay_period']['weeks'] || 2
+      @ppw ||= @settings['pay_period']['weeks'] rescue 2
     end
 
     def determine_time_interval!
@@ -37,7 +58,7 @@ module HarvestWheelman
           @to = date_to_string(1.day.before(ppw.weeks.after(string_to_date(from))))
         end
       elsif date_valid? to
-        puts "Setting FROM via the TO based on a ppw week pay period"
+        puts "Setting FROM via the TO based on a #{ppw} week pay period"
         @from = date_to_string(1.day.after(ppw.weeks.before(string_to_date(to))))
       else
         raise UnknownTimeIntervalError
@@ -62,11 +83,11 @@ module HarvestWheelman
     end
 
     def from
-      @from ||= @settings['pay_period']['from']
+      @from ||= @settings['pay_period']['from'] rescue nil
     end
 
     def to
-      @to ||= @settings['pay_period']['to']
+      @to ||= @settings['pay_period']['to'] rescue nil
     end
 
     def site
@@ -82,7 +103,7 @@ module HarvestWheelman
     end
 
     def whoami
-      `whoami`.strip
+      `whoami`.strip.capitalize
     end
 
     def pdf_name
@@ -94,50 +115,75 @@ module HarvestWheelman
     end
 
     def body
-      %{Hello,
-  Attached is my timesheet for pay period #{from} - #{to}
-  Thank you,
-  #{whoami}}
+      <<-EOF
+Hello,
+Attached is my timesheet for pay period #{from} - #{to}
+
+Thank you,
+#{whoami}
+
+Generated using http://rubygems.org/gems/harvest_wheelman
+Version #{HarvestWheelman::VERSION}
+      EOF
     end
 
     def drive_to_pdf
       begin
-        @driver = Selenium::WebDriver.for :firefox
+        puts "Starting chromedriver"
+        @driver = Selenium::WebDriver.for :chrome
         @base_url = "https://#{site}.harvestapp.com/"
+        puts "Driving to #{@base_url}"
         @driver.manage.timeouts.implicit_wait = 30
         @verification_errors = []
         @driver.get(@base_url + "/account/login")
+        puts "Signing in as #{user}"
         @driver.find_element(:id, "email").clear
         @driver.find_element(:id, "email").send_keys user
         @driver.find_element(:id, "user_password").clear
         @driver.find_element(:id, "user_password").send_keys pass
         @driver.find_element(:id, "sign-in-button").click
+        puts "Heading to the Reports area"
         @driver.find_element(:link, "Reports").click
         @driver.find_element(:css, "span.btn-dropdown").click
         @driver.find_element(:link, "Custom").click
 
+        puts "Entering custom dates"
         @driver.find_element(:id, "start_date").clear
         @driver.find_element(:id, "start_date").send_keys from
         @driver.find_element(:id, "end_date").clear
         @driver.find_element(:id, "end_date").send_keys to
 
+        puts "Requesting detailed report"
         @driver.find_element(:link, "Update Report").click
         @driver.find_element(:link, "Detailed Report").click
         @driver.find_elements(:tag_name => 'option').find{|i| i.text == 'Task'}.click
         
         puts "Print as PDF. Filename:\n#{pdf_name}"
-        pbcopy pdf_name
+        if pbcopy pdf_name
+          puts "Filename is currently in your pasteboard for file save dialog..."
+        end
         @driver.execute_script('window.print()')
         @driver.find_element(:xpath, "//nav[@id='nav']/div/ul/li[4]/a").click
         @driver.find_element(:link, "Sign Out").click
-        @driver.quit
         puts "Email your PDF."
-        puts "Subject:\n#{subject}"
-        puts "Body:\n#{body}"
-        pbcopy "#{subject}\n#{body}"
+        pbcopy subject
+        puts <<-EOF
+        -------------------------------
+#{subject}
+
+
+#{body}
+        -------------------------------
+        EOF
       rescue Exception => ex
-        "There was a problem!\n#{ex.backtrace}\n\n#{ex.message}\nExit!"
+puts <<-EOF
+Error: #{ex.message}
+------------------
+#{ex.backtrace.join("\n")} 
+EOF
+      ensure
+        @driver.quit
       end
     end
   end
-end
+end

data/lib/harvest_wheelman/version.rb

@@ -1,3 +1,3 @@
 module HarvestWheelman
-  VERSION = "0.0.1"
+  VERSION = "1.0.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: harvest_wheelman
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 1.0.0
 platform: ruby
 authors:
 - Keyvan Fatehi
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-05-20 00:00:00.000000000 Z
+date: 2014-08-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -133,7 +133,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.0.3
+rubygems_version: 2.0.14
 signing_key: 
 specification_version: 4
 summary: Logs into Harvest and generates a PDF report