Update utils.py

utils.py

@@ -18,6 +18,7 @@ class StockDataFetcher:
         self.search_url = "https://groww.in/v1/api/search/v1/entity"
         self.news_url = "https://groww.in/v1/api/stocks_company_master/v1/company_news/groww_contract_id/"
         self.all_stocks_url = "https://groww.in/v1/api/stocks_data/v1/all_stocks"
+        self.nearest_expiries = "https://groww.in/v1/api/stocks_fo_data/v1/nearest_expiries?instrumentType=INDEX"
 
         self.indian_timezone = pytz.timezone('Asia/Kolkata')
         self.utc_timezone = pytz.timezone('UTC')
@@ -82,9 +83,19 @@ class StockDataFetcher:
           print(f"An error occurred: {e}")
           return None
 
+    def fetch_nearest_expiries(self):
+      try:
+          response = requests.get(self.nearest_expiries, headers=self.headers).json()
+          return response
+      except Exception as e:
+          print(f"An error occurred: {e}")
+          return None
+
     def fetch_option_chain(self, symbol):
+
       if symbol.upper() == "BANKNIFTY":
         symbol = "nifty-bank"
+
       response = requests.get(self.option_chain + symbol, headers=self.headers)
       data = response.json()['optionChain']['optionChains']
       ltp = response.json()['livePrice']['value']
@@ -102,7 +113,6 @@ class StockDataFetcher:
 
       chain = pd.DataFrame(chain)
       index = chain[(chain['strikePrice'] >= ltp)].head(1).index[0]
-      print(response.json()['livePrice'])
       chain = chain[index-6:index+7].reset_index(drop=True)
       optin_exp = chain['Symbol_CE'][0][:-7]
       return chain, optin_exp, index_ltp
@@ -122,7 +132,6 @@ class StockDataFetcher:
           entity = data['content'][0]
           return {"ID": entity['id'], "title": entity['title'], "NSE_Symbol": entity['nse_scrip_code'], "contract_id" : entity["groww_contract_id"]}
       except requests.exceptions.RequestException as e:
-          print(f"Error during API request: {e}")
           return None
 
     def fetch_stock_news(self, symbol, page=1, size=1):
@@ -133,7 +142,6 @@ class StockDataFetcher:
       try:
         symbol_id = self.search_entity(symbol.upper())['contract_id']
         response = requests.get(self.news_url + symbol_id, headers=self.headers, params=params).json()['results']
-        print(response)
         news = []
         for i in range(len(response)):
           Title = response[i]['title']
@@ -161,7 +169,6 @@ class StockDataFetcher:
         news_table = pd.DataFrame(news)
         return news_table
       except:
-        print("Something went wrong")
         return None
 
     def fetch_all_stock(self):
@@ -206,7 +213,7 @@ class StockDataFetcher:
         index_symbol = "NIFTY"
         rounding_value = 50
 
-      elif symbol.upper() == "NIFTY-BANK":
+      elif symbol.upper() == "BANKNIFTY":
         index_symbol = "BANKNIFTY"
         rounding_value = 100
 
@@ -214,7 +221,7 @@ class StockDataFetcher:
         pass
 
       stock_data = self.fetch_stock_data(index_symbol, intervals, days)
-      chain, exp = self.fetch_option_chain(symbol.upper())
+      chain, exp, index_ltp = self.fetch_option_chain(symbol.upper())
       stock_data['RSI'] = ta.momentum.rsi(stock_data['Close'], window=14)
       stock_data = stock_data.drop(columns=['Volume'])
       stock_data['Prev_RSI'] = stock_data['RSI'].shift(1)