Update app.py

Added date to stock report & handled exceptions

app.py

@@ -63,36 +63,56 @@ def Client():
     return OpenAI(api_key = key)
 
 def get_stock_report():
-    stock_data = {}
-    global stock_data_path
-    with open(stock_data_path, 'rt') as fp:
-        lines = fp.readlines()
-    for line in lines:
-        (name, symbol, shares) = line.rstrip().split(',')
-        stock_data[symbol] = {"symbol": symbol, "name": name, "shares": shares, "closing": '0'}
-    for symbol in stock_data.keys():
-        stock_data[symbol]['closing'] = f'{get_last_closing(symbol):.2f}'
-    total_value = 0.0
-    rv = ''
-    for item in stock_data.values():
-        rv += str(item) + '\n'
-        total_value += float(item['closing']) * float(item['shares'])
-    rv += (f'\nTotal value = {total_value:.2f}')
+    try:
+        stock_data = {}
+        global stock_data_path
+        error_msg = ''
+        with open(stock_data_path, 'rt') as fp:
+            lines = fp.readlines()
+        for line in lines:
+            (name, symbol, shares) = line.rstrip().split(',')
+            name = name.strip()
+            symbol = symbol.strip()
+            shares = shares.strip()
+            stock_data[symbol] = {"symbol": symbol, "name": name, "shares": shares, "closing": '0'}
+        for symbol in stock_data.keys():
+            (closing_price, closing_date) = get_last_closing(symbol)
+            if closing_price == 0:
+                error_msg += f'Error getting closing for {symbol}\n'
+            stock_data[symbol]['closing'] = f'{closing_price:.2f}'
+        total_value = 0.0
+        rv = f'At closing on {closing_date}:\n'
+        for item in stock_data.values():
+            rv += str(item) + '\n'
+            total_value += float(item['closing']) * float(item['shares'])
+        rv += (f'\nTotal value = {total_value:.2f}\n')
+        if len(error_msg) > 0:
+            rv += error_msg
+    except:
+        rv = 'Error getting stock report'
     return rv
 
 def get_last_closing(symbol, timeout=10):
-    today = datetime.today()+timedelta(days=1)
-    five_days_ago = today - timedelta(days=5)
-    end = today.strftime('%Y-%m-%d')
-    start = five_days_ago.strftime('%Y-%m-%d')
-    df = yf.download(symbol,
-                    start = start,
-                    end = end,
-                    progress = False,
-                    timeout=timeout,
-                    )
-    # print(df)
-    return df.iat[-1,1]
+    try:
+        today = datetime.today()+timedelta(days=1)
+        five_days_ago = today - timedelta(days=5)
+        end = today.strftime('%Y-%m-%d')
+        start = five_days_ago.strftime('%Y-%m-%d')
+        df = yf.download(symbol,
+                        start = start,
+                        end = end,
+                        progress = False,
+                        timeout=timeout,
+                        )
+        # print(df)
+        closing_date = 'unknown'
+        data_top = df.tail(1)
+        for row in data_top.index:
+            closing_date = row.strftime('%Y-%m-%d')
+            # print(closing_date)
+        return (df.iat[-1,1], closing_date)
+    except:
+        return (0.0, "0000-00-00")
 
 def create_stock_data_file(txt):
     with open(stock_data_path, 'wt') as fp: