Create stock.py

app/stock.py

@@ -0,0 +1,386 @@
+# stock.py — compact, merged, single-file (collision-safe)
+
+import traceback
+import pandas as pd
+import yfinance as yf
+from datetime import datetime as dt
+
+import persist
+import common
+from backblaze import *
+import ta_indi_pat
+
+
+# ================================================================
+#                    BASIC YFINANCE FETCHERS
+# ================================================================
+
+def yfinfo(symbol):
+    return yf.Ticker(symbol + ".NS").info
+
+
+def qresult(symbol):
+    return yf.Ticker(symbol + ".NS").quarterly_financials
+
+
+def result(symbol):
+    return yf.Ticker(symbol + ".NS").financials
+
+
+def balance(symbol):
+    return yf.Ticker(symbol + ".NS").balance_sheet
+
+
+def cashflow(symbol):
+    return yf.Ticker(symbol + ".NS").cashflow
+
+
+def dividend(symbol):
+    return yf.Ticker(symbol + ".NS").dividends.to_frame("Dividend")
+
+
+def split(symbol):
+    return yf.Ticker(symbol + ".NS").splits.to_frame("Split")
+
+
+def intraday(symbol):
+    print(f"[{dt.now().strftime('%Y-%m-%d %H:%M:%S')}] yf called for {symbol}")
+    return yf.download(symbol + ".NS", period="1d", interval="5m").round(2)
+
+
+def daily(symbol):
+    print(f"[{dt.now().strftime('%Y-%m-%d %H:%M:%S')}] yf called for {symbol}")
+    return yf.download(symbol + ".NS", period="1y", interval="1d").round(2)
+
+
+# ================================================================
+#              HTML HELPERS (delegated to common)
+# ================================================================
+
+format_number = common.format_number
+format_large_number = common.format_large_number
+make_table = common.make_table
+html_card = common.html_card
+html_section = common.html_section
+html_error = common.html_error
+clean_df = common.clean_df
+safe_get = common.safe_get
+wrap_html = common.wrap_html
+
+#build_chart = chart_builder.build_chart
+talib_df = ta_indi_pat.talib_df
+
+
+# ================================================================
+#                         INTRADAY
+# ================================================================
+
+def fetch_intraday(symbol, indicators=None):
+    key = f"intraday_{symbol}"
+
+    if persist.exists(key, "html"):
+        cached = persist.load(key, "html")
+        if cached is not False:
+            print(f"[{dt.now().strftime('%Y-%m-%d %H:%M:%S')}] Using cached intraday for {symbol}")
+            return cached
+
+    try:
+        df = intraday(symbol)
+        if df is None or df is False or df.empty:
+            return wrap_html(f"<h1>No intraday data for {symbol}</h1>")
+
+        if isinstance(df.columns, pd.MultiIndex):
+            df.columns = df.columns.get_level_values(0)
+
+        upload_file("eshanhf", f"intraday/{symbol}.csv", df)
+
+        df_display = df.tail(50).copy()
+        df_display.reset_index(inplace=True)
+
+        html = wrap_html(
+            f"<h2>Last 50 Rows</h2>{make_table(df_display)}",
+            title=f"{symbol} Intraday"
+        )
+
+        persist.save(key, html, "html")
+        return html
+
+    except Exception as e:
+        print(f"[{dt.now().strftime('%Y-%m-%d %H:%M:%S')}] Error fetch_intraday: {e}")
+        return wrap_html(f"<h1>Error: {e}</h1>")
+
+
+# ================================================================
+#                           DAILY
+# ================================================================
+
+def fetch_daily(symbol):
+    key = f"daily_{symbol}"
+
+    if persist.exists(key, "html"):
+        cached = persist.load(key, "html")
+        if cached is not False:
+            print(f"[{dt.now().strftime('%Y-%m-%d %H:%M:%S')}] Using cached daily for {symbol}")
+            return cached
+
+    try:
+        df = daily(symbol)
+        if df is None or df is False or df.empty:
+            return wrap_html(f"<h1>No daily data for {symbol}</h1>")
+
+        upload_file("eshanhf", f"daily/{symbol}.csv", df)
+
+        df_display = df.tail(50).copy()
+        df_display.reset_index(inplace=True)
+
+        html = wrap_html(
+            f"<h2>Last 50 Rows</h2>{make_table(df_display)}",
+            title=f"{symbol} Daily"
+        )
+
+        persist.save(key, html, "html")
+        return html
+
+    except Exception as e:
+        print(f"[{dt.now().strftime('%Y-%m-%d %H:%M:%S')}] Error fetch_daily: {e}")
+        return wrap_html(f"<h1>Error: {e}</h1>")
+
+
+# ================================================================
+#                        QUARTERLY
+# ================================================================
+
+def fetch_qresult(symbol):
+    key = f"qresult_{symbol}"
+
+    if persist.exists(key, "html"):
+        cached = persist.load(key, "html")
+        if cached is not False:
+            print(f"[{dt.now().strftime('%Y-%m-%d %H:%M:%S')}] Using cached quarterly for {symbol}")
+            return cached
+
+    try:
+        df = qresult(symbol)
+        if df.empty:
+            return wrap_html(f"<h1>No quarterly results for {symbol}</h1>")
+
+        upload_file("eshanhf", f"qresult/{symbol}.csv", df)
+
+        df_display = df.copy()
+        for col in df_display.columns:
+            df_display[col] = df_display[col].apply(
+                lambda x: format_large_number(x) if isinstance(x, (int, float)) else x
+            )
+
+        df_display.reset_index(inplace=True)
+
+        html = wrap_html(make_table(df_display), title=f"{symbol} Quarterly Results")
+        persist.save(key, html, "html")
+        return html
+
+    except Exception as e:
+        print(f"[{dt.now().strftime('%Y-%m-%d %H:%M:%S')}] Error fetch_qresult: {e}")
+        return wrap_html(html_error(f"Quarterly Error: {e}"))
+
+
+# ================================================================
+#                          ANNUAL
+# ================================================================
+
+def fetch_result(symbol):
+    key = f"result_{symbol}"
+
+    if persist.exists(key, "html"):
+        cached = persist.load(key, "html")
+        if cached is not False:
+            print(f"[{dt.now().strftime('%Y-%m-%d %H:%M:%S')}] Using cached annual for {symbol}")
+            return cached
+
+    try:
+        df = result(symbol)
+        if df.empty:
+            return wrap_html(f"<h1>No annual results for {symbol}</h1>")
+
+        upload_file("eshanhf", f"result/{symbol}.csv", df)
+
+        df_display = df.copy()
+        for col in df_display.columns:
+            df_display[col] = df_display[col].apply(
+                lambda x: format_large_number(x) if isinstance(x, (int, float)) else x
+            )
+
+        df_display.reset_index(inplace=True)
+
+        html = wrap_html(make_table(df_display), title=f"{symbol} Annual Results")
+        persist.save(key, html, "html")
+        return html
+
+    except Exception as e:
+        print(f"[{dt.now().strftime('%Y-%m-%d %H:%M:%S')}] Error fetch_result: {e}")
+        return wrap_html(html_error(f"Annual Error: {e}"))
+
+
+# ================================================================
+#                        BALANCE SHEET
+# ================================================================
+
+def fetch_balance(symbol):
+    key = f"balance_{symbol}"
+
+    if persist.exists(key, "html"):
+        cached = persist.load(key, "html")
+        if cached is not False:
+            print(f"[{dt.now().strftime('%Y-%m-%d %H:%M:%S')}] Using cached balance for {symbol}")
+            return cached
+
+    try:
+        df = balance(symbol)
+        if df.empty:
+            return wrap_html(f"<h1>No balance sheet for {symbol}</h1>")
+
+        upload_file("eshanhf", f"balance/{symbol}.csv", df)
+
+        df_display = df.copy()
+        for col in df_display.columns:
+            df_display[col] = df_display[col].apply(
+                lambda x: format_large_number(x) if isinstance(x, (int, float)) else x
+            )
+
+        df_display.reset_index(inplace=True)
+
+        html = wrap_html(make_table(df_display), title=f"{symbol} Balance Sheet")
+        persist.save(key, html, "html")
+        return html
+
+    except Exception as e:
+        print(f"[{dt.now().strftime('%Y-%m-%d %H:%M:%S')}] Error fetch_balance: {e}")
+        return wrap_html(html_error(f"Balance Error: {e}"))
+
+
+# ================================================================
+#                          CASHFLOW
+# ================================================================
+
+def fetch_cashflow(symbol):
+    key = f"cashflow_{symbol}"
+
+    if persist.exists(key, "html"):
+        cached = persist.load(key, "html")
+        if cached is not False:
+            print(f"[{dt.now().strftime('%Y-%m-%d %H:%M:%S')}] Using cached cashflow for {symbol}")
+            return cached
+
+    try:
+        df = cashflow(symbol)
+        if df.empty:
+            return wrap_html(f"<h1>No cashflow for {symbol}</h1>")
+
+        upload_file("eshanhf", f"cashflow/{symbol}.csv", df)
+
+        df_display = df.copy()
+        for col in df_display.columns:
+            df_display[col] = df_display[col].apply(
+                lambda x: format_large_number(x) if isinstance(x, (int, float)) else x
+            )
+
+        df_display.reset_index(inplace=True)
+
+        html = wrap_html(make_table(df_display), title=f"{symbol} Cash Flow")
+        persist.save(key, html, "html")
+        return html
+
+    except Exception as e:
+        print(f"[{dt.now().strftime('%Y-%m-%d %H:%M:%S')}] Error fetch_cashflow: {e}")
+        return wrap_html(html_error(f"Cash Flow Error: {e}"))
+
+
+# ================================================================
+#                         DIVIDEND
+# ================================================================
+
+def fetch_dividend(symbol):
+    key = f"dividend_{symbol}"
+
+    if persist.exists(key, "html"):
+        cached = persist.load(key, "html")
+        if cached is not False:
+            print(f"[{dt.now().strftime('%Y-%m-%d %H:%M:%S')}] Using cached dividend for {symbol}")
+            return cached
+
+    try:
+        df = dividend(symbol)
+        if df.empty:
+            return wrap_html(f"<h1>No dividend history for {symbol}</h1>")
+
+        df_display = df.copy()
+        df_display.reset_index(inplace=True)
+
+        html = wrap_html(make_table(df_display), title=f"{symbol} Dividends")
+        persist.save(key, html, "html")
+        return html
+
+    except Exception as e:
+        print(f"[{dt.now().strftime('%Y-%m-%d %H:%M:%S')}] Error fetch_dividend: {e}")
+        return wrap_html(html_error(f"Dividend Error: {e}"))
+
+
+# ================================================================
+#                            SPLIT
+# ================================================================
+
+def fetch_split(symbol):
+    key = f"split_{symbol}"
+
+    if persist.exists(key, "html"):
+        cached = persist.load(key, "html")
+        if cached is not False:
+            print(f"[{dt.now().strftime('%Y-%m-%d %H:%M:%S')}] Using cached split for {symbol}")
+            return cached
+
+    try:
+        df = split(symbol)
+        if df.empty:
+            return wrap_html(f"<h1>No splits for {symbol}</h1>")
+
+        df_display = df.copy()
+        df_display.reset_index(inplace=True)
+
+        html = wrap_html(make_table(df_display), title=f"{symbol} Splits")
+        persist.save(key, html, "html")
+        return html
+
+    except Exception as e:
+        print(f"[{dt.now().strftime('%Y-%m-%d %H:%M:%S')}] Error fetch_split: {e}")
+        return wrap_html(html_error(f"Split Error: {e}"))
+
+
+# ================================================================
+#                           EARNINGS
+# ================================================================
+
+def fetch_other(symbol):
+    key = f"other_{symbol}"
+
+    if persist.exists(key, "html"):
+        cached = persist.load(key, "html")
+        if cached is not False:
+            print(f"[{dt.now().strftime('%Y-%m-%d %H:%M:%S')}] Using cached earnings for {symbol}")
+            return cached
+
+    try:
+        ticker = yf.Ticker(symbol + ".NS")
+        df = ticker.earnings
+
+        if df.empty:
+            return wrap_html(f"<h1>No earnings data for {symbol}</h1>")
+
+        df_display = df.copy()
+        df_display.reset_index(inplace=True)
+
+        html = wrap_html(make_table(df_display), title=f"{symbol} Earnings")
+        persist.save(key, html, "html")
+        return html
+
+    except Exception as e:
+        print(f"[{dt.now().strftime('%Y-%m-%d %H:%M:%S')}] Error fetch_other: {e}")
+        return wrap_html(html_error(f"Earnings Error: {e}"))