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)