ytterb 0.1.1 → 0.1.147.77920

data/lib/ytterb/stock_symbol_loader.rb

@@ -0,0 +1,81 @@
+require 'csv'
+
+module Ytterb
+
+  class StockSymbol
+    def initialize(options={})
+      @options = {}
+      options.reject {|k,v| !k }.each do |k,v|
+        sym_k = k.gsub(/\s+/,"_").downcase.to_sym
+        @options[sym_k] = v
+      end
+    end
+
+    def attributes
+      return @options.keys
+    end
+
+    def method_missing(meth, *args, &block) 
+      if @options.has_key?(meth)
+        return @options[meth]
+      else
+        # don't know how to handle it
+        super
+      end
+    end
+
+    def to_s
+      @options.to_s
+    end
+  end
+
+
+  class StockSymbolLoader
+    def initialize(stock_symbol_file, exchange)
+      @stock_symbol_file = stock_symbol_file
+      @exchange = exchange
+    end
+
+    def parse
+      CSV.foreach(@stock_symbol_file, :headers => true) do |obj|
+        yield StockSymbol.new(obj.to_hash.merge("Exchange" => @exchange))
+      end
+    end
+  end
+
+  class StockMarketLoader
+    def initialize
+      @stock_symbols = []
+      path = File.join(File.expand_path(File.dirname(__FILE__)),"raw_symbol_data")
+      Dir.entries(path).select {|f| !File.directory?(f) and /companylist_[a-zA-Z]+\.csv/ =~ f }.each do |file_to_process|
+        market = /companylist_(?<market>[a-zA-Z]+)\.csv/.match(file_to_process)[:market]
+        StockSymbolLoader.new(File.join(path,file_to_process),market).parse do |stock_symbol|
+          @stock_symbols << stock_symbol
+        end
+      end
+    end
+
+    def industries
+      return @industries if @industries
+      @raw_industries = {}
+      @stock_symbols.each do |stock_symbol|
+        @raw_industries[stock_symbol.industry] = 1
+      end
+      @industries = @raw_industries.keys
+    end
+
+    def sectors
+      return @sectors if @sectors
+      @raw_sectors = {}
+      @stock_symbols.each do |stock_symbol|
+        @raw_sectors[stock_symbol.sector] = 1
+      end
+      @sectors = @raw_sectors.keys
+    end
+
+    def stock_symbols
+      @stock_symbols
+    end
+  end
+
+end

data/lib/ytterb/version.rb

@@ -1,4 +1,4 @@
 module Ytterb
   # ytterb version
-  VERSION = "0.1.1"
+  VERSION = "0.1"
 end

data/lib/ytterb/yql_client.rb

@@ -0,0 +1,38 @@
+require 'rest-client'
+require 'cgi'
+require 'json'
+require 'date'
+
+module Ytterb
+
+  class YQLClient
+    def intialize
+      @endpoint = "http://query.yahooapis.com/v1/public/yql"
+    end
+
+    def get_yql_endpoint
+      @endpoint
+    end
+
+    def run_select_query(query)
+      RestClient.get("#{get_yql_endpoint}?q=#{CGI::escape(query)}&format=json&env=store://datatables.org/alltableswithkeys") do |response, request, result|
+        raise "Failed performing YQLQuery\n Request: #{request.inspect}\n Response #{response.inspect}" unless response.code == 200
+        return JSON.parse(response)
+      end
+    end
+
+    def get_symbol_current(my_symbol)
+      run_select_query("select * from yahoo.finance.quotes where symbol in (\"#{my_symbol}\")")["query"]["results"]
+    end
+
+    def get_symbol_historical(my_symbol, start_date = nil, end_date = nil)
+      requested_end = Date.parse(end_date) if end_date
+      requested_end = Date.today unless requested_end
+
+      requested_start = Date.parse(start_date) if start_date
+      requested_start = requested_end - 30 unless requested_start
+      run_select_query("select * from yahoo.finance.historicaldata where symbol in (\"#{my_symbol}\") and startDate='#{requested_start}' and endDate='#{requested_end}'")["query"]["results"]
+    end
+    
+  end
+end

data/ytterb.gemspec

@@ -1,10 +1,14 @@
 # -*- encoding: utf-8 -*-
 
+require 'date'
 require File.expand_path('../lib/ytterb/version', __FILE__)
 
+t = Time.now
+minor_rev = t.hour * 3600 + t.min * 60 + t.sec
+
 Gem::Specification.new do |gem|
   gem.name          = "ytterb"
-  gem.version       = Ytterb::VERSION
+  gem.version       = "#{Ytterb::VERSION}.#{Date.today.yday}.#{minor_rev}"
   gem.summary       = %q{ytterb finance data gen}
   gem.description   = %q{pulls financial data data via yahoo's YQL apis}
   gem.license       = "MIT"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ytterb
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.147.77920
 platform: ruby
 authors:
 - thext
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-05-25 00:00:00.000000000 Z
+date: 2014-05-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -94,7 +94,12 @@ files:
 - README.md
 - Rakefile
 - lib/ytterb.rb
+- lib/ytterb/raw_symbol_data/companylist_amex.csv
+- lib/ytterb/raw_symbol_data/companylist_nasdaq.csv
+- lib/ytterb/raw_symbol_data/companylist_nyse.csv
+- lib/ytterb/stock_symbol_loader.rb
 - lib/ytterb/version.rb
+- lib/ytterb/yql_client.rb
 - spec/spec_helper.rb
 - spec/ytterb_spec.rb
 - ytterb.gemspec