stocktracker 0.0.3 → 0.0.4

data/lib/stocktracker.rb

@@ -2,9 +2,17 @@ require "stocktracker/version"
 require "stocktracker/yahoofinance"
 
 module StockTracker
+
+  def self.cache
+    @cache ||= Redis.new
+  end
+
   class CurrentQuote
     include YahooFinance
 
+    def cache
+      StockTracker.cache
+    end
 
     attr_accessor :symbol, :results
     def initialize(symbol)
@@ -15,25 +23,53 @@ module StockTracker
     private
 
       def yahoo_quote
-        YahooFinance::CurrentQuote.new(symbol).results
+        if cache.exists(cache_key)
+          Marshal.load(cache.get(cache_key))
+        else
+          results = YahooFinance::CurrentQuote.new(symbol).results
+          cache.set(cache_key, Marshal.dump(results))
+          cache.expire(cache_key, 600)
+          results
+        end
+      end
+
+      def cache_key
+        "StockTracker:current:#{symbol}"
       end
   end
 
   class PastQuote
     include YahooFinance
 
+    def cache
+      StockTracker.cache
+    end
+
     attr_accessor :symbol, :date, :results
     def initialize(symbol, date)
       self.symbol = symbol
       self.date = date
-      self.results = yahoo_historical_quote
+      self.results = yahoo_past_quote
     end
 
     private
 
-      def yahoo_historical_quote
-        YahooFinance::PastQuote.new(symbol, date).results
+
+      def yahoo_past_quote
+        if cache.exists(cache_key)
+          Marshal.load(cache.get(cache_key))
+        else
+          results = YahooFinance::PastQuote.new(symbol, date).results
+          cache.set(cache_key, Marshal.dump(results))
+          cache.expire(cache_key, 600)
+          results
+        end
       end
 
+      def cache_key
+        "StockTracker:past:#{symbol}:#{date.to_s}"
+      end
+
+
   end
 end

data/lib/stocktracker/version.rb

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

data/lib/stocktracker/yahoofinance.rb

@@ -38,7 +38,9 @@ module YahooFinance
     end
 
     def get_current_quote
-      current_map(yahoo_quote)
+      current_map
+    rescue
+      nil
     end
 
     def yahoo_quote
@@ -53,13 +55,13 @@ module YahooFinance
       CURRENT_FORMAT.keys.join
     end
 
-    def current_values(quote)
+    def current_values
       CSV.parse(yahoo_quote)[0]
     end
 
-    def current_map(quote)
+    def current_map
       map = {}
-      values = current_values(quote)
+      values = current_values
       CURRENT_FORMAT.each_with_index do |v,i|
         if v[1].is_a?(Array)
           map[v[1][0]] = values[i].send(v[1][1])
@@ -87,16 +89,18 @@ module YahooFinance
 
     def initialize(symbol, date)
       self.symbol= symbol
-      self.start_date= Date.parse(date)
-      self.end_date= Date.parse(date)
+      self.start_date= date
+      self.end_date= date
       self.results= get_past_quote
     end
 
     def get_past_quote
-      past_map(yahoo_quote)
+      past_map
+    rescue
+      nil
     end
 
-    def yahoo_quote
+    def yahoo_past_quote
       open(yahoo_past_url).read
     end
 
@@ -112,13 +116,13 @@ module YahooFinance
     end
 
 
-    def past_values(quote)
-      CSV.parse(yahoo_quote)[1]
+    def past_values
+      CSV.parse(yahoo_past_quote)[1]
     end
 
-    def past_map(quote)
+    def past_map
       map = {}
-      values = past_values(quote)
+      values = past_values
       PAST_FORMAT.each_with_index do |v,i|
         if v.is_a?(Array)
           map[v[0]] = values[i].send(v[1])

data/stocktracker.gemspec

@@ -8,8 +8,8 @@ Gem::Specification.new do |s|
   s.authors     = ["Aaron Worsham"]
   s.email       = ["aaronworsham@gmail.com"]
   s.homepage    = "https://github.com/aaronworsham/stocktracker"
-  s.summary     = %q{Returns Yahoo Finance Quotes for stock symbols}
-  s.description = %q{Returns Yahoo Finance Quotes for stock symbols}
+  s.summary     = %q{Returns Yahoo Finance Quotes for stock symbols.}
+  s.description = %q{Queries Yahoo Finance for Quotes from current and past.  Uses Redis to cache results for period of time}
 
   s.rubyforge_project = "stocktracker"
 
@@ -21,4 +21,5 @@ 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 'redis'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: stocktracker
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
   prerelease: 
 platform: ruby
 authors:
@@ -9,11 +9,11 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2011-10-16 00:00:00.000000000Z
+date: 2011-10-17 00:00:00.000000000Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake
-  requirement: &2166372700 !ruby/object:Gem::Requirement
+  requirement: &2153516340 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -21,10 +21,10 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *2166372700
+  version_requirements: *2153516340
 - !ruby/object:Gem::Dependency
   name: rspec
-  requirement: &2166372280 !ruby/object:Gem::Requirement
+  requirement: &2153515680 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -32,8 +32,20 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *2166372280
-description: Returns Yahoo Finance Quotes for stock symbols
+  version_requirements: *2153515680
+- !ruby/object:Gem::Dependency
+  name: redis
+  requirement: &2153515040 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *2153515040
+description: Queries Yahoo Finance for Quotes from current and past.  Uses Redis to
+  cache results for period of time
 email:
 - aaronworsham@gmail.com
 executables: []
@@ -73,7 +85,7 @@ rubyforge_project: stocktracker
 rubygems_version: 1.8.10
 signing_key: 
 specification_version: 3
-summary: Returns Yahoo Finance Quotes for stock symbols
+summary: Returns Yahoo Finance Quotes for stock symbols.
 test_files:
 - spec/spec_helper.rb
 - spec/stocktracker_spec.rb