60 lines
1.7 KiB
Python
60 lines
1.7 KiB
Python
import finnhub
|
|
import time
|
|
import csv
|
|
import pytz
|
|
from datetime import datetime
|
|
|
|
APIkey = "c24qddqad3ickpckgg80"
|
|
sandboxAPIkey = "sandbox_c24qddqad3ickpckgg8g"
|
|
finnhubClient = finnhub.Client(api_key=APIkey)
|
|
|
|
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)
|
|
|
|
|
|
|
|
|
|
|
|
def getStockPrices():
|
|
|
|
if opening < NY_time < closing:
|
|
apiCalledError = False
|
|
stock_info = []
|
|
|
|
symbols = []
|
|
f = open('csv/tickers.csv', 'r')
|
|
CSV = csv.reader(f)
|
|
|
|
for row in CSV:
|
|
symbol = row[0]
|
|
symbols.append(symbol)
|
|
f.close()
|
|
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]
|
|
|
|
CSV = open('csv/tickers.csv', 'w+')
|
|
for i, symbol in enumerate(symbols):
|
|
symbol + ',' + str(current_prices[i]) + ',' + str(opening_prices[i]) + '\n'
|
|
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
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
sleeptime = 2 #minutes
|
|
while True:
|
|
getStockPrices()
|
|
time.sleep(sleeptime*60)
|