Update qresult.py

qresult.py

@@ -1,104 +1,35 @@
 # qresult.py
 import yfinance as yf
 import pandas as pd
-from common import (
-    format_large_number,
-    format_timestamp_to_date,
-    format_number,
-    wrap_html,
-    STYLE_BLOCK
-)
+from common import make_table, wrap_html, format_large_number, html_error
 
 def fetch_qresult(symbol):
-    yfsymbol = f"{symbol}.NS"
-
+    """
+    Fetch quarterly financials for a stock symbol and return HTML
+    """
+    yfsymbol = symbol + ".NS"
     try:
         ticker = yf.Ticker(yfsymbol)
+        df = ticker.quarterly_financials
 
-        # Fetch data
-        q = ticker.quarterly_financials
-        y = ticker.financials
-        bs = ticker.balance_sheet
-        cf = ticker.cashflow
+        if df.empty:
+            return wrap_html(f"<h1>No quarterly results available for {symbol}</h1>")
 
-        if q is None or q.empty:
-            return wrap_html(
-                f"Quarterly Results: {symbol}",
-                "<h2>No quarterly data available</h2>"
+        # Format numeric columns
+        df_formatted = df.copy()
+        for col in df_formatted.columns:
+            df_formatted[col] = df_formatted[col].apply(
+                lambda x: format_large_number(x) if isinstance(x, (int, float)) else x
             )
 
-        # Transpose for nicer table format
-        q_t = q.T
-        y_t = y.T if y is not None else None
-        bs_t = bs.T if bs is not None else None
-        cf_t = cf.T if cf is not None else None
-
-        # ------------------------
-        def format_df(df):
-            df_formatted = df.copy()
-    
-            # Format numeric columns
-            for col in df_formatted.columns:
-                df_formatted[col] = df_formatted[col].apply(
-                    lambda x: format_number(x) if isinstance(x, (int, float)) else x
-                )
-
-            # Fix index (assume index is datetime-like)
-            df_formatted.index = [
-                format_timestamp_to_date(i.timestamp()) if hasattr(i, "timestamp") else str(i)
-                for i in df_formatted.index
-            ]
-
-            return df_formatted
-
-
-        q_html = format_df(q_t).to_html(classes="styled-table", border=0)
-
-        y_html = ""
-        if y_t is not None:
-            y_html = format_df(y_t).to_html(classes="styled-table", border=0)
-
-        bs_html = ""
-        if bs_t is not None:
-            bs_html = format_df(bs_t).to_html(classes="styled-table", border=0)
-
-        cf_html = ""
-        if cf_t is not None:
-            cf_html = format_df(cf_t).to_html(classes="styled-table", border=0)
-
-        # Build sections
-        content = f"""
-        <div class='big-box'>
-            <h2>Quarterly Results</h2>
-            {q_html}
-        </div>
-        """
-
-        if y_html:
-            content += f"""
-            <div class='big-box'>
-                <h2>Annual Results</h2>
-                {y_html}
-            </div>
-            """
-
-        if bs_html:
-            content += f"""
-            <div class='big-box'>
-                <h2>Balance Sheet</h2>
-                {bs_html}
-            </div>
-            """
+        # Format index (dates)
+        df_formatted.index = [str(i.date()) if hasattr(i, "date") else str(i) for i in df_formatted.index]
 
-        if cf_html:
-            content += f"""
-            <div class='big-box'>
-                <h2>Cash Flow</h2>
-                {cf_html}
-            </div>
-            """
+        # Convert to pretty HTML table
+        table_html = make_table(df_formatted)
 
-        return wrap_html(f"Quarterly Results — {symbol}", content)
+        # Wrap into full HTML page
+        return wrap_html(table_html, title=f"{symbol} - Quarterly Results")
 
     except Exception as e:
-        return wrap_html("Error", f"<h3>Error fetching results</h3><p>{str(e)}</p>")
+        return wrap_html(html_error(f"Failed to fetch quarterly results: {e}"))