lib/sqa/stock.rb in sqa-0.0.12 vs lib/sqa/stock.rb in sqa-0.0.13

@@ -2,41 +2,53 @@
 
 class SQA::Stock
   attr_accessor :company_name
   attr_accessor :df             # The DataFrane
   attr_accessor :ticker
+  attr_accessor :type           # type of data store (default is CSV)
   attr_accessor :indicators
 
-  def initialize(ticker:, source: :yahoo_finance, type: :csv)
+  def initialize(
+        ticker:,
+        source: :alpha_vantage,
+        type:   :csv
+      )
+    raise "Invalid Ticker #{ticker}" unless SQA::Ticker.valid?(ticker)
+
+    # TODO: Change API on lookup to return array instead of hash
+    #       Could this also incorporate the validation process to
+    #       save an additiona hash lookup?
+
+    entry         = SQA::Ticker.lookup(ticker)
+
     @ticker       = ticker.downcase
-    @company_name = "Company Name"
+    @company_name = entry[:name]
+    @exchange     = entry[:exchange]
     @klass        = "SQA::DataFrame::#{source.to_s.camelize}".constantize
     @type         = type
-    @filename     = "#{@ticker}.#{type}"
     @indicators   = OpenStruct.new
 
     update_the_dataframe
   end
 
 
   def update_the_dataframe
-    df1 = @klass.load(@filename)
+    df1 = @klass.load(@ticker, type)
     df2 = @klass.recent(@ticker)
-    @df = @klass.append(df1, df2)
 
-    if @df.nrows > df1.nrows
-      @df.send("to_#{@type}", SQA::DataFrame.path(@filename))
+    df1_nrows = df1.nrows
+    @df       = @klass.append(df1, df2)
+
+    if @df.nrows > df1_nrows
+      @df.send("to_#{@type}", SQA.data_dir + "#{ticker}.csv")
     end
 
     # Adding a ticker vector in case I want to do
     # some multi-stock analysis in the same data frame.
+    # For example to see how one stock coorelates with another.
     @df[:ticker]  = @ticker
   end
 
   def to_s
     "#{ticker} with #{@df.size} data points from #{@df.timestamp.first} to #{@df.timestamp.last}"
   end
 end
-
-__END__
-
-aapl = Stock.new('aapl', SQA::Datastore::CSV)