Update utils.py

utils.py

@@ -12,6 +12,7 @@ class StockDataFetcher:
         self.base_url = "https://groww.in/v1/api/charting_service/v3/chart/exchange/NSE/segment/CASH/"
         self.base_fno_url = "https://groww.in/v1/api/stocks_fo_data/v3/charting_service/chart/exchange/NSE/segment/FNO/"
         self.latest_stock_price = "https://groww.in/v1/api/stocks_data/v1/tr_live_prices/exchange/NSE/segment/CASH/"
+        self.latest_index_price = "https://groww.in/v1/api/stocks_data/v1/tr_live_indices/exchange/NSE/segment/CASH/"
         self.latest_option_price = "https://groww.in/v1/api/stocks_fo_data/v1/tr_live_prices/exchange/NSE/segment/FNO/"
         self.option_chain = "https://groww.in/v1/api/option_chain_service/v1/option_chain/derivatives/"
         self.search_url = "https://groww.in/v1/api/search/v1/entity"
@@ -63,12 +64,16 @@ class StockDataFetcher:
       try:
         if symbol[-2:].upper() == "PE" or symbol[-2:].upper() == "CE" or symbol[-3:].upper() == "FUT":
           response = requests.get(self.latest_option_price + symbol.upper() + "/latest", headers=self.headers)
+        elif symbol.upper() == "NIFTY" or symbol.upper() == "BANKNIFTY":
+          response = requests.get(self.latest_index_price + symbol.upper() + "/latest", headers=self.headers)
         else:
           response = requests.get(self.latest_stock_price + symbol.upper() + "/latest", headers=self.headers)
         if response.status_code == 200:
             data = response.json()
-            latest_price = data.get('ltp')
-            print(symbol, 'Price: ', latest_price)
+            if 'ltp' in data:
+              latest_price = data['ltp']
+            else:
+              latest_price = data['value']
             return latest_price
         else:
             print(f"Failed to fetch data. Status code: {response.status_code}")