import finnhub import time import csv import pytz from datetime import datetime import datetime as dt import sys, os, base64, hashlib, hmac import requests from pycoingecko import CoinGeckoAPI from newsapi import NewsApiClient def readCSV(file_path, max_stocks): symbols = [] stock_info = {} f = open(file_path, 'r') CSV = csv.reader(f) next(CSV) i = 0 for row in CSV: if i < max_stocks: i += 1 try: symbol, current_price, opening_price = row symbols.append(symbol) stock_info[symbol] = [current_price, opening_price] except: symbol = row[0] symbols.append(symbol) stock_info[symbol] = [] else: print('max stocks exceeded') break f.close() return symbols, stock_info def emptyInfo(symbols, stock_info): update = False for symbol in symbols: if len(stock_info[symbol]) == 1: # stock with no info update = True return update def updateUpdate(NY_time): NY_str = NY_time.strftime("%d/%m/%Y %H:%M:%S") f = open('csv/last_update.csv', 'w+') f.write(NY_str + '\n') f.close() def updateStockPrices(symbols): try: quotes = [finnhubClient.quote(symbol) for symbol in symbols] current_prices = [quote['c'] for quote in quotes] opening_prices = [quote['o'] for quote in quotes] print(current_prices) print(opening_prices) print(current_prices, opening_prices) CSV = open('csv/tickers.csv', 'w+') CSV.write('name,current,opening\n') for i, symbol in enumerate(symbols): CSV.write(symbol + ',' + str(current_prices[i]) + ',' + str(opening_prices[i]) + '\n') CSV.close() print('API called successfully') except Exception as e: print("Could not fetch data - API CALLS REACHED? - Will display old image") print(e) apiCalledError = True def updateStockPricesIEX(symbols): symbols_str = ','.join(symbols) method = 'GET' host = 'https://cloud.iexapis.com/stable' lastEndpoint = '/tops/last' querystring = '?symbols=' + symbols_str +'&token='+iexAPIkey last_request_url = host + lastEndpoint + querystring print('\nBEGIN REQUEST++++++++++++++++++++++++++++++++++++') print('Request URL = ' + last_request_url) last_response = requests.get(last_request_url) if last_response.status_code == 200: print('last success') current_prices = [] for stock in last_response.json(): current_prices.append(stock['price']) for symbol in symbols: symbol = 'MSFT' method = 'GET' host = 'https://cloud.iexapis.com/stable' lastEndpoint = '/tops/last' intradayEndpoint = '/stock/'+ symbol+ '/intraday-prices' querystring = '?chartIEXOnly=true&token='+iexAPIkey intraday_request_url = host + intradayEndpoint + querystring print('\nBEGIN REQUEST++++++++++++++++++++++++++++++++++++') print('Request URL = ' + intraday_request_url) intraday_response = requests.get(intraday_request_url) print('\nRESPONSE++++++++++++++++++++++++++++++++++++') print('Response code: %d\n' % intraday_response.status_code) print(intraday_response.text) for time_point in intraday_response.json(): print(time_point['date']) print(time_point['open']) CSV = open('csv/tickers.csv', 'w+') CSV.write('name,current,opening\n') for i, symbol in enumerate(symbols): CSV.write(symbol + ',' + str(current_prices[i]) + ',' + str(opening_prices[i]) + '\n') CSV.close() print(lasts) def updateCrypto(coins): response = coingecko_client.get_price(ids=','.join(coins), vs_currencies = 'usd', include_24hr_change=True) CSV = open('csv/crypto.csv', 'w+') CSV.write('name,current,24hr change\n') for coin in coins: CSV.write(coin+ ',' + str(response[coin]['usd']) + ',' + str(response[coin]['usd_24h_change']) + '\n') CSV.close() def updateNews(): top_headlines = newsapi.get_top_headlines() headline_titles = [top_headline['title'] for top_headline in top_headlines['articles']] CSV = open('csv/news.csv', 'w+') CSV.write('headline\n') for title in headline_titles: CSV.write(title + '\n') CSV.close() if __name__ == '__main__': newsapi = NewsApiClient(api_key='cf08652bd17647b89aaf469a1a8198a9') finnhubAPIkey = "c24qddqad3ickpckgg80" #Finnhub finnhubsandboxAPIkey = "sandbox_c24qddqad3ickpckgg8g" #Finnhub sleeptime = 2 #minutes finnhubClient = finnhub.Client(api_key=finnhubAPIkey) max_stocks = 200 iexAPIkey = 'pk_68ef6a15902c41f887f0b544a0ca17cf' #IEX iexSandboxAPIkey = 'Tpk_0078dff413ef4f979137f7111452dc4b' #updateStockPricesIEX(symbols) finnhubClient = finnhub.Client(api_key=finnhubAPIkey) coingecko_client = CoinGeckoAPI() NY_zone = pytz.timezone('America/New_York') NY_time = datetime.now(NY_zone) opening = NY_time.replace(hour=9, minute=30, second=0, microsecond=0) closing = NY_time.replace(hour=16, minute=0, second=0, microsecond=0) NY_time = datetime.now(NY_zone) print(NY_time) symbols, stock_info = readCSV('csv/tickers.csv', max_stocks) updateStockPrices(symbols) updateUpdate(NY_time) while True: coins, coin_info = readCSV('csv/crypto.csv', max_stocks) updateCrypto(coins) updateNews() NY_time = datetime.now(NY_zone) symbols, stock_info = readCSV('csv/tickers.csv', max_stocks) if opening < NY_time < closing and datetime.today().weekday() < 5: # we need to do real time updating print('market open') updateStockPrices(symbols) updateUpdate(NY_time) elif emptyInfo(symbols, stock_info): # if theres any empty stocks updateStockPrices(symbols) updateUpdate(NY_time) else: # update if last update was before the previous days closing f = open('csv/last_update.csv', 'r') CSV = csv.reader(f) last_update_str = next(CSV)[0] print(last_update_str) last_update = datetime.strptime(last_update_str, "%d/%m/%Y %H:%M:%S") yday_closing = closing - dt.timedelta(days=1) yday_str = yday_closing.strftime("%d/%m/%Y %H:%M:%S") yday_closing = datetime.strptime(yday_str, "%d/%m/%Y %H:%M:%S") print(last_update < yday_closing) if last_update < yday_closing: updateStockPrices(symbols) updateUpdate(NY_time) time.sleep(sleeptime*60)