Update utils.py

utils.py

@@ -97,16 +97,29 @@ def random_test(strategy: str, period: str, interval: str, no_of_stocks: int = 5
     for i in random_indices:
         # Fetching ohlc of random ticker_symbol.
         ticker_symbol = nifty50['YahooEquiv'].values[i]
-        data = fetch(ticker_symbol, period, interval)
-
-        if strategy == "Order Block":
-            backtest_results = smc_backtest(data, kwargs['swing_hl'])
-        elif strategy == "Order Block with EMA":
-            backtest_results = smc_ema_backtest(data, kwargs['ema1'], kwargs['ema2'], kwargs['cross_close'])
-        elif strategy == "Structure trading":
-            backtest_results = smc_structure_backtest(data, kwargs['swing_hl'])
-        else:
-            raise Exception('Strategy not found')
+        retries = 3
+        for i in range(retries):
+            try:
+                data = fetch(ticker_symbol, period, interval)
+            except:
+                raise Exception(f"{ticker_symbol} data fetch failed")
+    
+            if len(data) == 0:
+                raise Exception(f"{ticker_symbol} ohlc is empty")
+            else:
+                break
+        
+        try:
+            if strategy == "Order Block":
+                backtest_results = smc_backtest(data, kwargs['swing_hl'])
+            elif strategy == "Order Block with EMA":
+                backtest_results = smc_ema_backtest(data, kwargs['ema1'], kwargs['ema2'], kwargs['cross_close'])
+            elif strategy == "Structure trading":
+                backtest_results = smc_structure_backtest(data, kwargs['swing_hl'])
+            else:
+                raise Exception('Strategy not found')
+        except:
+            raise Exception(f"{ticker_symbol} strategy run failed")
 
         with open("bokeh_graph.html", 'r', encoding='utf-8') as f:
             plot = f.read()