fintic-tracker/api_caller.py

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)