lib/sqa/stock.rb in sqa-0.0.6 vs lib/sqa/stock.rb in sqa-0.0.7
- old
+ new
@@ -4,13 +4,15 @@
attr_accessor :company_name
attr_accessor :df # The DataFrane
attr_accessor :ticker
def initialize(ticker:, source: :yahoo_finance, type: :csv)
- @ticker = ticker
+ @ticker = ticker.downcase
@company_name = "Company Name"
klass = "SQA::DataFrame::#{source.to_s.camelize}".constantize
- @df = klass.send("from_#{type.downcase}", ticker)
+ filename = "#{@ticker}.#{type}"
+ @df = klass.send(:load, filename)
+ @df[:ticker] = ticker
end
def to_s
"#{ticker} with #{@df.size} data points from #{@df.timestamp.first} to #{@df.timestamp.last}"
end