Update nse.py

nse.py

@@ -1,131 +1,134 @@
 import requests
 import pandas as pd
 from datetime import datetime
-# ======================================================
-# Base Headers (mandatory for NSE API)
-# ======================================================
+
+# ================================
+# NSE Base Settings
+# ================================
 NSE_HEADERS = {
-    "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36",
+    "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 "
+                  "(KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36",
     "Accept-Language": "en-US,en;q=0.9",
 }
 
+# Persistent session
+session = requests.Session()
+session.get("https://www.nseindia.com", headers=NSE_HEADERS, timeout=5)
+
 
-# ======================================================
-# Helper: Convert DataFrame to HTML (single block)
-# ======================================================
+# ================================
+# Helper: Convert DataFrame to HTML
+# ================================
 def _to_html(title, df):
-    if df is None or len(df) == 0:
+    if df is None or df.empty:
         return f"<h3>{title}</h3><p>No data available</p>"
     return f"<h3>{title}</h3>" + df.to_html(index=False, border=1)
 
 
-# ======================================================================
-# STOCK QUOTE (similar to get_quote)
-# ======================================================================
-def nse_stock(symbol):
-    url = f"https://www.nseindia.com/api/quote-equity?symbol={symbol.upper()}"
+# ================================
+# Helper: Perform GET request
+# ================================
+def _get_json(url):
     try:
-        r = requests.get(url, headers=NSE_HEADERS)
-        data = r.json()
+        r = session.get(url, headers=NSE_HEADERS, timeout=5)
+        r.raise_for_status()
+        return r.json()
+    except Exception as e:
+        return {"error": str(e)}
 
-        info = pd.json_normalize(data.get("info", {}))
-        price = pd.json_normalize(data.get("priceInfo", {}))
-        meta = pd.json_normalize(data.get("metadata", {}))
 
-        html = ""
-        html += _to_html("Stock Info", info)
-        html += _to_html("Price Info", price)
-        html += _to_html("Metadata", meta)
+# ================================
+# 1. Stock Quote
+# ================================
+def nse_stock(symbol):
+    url = f"https://www.nseindia.com/api/quote-equity?symbol={symbol.upper()}"
+    data = _get_json(url)
+
+    if "error" in data:
+        return f"<p>Error fetching stock: {data['error']}</p>"
 
-        return html
+    info = pd.json_normalize(data.get("info", {}))
+    price = pd.json_normalize(data.get("priceInfo", {}))
+    meta = pd.json_normalize(data.get("metadata", {}))
 
-    except Exception as e:
-        return f"<p>Error fetching stock: {e}</p>"
+    html = ""
+    html += _to_html("Stock Info", info)
+    html += _to_html("Price Info", price)
+    html += _to_html("Metadata", meta)
+    return html
 
 
-# ======================================================================
-# OPTION CHAIN (similar to get_option_chain)
-# ======================================================================
+# ================================
+# 2. F&O / Option Chain
+# ================================
 def nse_fno(symbol):
     url = f"https://www.nseindia.com/api/option-chain-equities?symbol={symbol.upper()}"
-    try:
-        r = requests.get(url, headers=NSE_HEADERS)
-        data = r.json()
+    data = _get_json(url)
 
-        records = data.get("records", {})
+    if "error" in data:
+        return f"<p>Error fetching FNO: {data['error']}</p>"
 
-        all_data = pd.DataFrame(records.get("data", []))
-        ce_list = [x["CE"] for x in records.get("data", []) if "CE" in x]
-        pe_list = [x["PE"] for x in records.get("data", []) if "PE" in x]
+    records = data.get("records", {})
+    all_data = pd.DataFrame(records.get("data", []))
 
-        df_ce = pd.DataFrame(ce_list)
-        df_pe = pd.DataFrame(pe_list)
+    ce_list = [x["CE"] for x in records.get("data", []) if "CE" in x]
+    pe_list = [x["PE"] for x in records.get("data", []) if "PE" in x]
 
-        html = ""
-        html += _to_html("F&O Combined", all_data)
-        html += _to_html("CALL Options (CE)", df_ce)
-        html += _to_html("PUT Options (PE)", df_pe)
+    df_ce = pd.DataFrame(ce_list)
+    df_pe = pd.DataFrame(pe_list)
 
-        return html
-
-    except Exception as e:
-        return f"<p>Error fetching FNO: {e}</p>"
+    html = ""
+    html += _to_html("F&O Combined", all_data)
+    html += _to_html("CALL Options (CE)", df_ce)
+    html += _to_html("PUT Options (PE)", df_pe)
+    return html
 
 
-# ======================================================================
-# FUTURES DATA
-# ======================================================================
+# ================================
+# 3. Futures Data
+# ================================
 def nse_future(symbol):
     url = f"https://www.nseindia.com/api/quote-derivative?symbol={symbol.upper()}"
-    try:
-        r = requests.get(url, headers=NSE_HEADERS)
-        data = r.json()
-
-        futures = pd.DataFrame(data.get("stocks", []))
-        meta = pd.json_normalize(data.get("info", {}))
+    data = _get_json(url)
 
-        html = ""
-        html += _to_html("Futures Data", futures)
-        html += _to_html("Metadata", meta)
+    if "error" in data:
+        return f"<p>Error fetching futures: {data['error']}</p>"
 
-        return html
+    futures = pd.DataFrame(data.get("stocks", []))
+    meta = pd.json_normalize(data.get("info", {}))
 
-    except Exception as e:
-        return f"<p>Error fetching futures: {e}</p>"
+    html = ""
+    html += _to_html("Futures Data", futures)
+    html += _to_html("Metadata", meta)
+    return html
 
 
-# ======================================================================
-# 52-WEEK HIGH / LOW
-# ======================================================================
+# ================================
+# 4. 52-Week High / Low
+# ================================
 def nse_high_low():
     url = "https://www.nseindia.com/api/market-data-52Week"
-    try:
-        r = requests.get(url, headers=NSE_HEADERS)
-        data = r.json()
+    data = _get_json(url)
 
-        high = pd.DataFrame(data.get("FiftyTwoWeekHigh", []))
-        low = pd.DataFrame(data.get("FiftyTwoWeekLow", []))
+    if "error" in data:
+        return f"<p>Error fetching high-low: {data['error']}</p>"
 
-        html = ""
-        html += _to_html("52 Week High", high)
-        html += _to_html("52 Week Low", low)
+    high = pd.DataFrame(data.get("FiftyTwoWeekHigh", []))
+    low = pd.DataFrame(data.get("FiftyTwoWeekLow", []))
 
-        return html
+    html = ""
+    html += _to_html("52 Week High", high)
+    html += _to_html("52 Week Low", low)
+    return html
 
-    except Exception as e:
-        return f"<p>Error fetching high-low: {e}</p>"
-
-
-# ======================================================================
-# BHAVCOPY
-# ======================================================================
 
+# ================================
+# 5. Bhavcopy (date format flexible)
+# ================================
 def nse_bhav(date_str):
     """
-    Fetch NSE Bhavcopy for given date.
-    date_str: "DD-MM-YYYY" or "DD/MM/YYYY" or "DDMMYYYY"
+    date_str: DD-MM-YYYY, DD/MM/YYYY, or DDMMYYYY
     """
-    # Normalize date
     try:
         if "-" in date_str:
             dt = datetime.strptime(date_str, "%d-%m-%Y")
@@ -140,77 +143,62 @@ def nse_bhav(date_str):
 
     date_api = dt.strftime("%d-%m-%Y")
     url = f"https://www.nseindia.com/api/reports?archives=true&date={date_api}&type=equities&mode=single"
+    data = _get_json(url)
 
-    try:
-        r = session.get(url, headers=NSE_HEADERS, timeout=5)
-        r.raise_for_status()
-        data = r.json()
-        df = pd.DataFrame(data.get("data", []))
-
-        if df.empty:
-            return f"<h3>Bhavcopy {date_api}</h3><p>No data available</p>"
+    if "error" in data:
+        return f"<p>Error fetching bhavcopy: {data['error']}</p>"
 
-        return _to_html(f"Bhavcopy {date_api}", df)
+    df = pd.DataFrame(data.get("data", []))
+    return _to_html(f"Bhavcopy {date_api}", df)
 
-    except Exception as e:
-        return f"<p>Error fetching bhavcopy: {e}</p>"
 
-# ======================================================================
-# ALL NSE INDICES LIST
-# ======================================================================
+# ================================
+# 6. All NSE Indices
+# ================================
 def nse_indices():
     url = "https://www.nseindia.com/api/allIndices"
-    try:
-        r = requests.get(url, headers=NSE_HEADERS)
-        data = r.json()
+    data = _get_json(url)
 
-        df = pd.DataFrame(data.get("data", []))
+    if "error" in data:
+        return f"<p>Error fetching indices: {data['error']}</p>"
 
-        return _to_html("All NSE Indices", df)
-
-    except Exception as e:
-        return f"<p>Error fetching indices: {e}</p>"
+    df = pd.DataFrame(data.get("data", []))
+    return _to_html("All NSE Indices", df)
 
 
-# ======================================================================
-# NSE OPEN MARKET DATA (same endpoint used in nsepython)
-# ======================================================================
-def nse_open(index_name):
+# ================================
+# 7. NSE Open Market Data
+# ================================
+def nse_open(index_name="NIFTY 50"):
     url = f"https://www.nseindia.com/api/equity-stockIndices?index={index_name.replace(' ', '%20')}"
-    try:
-        r = requests.get(url, headers=NSE_HEADERS)
-        data = r.json()
-
-        meta = pd.json_normalize(data.get("metadata", {}))
-        df = pd.DataFrame(data.get("data", []))
+    data = _get_json(url)
 
-        html = ""
-        html += _to_html("Index Metadata", meta)
-        html += _to_html("Index Open Data", df)
+    if "error" in data:
+        return f"<p>Error fetching open data: {data['error']}</p>"
 
-        return html
+    meta = pd.json_normalize(data.get("metadata", {}))
+    df = pd.DataFrame(data.get("data", []))
 
-    except Exception as e:
-        return f"<p>Error fetching open data: {e}</p>"
+    html = ""
+    html += _to_html("Index Metadata", meta)
+    html += _to_html("Index Open Data", df)
+    return html
 
 
-# ======================================================================
-# NSE PRE-OPEN MARKET DATA
-# ======================================================================
-def nse_preopen(index_name):
+# ================================
+# 8. NSE Pre-Open Market Data
+# ================================
+def nse_preopen(index_name="NIFTY 50"):
     url = "https://www.nseindia.com/api/market-data-pre-open?key=NIFTY"
-    try:
-        r = requests.get(url, headers=NSE_HEADERS)
-        data = r.json()
+    data = _get_json(url)
 
-        df = pd.DataFrame(data.get("data", []))
-        meta = pd.json_normalize(data.get("metadata", {}))
+    if "error" in data:
+        return f"<p>Error fetching preopen: {data['error']}</p>"
 
-        html = ""
-        html += _to_html("Pre-Open Metadata", meta)
-        html += _to_html("Pre-Open Market Data", df)
+    df = pd.DataFrame(data.get("data", []))
+    meta = pd.json_normalize(data.get("metadata", {}))
 
-        return html
-
-    except Exception as e:
-        return f"<p>Error fetching preopen: {e}</p>"
+    html = ""
+    html += _to_html("Pre-Open Metadata", meta)
+    html += _to_html("Pre-Open Market Data", df)
+    return html