Skip to content

Commit

Permalink
Worldtradedata took a dump, back to alphavantage
Browse files Browse the repository at this point in the history
  • Loading branch information
@onewheelskyward
onewheelskyward committed Apr 9, 2020
1 parent f08e050 commit e77bedd
Show file tree
Hide file tree
Showing 4 changed files with 244 additions and 220 deletions.
64 changes: 64 additions & 0 deletions lib/lita/handlers/alphavantage_quote.rb
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,64 @@
class AlphaVantageQuote
attr_reader :open, :high, :low, :price, :volume, :trading_day, :prev_close, :change, :change_percent, :exchange, :error, :name, :message, :is_index
attr_accessor :symbol

def initialize(symbol, apikey)
# Lita.logger.debug "parsing: #{json_blob}"
# hash = JSON.parse(json_blob)
@symbol = symbol
@apikey = apikey
@is_index = false

response = call_api
hash = JSON.parse response

quote = hash["Global Quote"]

quote.keys.each do |key|
case key
when "01. symbol"
@symbol = quote[key]
when "02. open"
@open = self.fix_number quote[key]
when "03. high"
@high = self.fix_number quote[key]
when "04. low"
@low = self.fix_number quote[key]
when "05. price"
@price = self.fix_number quote[key]
when "06. volume"
@volume = quote[key]
when "07. latest trading day"
@trading_day = quote[key]
when "08. previous close"
@prev_close = self.fix_number quote[key]
when "09. change"
@change = self.fix_number quote[key]
when "10. change percent"
@change_percent = self.fix_number quote[key]
end
end
end

def fix_number(price_str)
price_str.to_f.round(2)
end

def is_index?
@is_index
end

def call_api
url = "https://www.alphavantage.co/query"
params = {function: 'GLOBAL_QUOTE', symbol: @symbol, apikey: @apikey}
Lita.logger.debug "#{url} #{params.inspect}"
response = RestClient.get url, {params: params}

# if @exchange
# params[:stock_exchange] = @exchange
# end

Lita.logger.debug "response: #{response}"
response
end
end
43 changes: 43 additions & 0 deletions lib/lita/handlers/irc_colors.rb
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,43 @@
module IrcColors
prefix = "\x03"
@white = "#{prefix}00"
@black = "#{prefix}01"
@blue = "#{prefix}02"
@green = "#{prefix}03"
@red = "#{prefix}04"
@brown = "#{prefix}05"
@purple = "#{prefix}06"
@orange = "#{prefix}07"
@yellow = "#{prefix}08"
@lime = "#{prefix}09"
@teal = "#{prefix}10"
@aqua = "#{prefix}11"
@royal = "#{prefix}12"
@pink = "#{prefix}13"
@grey = "#{prefix}14"
@silver = "#{prefix}15"
@reset = prefix

class << self
attr_reader :white,
:black,
:blue,
:green,
:red,
:brown,
:purple,
:orange,
:yellow,
:lime,
:teal,
:aqua,
:royal,
:pink,
:grey,
:silver,
:reset
end

def initialize
end
end
229 changes: 9 additions & 220 deletions lib/lita/handlers/onewheel_finance.rb
View file
Original file line number Diff line number Diff line change
@@ -1,225 +1,18 @@
require_relative 'irc_colors'
require_relative 'alphavantage_quote'
require_relative 'worldtradedata_quote'
require 'rest-client'

class IrcColors
prefix = "\x03"
@white = "#{prefix}00"
@black = "#{prefix}01"
@blue = "#{prefix}02"
@green = "#{prefix}03"
@red = "#{prefix}04"
@brown = "#{prefix}05"
@purple = "#{prefix}06"
@orange = "#{prefix}07"
@yellow = "#{prefix}08"
@lime = "#{prefix}09"
@teal = "#{prefix}10"
@aqua = "#{prefix}11"
@royal = "#{prefix}12"
@pink = "#{prefix}13"
@grey = "#{prefix}14"
@silver = "#{prefix}15"
@reset = prefix

class << self
attr_reader :white,
:black,
:blue,
:green,
:red,
:brown,
:purple,
:orange,
:yellow,
:lime,
:teal,
:aqua,
:royal,
:pink,
:grey,
:silver,
:reset
end

def initialize
end
end

class AlphaVantageQuote
attr_reader :symbol, :open, :high, :low, :price, :volume, :trading_day, :prev_close, :change, :change_percent

def initialize(json_blob)
Lita.logger.debug "parsing: #{json_blob}"
hash = JSON.parse(json_blob)
quote = hash["Global Quote"]

quote.keys.each do |key|
case key
when "01. symbol"
@symbol = quote[key]
when "02. open"
@open = self.fix_number quote[key]
when "03. high"
@high = self.fix_number quote[key]
when "04. low"
@low = self.fix_number quote[key]
when "05. price"
@price = self.fix_number quote[key]
when "06. volume"
@volume = quote[key]
when "07. latest trading day"
@trading_day = quote[key]
when "08. previous close"
@prev_close = self.fix_number quote[key]
when "09. change"
@change = self.fix_number quote[key]
when "10. change percent"
@change_percent = self.fix_number quote[key]
end
end
end

def fix_number(price_str)
price_str.to_f.round(2)
end
end

class WorldTradeDataQuote
attr_reader :open, :high, :low, :price, :volume, :trading_day, :prev_close, :change, :change_percent, :exchange, :error, :name, :message, :is_index
attr_accessor :symbol

def initialize(symbol, api_key)
@base_uri = 'https://api.worldtradingdata.com/api/v1'
@symbol = symbol
@api_key = api_key

if @symbol[':']
(@exchange, @symbol) = @symbol.split /:/
end

if @symbol['^']
@is_index = true
else
@is_index = false
end

self.call_api

hash = JSON.parse(@response)

# We couldn't find the stock. Let's look for it real quick.
if hash['Message'].to_s.include? 'Error'
@error = true
self.run_search

if @message
@error = true
return
else
self.call_api
hash = JSON.parse(@response)
@error = false
end
else
@error = false
end

unless hash['data']
@message = "Error getting data for #{@symbol}"
@error = true
return
end

quote = hash['data'][0]

quote.keys.each do |key|
case key
when "symbol"
@symbol = quote[key]
when "price_open"
@open = self.fix_number quote[key]
when "day_high"
@high = self.fix_number quote[key]
when "day_low"
@low = self.fix_number quote[key]
when "price"
@price = self.fix_number quote[key]
when "volume"
@volume = quote[key].to_i
when "last_trade_time"
@trading_day = quote[key]
when "08. previous close"
@prev_close = self.fix_number quote[key]
when "day_change"
@change = self.fix_number quote[key]
when "change_pct"
@change_percent = self.fix_number quote[key]
when 'stock_exchange_short'
@exchange = quote[key].sub /NYSEARCA/, 'NYSE'
when 'name'
@name = quote[key]
end
end

def is_index?
is_index
end
end

# Let's see what we can get from the api.
def call_api
url = "#{@base_uri}/stock"
params = {symbol: @symbol, api_token: @api_key}

if @exchange
params[:stock_exchange] = @exchange
end

Lita.logger.debug "call_api: #{url} #{params.inspect}"

@response = RestClient.get url, {params: params}

Lita.logger.debug "response: #{@response}"
end

def run_search
url = "#{@base_uri}/stock_search"
params = {search_term: @symbol,
search_by: 'symbol,name',
stock_exchange: 'NASDAQ,NYSE',
limit: 5,
page: 1,
api_token: @api_key
}

Lita.logger.debug "run_search: #{url} #{params.inspect}"

response = RestClient.get url, {params: params}

Lita.logger.debug "response: #{response}"
result = JSON.parse(response)

if result['total_returned'] == 1
@symbol = result['data'][0]['symbol']
elsif result['total_returned'].to_i > 1
Lita.logger.debug "many search results: #{result.inspect}"
x = result['data'].map { |k| k.values[0] }
@message = "`#{symbol}` not found, did you mean one of #{x.join(', ')}?"
end
end

def fix_number(price_str)
price_str.to_f.round(2)
end
end

module Lita
module Handlers
class OnewheelFinance < Handler
config :apikey, required: true
route /qu*o*t*e*\s+(.+)/i, :handle_quote, command: true
# route /q2\s+(.+)/i, :handle_alphavantage, command: true

def handle_quote(response)
stock = handle_world_trade_data response.matches[0][0]
# stock = handle_world_trade_data response.matches[0][0]
stock = handle_alphavantage response.matches[0][0]

if stock.error
if stock.message
Expand Down Expand Up @@ -256,13 +49,9 @@ def handle_world_trade_data(symbol)
end

# deprecated for now
#def handle_alphavantage
# url = "https://www.alphavantage.co/query"
# params = {function: 'GLOBAL_QUOTE', symbol: response.matches[0][0], apikey: config.apikey}
# Lita.logger.debug "#{url} #{params.inspect}"
# resp = RestClient.get url, {params: params}
# stock = GlobalQuote.new resp
#end
def handle_alphavantage(symbol)
stock = AlphaVantageQuote.new symbol, config.apikey
end

Lita.register_handler(self)
end
Expand Down
Loading

0 comments on commit e77bedd

Please sign in to comment.