def get_message_stream(wl_id, params={}):
""" Gets up to 30 messages from Watchlist (wl_id) according to additional params
"""
all_params = ST_BASE_PARAMS.copy()
for k, v in params.iteritems():
all_params[k] = v
return R.get_json(ST_BASE_URL + 'streams/watchlist/{}.json'.format(wl_id), params=all_params)
def get_stock_stream(symbol, params={}):
""" gets stream of messages for given symbol
"""
all_params = ST_BASE_PARAMS.copy()
for k, v in params.iteritems():
all_params[k] = v
return R.get_json(ST_BASE_URL + 'streams/symbol/{}.json'.format(symbol), params=all_params)
def delete_from_watchlist(symbol, wl_id):
""" removes a single "symbols" (str) from watchlist. Returns True on success, False on failure
"""
params = ST_BASE_PARAMS.copy()
params['symbols'] = symbol
resp = R.post_json(ST_BASE_URL + 'watchlists/{}/symbols/destroy.json'.format(wl_id), params=params)
if resp['response']['status'] == 200:
return True
else:
return False
def add_to_watchlist(symbols, wl_id):
""" Adds list of symbols to our StockTwits watchlist. Returns list of new symbols added
"""
deadline = 30 * len(symbols)
symbols = ','.join(symbols) # must be a csv list
params = ST_BASE_PARAMS.copy()
params['symbols'] = symbols
resp = R.post_json(ST_BASE_URL + 'watchlists/{}/symbols/create.json'.format(wl_id), params=params, deadline=deadline)
if resp['response']['status'] == 200:
return [s['symbol'] for s in resp['symbols']]
else:
return []
