api call conditions improved

2021-05-06 20:59:27 +01:00
parent 18841c817f
commit 3d6fb3d722
8 changed files with 139 additions and 69 deletions
--- a/api_caller.py
+++ b/api_caller.py
@@ -4,58 +4,122 @@ import csv
 import pytz
 from datetime import datetime

-APIkey = "c24qddqad3ickpckgg80"
-sandboxAPIkey = "sandbox_c24qddqad3ickpckgg8g"
-finnhubClient = finnhub.Client(api_key=APIkey)
+import datetime as dt

-NY_zone = pytz.timezone('America/New_York')
-
-
-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():
-    NY_time = datetime.now(NY_zone)
+def readCSV():
    
-    
-    if opening < NY_time < closing:
-        apiCalledError = False
-        stock_info = []
-        
-        symbols = []
-        f = open('csv/tickers.csv', 'r')
-        CSV = csv.reader(f)
-        
-        for row in CSV:
+    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)
-        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]
+            stock_info[symbol] = []
            
-            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')
-                
+    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
-    
-    
-    
+        
+    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:
-        getStockPrices()
+        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)