stocktracker 0.0.2 → 0.0.3

data/README

@@ -0,0 +1,2 @@
+A simple Yahoo Finance wrapper to return current and historic stock quotes.
+

data/lib/stocktracker/version.rb

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

data/lib/stocktracker/yahoofinance.rb

@@ -1,11 +1,132 @@
+require 'open-uri'
+require 'csv'
+
+
+# modified version of YahooFinance gem by Nicholas Rahn <nick at transparentech.com>
+
 module YahooFinance
-  class StandardQuote
-    def values
-      ret = {}
-      @formathash.each_value do |val|
-        ret[val[0]] = send( val[0] ) unless send( val[0] ) == nil
+
+  class CurrentQuote
+
+    attr_accessor :symbol, :results
+
+    CURRENT_FORMAT = {
+      "s"   =>  :symbol,
+      "n"   =>  :name,
+      "l1"  =>  [ :last_trade, "to_f" ],
+      "d1"  =>  :date,
+      "t1"  =>  :time,
+      "c"   =>  :change,
+      "c1"  =>  [ :change_points, "to_f" ],
+      "p2"  =>  [ :change_percent, "to_f" ],
+      "p"   =>  [ :previous_close, "to_f" ],
+      "o"   =>  [ :open, "to_f" ],
+      "h"   =>  [ :day_high, "to_f" ],
+      "g"   =>  [ :day_low, "to_f" ],
+      "v"   =>  [ :volume, "to_i" ],
+      "m"   =>  :day_range,
+      "l"   =>  :last_trade_with_time,
+      "t7"  =>  :ticker_trend,
+      "a2"  =>  [ :average_daily_volume, "to_i" ],
+      "b"   =>  [ :bid, "to_f" ],
+      "a"   =>  [ :ask, "to_f" ]
+    }
+
+    def initialize(symbol)
+      self.symbol= symbol
+      self.results= get_current_quote
+    end
+
+    def get_current_quote
+      current_map(yahoo_quote)
+    end
+
+    def yahoo_quote
+      open(yahoo_current_url).read
+    end
+
+    def yahoo_current_url
+      "http://download.finance.yahoo.com/d/quotes.csv?s=#{self.symbol}&f=#{current_format}"
+    end
+
+    def current_format
+      CURRENT_FORMAT.keys.join
+    end
+
+    def current_values(quote)
+      CSV.parse(yahoo_quote)[0]
+    end
+
+    def current_map(quote)
+      map = {}
+      values = current_values(quote)
+      CURRENT_FORMAT.each_with_index do |v,i|
+        if v[1].is_a?(Array)
+          map[v[1][0]] = values[i].send(v[1][1])
+        else
+          map[v[1]] = values[i]
+        end
+      end
+      map
+    end
+  end
+
+  class PastQuote
+
+    attr_accessor :symbol, :results, :start_date, :end_date
+
+    PAST_FORMAT = [
+      :date,
+      [:open,"to_f"],
+      [:high,"to_f"],
+      [:low,"to_f"],
+      [:close,"to_f"],
+      [:volume,"to_f"],
+      [:adj_close,"to_f"]
+    ]
+
+    def initialize(symbol, date)
+      self.symbol= symbol
+      self.start_date= Date.parse(date)
+      self.end_date= Date.parse(date)
+      self.results= get_past_quote
+    end
+
+    def get_past_quote
+      past_map(yahoo_quote)
+    end
+
+    def yahoo_quote
+      open(yahoo_past_url).read
+    end
+
+    def yahoo_past_url
+      "http://itable.finance.yahoo.com/table.csv?"+
+      "s=#{symbol}&g=d&"+
+      "a=#{start_date.month-1}&"+
+      "b=#{start_date.mday}&"+
+      "c=#{start_date.year}&"+
+      "d=#{end_date.month-1}&"+
+      "e=#{end_date.mday}&"+
+      "f=#{end_date.year}"
+    end
+
+
+    def past_values(quote)
+      CSV.parse(yahoo_quote)[1]
+    end
+
+    def past_map(quote)
+      map = {}
+      values = past_values(quote)
+      PAST_FORMAT.each_with_index do |v,i|
+        if v.is_a?(Array)
+          map[v[0]] = values[i].send(v[1])
+        else
+          map[v] = values[i]
+        end
       end
-      ret
+      map
     end
   end
-end
+end

data/lib/stocktracker.rb

@@ -1,47 +1,38 @@
 require "stocktracker/version"
-require "yahoofinance"
 require "stocktracker/yahoofinance"
 
 module StockTracker
   class CurrentQuote
+    include YahooFinance
 
 
     attr_accessor :symbol, :results
     def initialize(symbol)
-      self.symbol = symbol
+      self.symbol = symbol.upcase
       self.results = yahoo_quote
     end
 
     private
 
       def yahoo_quote
-        YahooFinance::get_quotes(YahooFinance::StandardQuote, symbol)[symbol].values
+        YahooFinance::CurrentQuote.new(symbol).results
       end
   end
 
-  class HistoricalQuote
-    HISTORIC_ROWS = ["Date","Open","High","Low","Close","Volume","Adjusted Close"]
+  class PastQuote
+    include YahooFinance
 
-    attr_accessor :symbol, :start_date, :end_date, :results
-    def initialize(symbol, start_date, end_date)
+    attr_accessor :symbol, :date, :results
+    def initialize(symbol, date)
       self.symbol = symbol
-      self.start_date = start_date
-      self.end_date = end_date
+      self.date = date
       self.results = yahoo_historical_quote
     end
 
     private
 
       def yahoo_historical_quote
-        map_rows(YahooFinance::get_historical_quotes(symbol, start_date, end_date))
-      end
-
-      def map_rows(results)
-        map = {}
-        results[0].each_with_index do |r, i|
-          map[HISTORIC_ROWS[i]] = r.to_f
-        end
-        map
+        YahooFinance::PastQuote.new(symbol, date).results
       end
 
   end

data/spec/stocktracker_spec.rb

@@ -8,24 +8,24 @@ describe StockTracker  do
       end
       it "will return a current quote for Google" do
         @quote.results.should be
-        @quote.results["symbol"].should == "GOOG"
+        @quote.results[:symbol].should == "GOOG"
       end
 
       it 'will have a close greater than 0' do
-        @quote.results["lastTrade"].should > 0
+        @quote.results[:last_trade].should > 0
       end
     end
 
-    context :historical do
+    context :past do
       before(:each) do
-        @quote = StockTracker::HistoricalQuote.new("GOOG", Date.parse('01/10/2010'), Date.parse('01/10/2010'))
+        @quote = StockTracker::PastQuote.new("GOOG", '01/10/2010')
       end
       it "will return a current quote for Google" do
         @quote.results.should be
       end
 
       it 'will have a close greater than 0' do
-        @quote.results["Close"].should > 0
+        @quote.results[:close].should > 0
       end
     end
   end

data/stocktracker.gemspec

@@ -21,5 +21,4 @@ Gem::Specification.new do |s|
   # specify any dependencies here; for example:
   s.add_development_dependency 'rake'
   s.add_development_dependency 'rspec'
-  s.add_runtime_dependency "yahoofinance"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: stocktracker
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors:
@@ -13,7 +13,7 @@ date: 2011-10-16 00:00:00.000000000Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake
-  requirement: &2158448960 !ruby/object:Gem::Requirement
+  requirement: &2166372700 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -21,10 +21,10 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *2158448960
+  version_requirements: *2166372700
 - !ruby/object:Gem::Dependency
   name: rspec
-  requirement: &2158448540 !ruby/object:Gem::Requirement
+  requirement: &2166372280 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -32,18 +32,7 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *2158448540
-- !ruby/object:Gem::Dependency
-  name: yahoofinance
-  requirement: &2158448120 !ruby/object:Gem::Requirement
-    none: false
-    requirements:
-    - - ! '>='
-      - !ruby/object:Gem::Version
-        version: '0'
-  type: :runtime
-  prerelease: false
-  version_requirements: *2158448120
+  version_requirements: *2166372280
 description: Returns Yahoo Finance Quotes for stock symbols
 email:
 - aaronworsham@gmail.com
@@ -53,6 +42,7 @@ extra_rdoc_files: []
 files:
 - .gitignore
 - Gemfile
+- README
 - Rakefile
 - lib/stocktracker.rb
 - lib/stocktracker/version.rb