fintic-tracker/api_caller.py

62 lines
1.7 KiB
Python
Raw Normal View History

2021-05-05 15:22:01 +00:00
import finnhub
import time
import csv
2021-05-05 19:26:56 +00:00
import pytz
from datetime import datetime
2021-05-05 15:22:01 +00:00
APIkey = "c24qddqad3ickpckgg80"
sandboxAPIkey = "sandbox_c24qddqad3ickpckgg8g"
finnhubClient = finnhub.Client(api_key=APIkey)
2021-05-05 19:26:56 +00:00
NY_zone = pytz.timezone('America/New_York')
2021-05-05 19:26:56 +00:00
opening = NY_time.replace(hour=9, minute=30, second=0, microsecond=0)
closing = NY_time.replace(hour=16, minute=0, second=0, microsecond=0)
2021-05-05 15:22:01 +00:00
def getStockPrices():
NY_time = datetime.now(NY_zone)
2021-05-05 15:22:01 +00:00
2021-05-05 19:26:56 +00:00
if opening < NY_time < closing:
apiCalledError = False
stock_info = []
2021-05-05 15:22:01 +00:00
2021-05-05 19:26:56 +00:00
symbols = []
f = open('csv/tickers.csv', 'r')
CSV = csv.reader(f)
2021-05-05 15:22:01 +00:00
2021-05-05 19:26:56 +00:00
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
2021-05-05 15:22:01 +00:00
if __name__ == '__main__':
sleeptime = 2 #minutes
while True:
getStockPrices()
time.sleep(sleeptime*60)