fintic-tracker/api_caller.py

import finnhub
import time
import csv
import pytz
from datetime import datetime

import datetime as dt

def readCSV():
    
    symbols = []
    stock_info = {}
    f = open('csv/tickers.csv', 'r')
    CSV = csv.reader(f)
    next(CSV)
    for row in CSV:
        print(row)
        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] = []
            
    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):
    
    apiCalledError = False
    stock_info = []
    
    symbols = []
    f = open('csv/tickers.csv', 'r')
    CSV = csv.reader(f)
    next(CSV) #read through headers
    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+')
        CSV.write('name,current,opening\n')
        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__':
    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)

    NY_time = datetime.now(NY_zone)
    
    sleeptime = 2 #minutes
    
    while True:
        NY_time = datetime.now(NY_zone)
        
        symbols, stock_info = readCSV()
        print(type(NY_time))
        
        if opening < NY_time < closing: # we need to do real time updating
            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)