snowglobe-cli 0.1.9__tar.gz → 0.2.1__tar.gz

{snowglobe_cli-0.1.9 → snowglobe_cli-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowglobe-cli
-Version: 0.1.9
+Version: 0.2.1
 Summary: Explainable cost and access visibility for Snowflake
 Author-email: Jaryd Thornton <jaryd90@gmail.com>
 License-Expression: Apache-2.0

{snowglobe_cli-0.1.9 → snowglobe_cli-0.2.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "snowglobe-cli"
-version = "0.1.9"
+version = "0.2.1"
 description = "Explainable cost and access visibility for Snowflake"
 readme = "README.md"
 requires-python = ">=3.12.3"

{snowglobe_cli-0.1.9 → snowglobe_cli-0.2.1}/snowglobe/core/cost_service.py

@@ -951,3 +951,23 @@ class CostService:
         except Exception:
             # View may not exist if no MVs are used
             return pd.DataFrame(), None
+
+    def get_day_detail(self, date: str) -> pd.DataFrame:
+        """Warehouse credit breakdown for a single calendar day."""
+        conn = self.context.connect()
+        with conn:
+            rows = conn.query(f"""
+                SELECT WAREHOUSE_NAME,
+                       ROUND(SUM(CREDITS_USED), 4) AS CREDITS
+                FROM SNOWFLAKE.ACCOUNT_USAGE.WAREHOUSE_METERING_HISTORY
+                WHERE START_TIME::DATE = '{date}'
+                  AND WAREHOUSE_NAME IS NOT NULL
+                GROUP BY 1
+                ORDER BY 2 DESC
+                LIMIT 30
+            """)
+        if not rows:
+            return pd.DataFrame(columns=["WAREHOUSE_NAME", "CREDITS"])
+        df = pd.DataFrame(rows)
+        df["CREDITS"] = df["CREDITS"].astype(float)
+        return df

{snowglobe_cli-0.1.9 → snowglobe_cli-0.2.1}/snowglobe/core/report_service.py

@@ -37,8 +37,15 @@ class ReportService:
         ai_df, _, _ = self.cost_service.get_ai_costs(days)
         ai_total = float(ai_df["TOTAL_CREDITS"].sum()) if not ai_df.empty else 0
 
-        # Storage
+        # AI costs by user — exclude zero-cost users
+        ai_users_df, _, _ = self.cost_service.get_ai_costs_by_user(days)
+        if not ai_users_df.empty:
+            ai_users_df = ai_users_df[ai_users_df["TOTAL_CREDITS"] > 0].reset_index(drop=True)
+
+        # Storage — exclude databases with no stored data
         storage_df, _ = self.cost_service.get_storage_usage(days)
+        if not storage_df.empty:
+            storage_df = storage_df[storage_df["TOTAL_TB"] > 0].reset_index(drop=True)
         storage_total_tb = float(storage_df["TOTAL_TB"].sum()) if not storage_df.empty else 0
         storage_total_cost = float(storage_df["EST_MONTHLY_COST"].sum()) if not storage_df.empty else 0
         storage_rate = self.cost_service.get_storage_rate()
@@ -56,6 +63,7 @@ class ReportService:
             "cost_total": cost_total,
             "ai_costs": ai_df.to_dict("records") if not ai_df.empty else [],
             "ai_total": ai_total,
+            "ai_users": ai_users_df.to_dict("records") if not ai_users_df.empty else [],
             "storage": storage_df.to_dict("records") if not storage_df.empty else [],
             "storage_total_tb": storage_total_tb,
             "storage_total_cost": storage_total_cost,

{snowglobe_cli-0.1.9 → snowglobe_cli-0.2.1}/snowglobe/templates/report.md.j2

@@ -30,6 +30,18 @@
 
 ---
 
+## AI Costs by User
+
+| User | Functions | Analyst | Agent | Code | Intelligence | Total |
+|---|---:|---:|---:|---:|---:|---:|
+{% for row in ai_users -%}
+| {{ row.USER_NAME }} | {{ "%.2f"|format(row.CORTEX_FUNCTIONS) }} | {{ "%.2f"|format(row.CORTEX_ANALYST) }} | {{ "%.2f"|format(row.CORTEX_AGENT) }} | {{ "%.2f"|format(row.CORTEX_CODE) }} | {{ "%.2f"|format(row.SNOWFLAKE_INTELLIGENCE) }} | {{ "%.2f"|format(row.TOTAL_CREDITS) }} |
+{% else -%}
+*No AI usage in this period.*
+{% endfor -%}
+
+---
+
 ## Storage
 
 | Database | Size (TB) | Est. Monthly Cost |

{snowglobe_cli-0.1.9 → snowglobe_cli-0.2.1}/snowglobe/tui/screens/cost.py

@@ -38,11 +38,12 @@ class CostScreen(Vertical):
 
     # Active view: "summary" | "trend" | "top_queries" | "warehouses" | "users"
     # | "ai" | "ai_users" | "services" | "storage" | "replication" | "mv" | "budget"
-    # | "drill_service" | "drill_warehouse" | "drill_user"
+    # | "drill_service" | "drill_warehouse" | "drill_user" | "drill_day"
     _current_view: str = "summary"
     _drill_service: str | None = None
     _drill_warehouse: str | None = None
     _drill_user: str | None = None
+    _drill_day: str | None = None
     # Becomes True after the user explicitly picks a view, so Select.Changed
     # events fired during the initial mount don't auto-trigger a Snowflake fetch.
     _user_initiated: bool = False
@@ -136,6 +137,9 @@ class CostScreen(Vertical):
         elif self._current_view == "users":
             self._drill_user = key
             self._fetch_user_drill(key)
+        elif self._current_view == "trend":
+            self._drill_day = key
+            self._fetch_day_drill(key)
         elif self._current_view == "top_queries":
             # Hand off to Tune via the App's content switcher.
             try:
@@ -179,6 +183,8 @@ class CostScreen(Vertical):
             self._fetch_warehouse_drill(self._drill_warehouse)
         elif self._current_view == "drill_user" and self._drill_user:
             self._fetch_user_drill(self._drill_user)
+        elif self._current_view == "drill_day" and self._drill_day:
+            self._fetch_day_drill(self._drill_day)
 
     def action_back_from_drill(self) -> None:
         if self._current_view == "drill_service":
@@ -190,6 +196,9 @@ class CostScreen(Vertical):
         elif self._current_view == "drill_user":
             self._drill_user = None
             self._fetch_users()
+        elif self._current_view == "drill_day":
+            self._drill_day = None
+            self._fetch_trend()
 
     # --- Days helper --------------------------------------------------
 
@@ -268,7 +277,7 @@ class CostScreen(Vertical):
         avg = float(df["CREDITS"].mean())
         suffix = f"  ·  cached {cache_age}m ago" if cache_age else ""
         self._set_status(
-            f"Daily trend — {days}d  ·  total {total:,.2f}  ·  avg/day {avg:,.2f}{suffix}"
+            f"Daily trend — {days}d  ·  total {total:,.2f}  ·  avg/day {avg:,.2f}{suffix}  ·  click a date to drill in"
         )
 
         table = self.query_one(DataTable)
@@ -536,6 +545,45 @@ class CostScreen(Vertical):
                 avg_str,
             )
 
+    # --- Drill: day detail -------------------------------------------
+
+    def _fetch_day_drill(self, date: str) -> None:
+        self._current_view = "drill_day"
+        self._set_status(f"Loading breakdown for {date}…  Esc to return")
+        self._day_drill_worker(date=date)
+
+    @work(thread=True, exclusive=True, group="cost")
+    def _day_drill_worker(self, *, date: str) -> None:
+        try:
+            df = self.app.get_cost_service().get_day_detail(date)
+        except Exception as e:
+            self.app.call_from_thread(self._fetch_failed, e)
+            return
+        self.app.call_from_thread(self._render_day_drill, df, date)
+
+    def _render_day_drill(self, df: pd.DataFrame, date: str) -> None:
+        self._reset_table()
+        table = self.query_one(DataTable)
+        if df is None or df.empty:
+            table.add_columns("(no warehouse activity on this date)")
+            self._set_status(f"{date}  ·  no data  ·  Esc to return")
+            return
+        total = float(df["CREDITS"].sum())
+        max_c = float(df["CREDITS"].max()) if total > 0 else 1.0
+        table.add_columns("WAREHOUSE", "CREDITS", "% OF DAY", "BAR")
+        for _, row in df.iterrows():
+            credits = float(row["CREDITS"])
+            pct = (credits / total * 100) if total > 0 else 0
+            table.add_row(
+                str(row["WAREHOUSE_NAME"]),
+                f"{credits:,.4f}",
+                f"{pct:.1f}%",
+                _bar(credits / max_c),
+            )
+        self._set_status(
+            f"Day detail — {date}  ·  {total:,.4f} credits  ·  Esc to return"
+        )
+
     # --- View 6: AI services -----------------------------------------
 
     def _fetch_ai(self, force: bool = False) -> None:

{snowglobe_cli-0.1.9 → snowglobe_cli-0.2.1}/snowglobe/tui/screens/home.py

@@ -1,13 +1,8 @@
 """Home — landing dashboard.
 
-Three KPI cards (Cache / Connection / This week) at the top, quick-action
-hotkeys + recent expensive queries DataTable below. KPI data comes from
-local SQLite (instant); recent queries fire one `get_top_queries(7, 5)`
-worker on mount.
-
-Hotkeys jump to common workflows:
-  a  Access check        c  Cost summary        r  Refresh cache
-  w  Who-access          s  Risk scan
+Three KPI cards (Cache / Connection / This week) at the top.
+Below: 30d spend trend (left) and recent expensive queries (right).
+Both panels fire Snowflake workers on mount; KPI data is local/instant.
 """
 from datetime import datetime, timezone
 
@@ -48,16 +43,10 @@ class HomeScreen(Vertical):
                 yield Static("loading…",   id="kpi-week", classes="kpi-body")
 
         with Horizontal(id="home-body"):
-            with Vertical(id="home-actions", classes="panel"):
-                yield Static("Quick actions", classes="panel-title")
-                yield Static(
-                    "[a]  Access check\n"
-                    "[w]  Who-access\n"
-                    "[c]  Cost\n"
-                    "\\[s]  Risk scan\n"
-                    "\\[r]  Refresh\n",
-                    classes="actions-list",
-                )
+            with Vertical(id="home-trend", classes="panel"):
+                yield Static("30d spend trend", classes="panel-title")
+                yield DataTable(id="home-trend-table",
+                                cursor_type="row", zebra_stripes=True)
             with Vertical(id="home-queries", classes="panel"):
                 yield Static("Recent expensive queries (7d)", classes="panel-title")
                 yield DataTable(id="home-queries-table",
@@ -69,7 +58,8 @@ class HomeScreen(Vertical):
         self._refresh_kpis()
         # Re-render KPIs whenever the app-level cache age ticks (every 30s).
         self.watch(self.app, "cache_age_minutes", lambda _v: self._refresh_kpis())
-        # Fire one query fetch when Home first mounts.
+        # Fire both Snowflake workers on mount (different groups → concurrent).
+        self._fetch_trend()
         self._fetch_queries()
 
     # --- Hotkey action -----------------------------------------------
@@ -172,6 +162,40 @@ class HomeScreen(Vertical):
         )
         self.query_one("#kpi-week", Static).update(text)
 
+    # --- 30d spend trend (one Snowflake call on mount) ---------------
+
+    @work(thread=True, exclusive=True, group="trend")
+    def _fetch_trend(self) -> None:
+        try:
+            df, _ = self.app.get_cost_service().get_daily_trend(days=30)
+        except Exception as e:
+            self.app.call_from_thread(self._trend_failed, e)
+            return
+        self.app.call_from_thread(self._render_trend, df)
+
+    def _trend_failed(self, err: Exception) -> None:
+        table = self.query_one("#home-trend-table", DataTable)
+        table.clear(columns=True)
+        table.add_columns(f"(could not load — {err})")
+
+    def _render_trend(self, df: pd.DataFrame) -> None:
+        table = self.query_one("#home-trend-table", DataTable)
+        table.clear(columns=True)
+        if df is None or df.empty:
+            table.add_columns("(no trend data)")
+            return
+        table.add_columns("DATE", "CREDITS", "TREND")
+        max_c = float(df["CREDITS"].max()) if not df.empty else 1.0
+        for _, row in df.iterrows():
+            credits = float(row["CREDITS"])
+            bar_width = max(1, int(credits / max_c * 20)) if max_c > 0 else 0
+            table.add_row(
+                str(row["DATE"]),
+                f"{credits:,.2f}",
+                "█" * bar_width,
+                key=str(row["DATE"]),
+            )
+
     # --- Recent expensive queries (one Snowflake call on mount) -----
 
     @work(thread=True, exclusive=True, group="cost")
@@ -206,17 +230,27 @@ class HomeScreen(Vertical):
             )
 
     def on_data_table_row_selected(self, event: DataTable.RowSelected) -> None:
-        # Drill: open the query in the Tune screen.
         key = event.row_key.value if event.row_key else None
         if not key:
             return
-        try:
-            from textual.widgets import ContentSwitcher, Input
-            from snowglobe.tui.screens.tune import TuneScreen
-            switcher = self.app.query_one(ContentSwitcher)
-            switcher.current = "tune"
-            tune = self.app.query_one(TuneScreen)
-            tune.query_one("#tu-query-id", Input).value = key
-            self.app.notify(f"Loaded {key[:24]}… into Tune. Press Analyse.", timeout=4)
-        except Exception:
-            self.app.notify(f"Query: {key[:24]}…", timeout=4)
+
+        if event.data_table.id == "home-trend-table":
+            # Drill into that day on the Cost screen.
+            try:
+                from textual.widgets import ContentSwitcher
+                from snowglobe.tui.screens.cost import CostScreen
+                self.app.query_one(ContentSwitcher).current = "cost"
+                self.app.query_one(CostScreen)._fetch_day_drill(key)
+            except Exception as e:
+                self.app.notify(f"Could not open day detail: {e}", severity="warning", timeout=4)
+        else:
+            # Queries table — open in Tune.
+            try:
+                from textual.widgets import ContentSwitcher, Input
+                from snowglobe.tui.screens.tune import TuneScreen
+                self.app.query_one(ContentSwitcher).current = "tune"
+                tune = self.app.query_one(TuneScreen)
+                tune.query_one("#tu-query-id", Input).value = key
+                self.app.notify(f"Loaded {key[:24]}… into Tune. Press Analyse.", timeout=4)
+            except Exception:
+                self.app.notify(f"Query: {key[:24]}…", timeout=4)

{snowglobe_cli-0.1.9 → snowglobe_cli-0.2.1}/snowglobe/tui/screens/reports.py

@@ -14,7 +14,7 @@ from typing import Any
 
 from textual import work
 from textual.app import ComposeResult
-from textual.containers import Horizontal, Vertical
+from textual.containers import Horizontal, Vertical, VerticalScroll
 from textual.suggester import SuggestFromList
 from textual.widgets import Button, Input, Markdown, Select, Static
 
@@ -50,11 +50,11 @@ class ReportsScreen(Vertical):
                 yield Select(_TYPES, value="full", id="rp-type",
                              classes="form-select", allow_blank=False)
 
-            with Horizontal(classes="form-row"):
+            with Horizontal(classes="form-row", id="rp-days-row"):
                 yield Static("Window:", classes="form-label")
                 yield Input(value="30", id="rp-days", classes="form-input")
 
-            with Horizontal(classes="form-row"):
+            with Horizontal(classes="form-row", id="rp-top-row"):
                 yield Static("Top queries:", classes="form-label")
                 yield Input(value="10", id="rp-top", classes="form-input")
 
@@ -76,15 +76,16 @@ class ReportsScreen(Vertical):
             "Preview will appear here after Generate. Save writes to the output path.",
             id="rp-status", classes="hint",
         )
-        yield Markdown(
-            "No report yet — fill the form and click Generate.",
-            id="rp-preview",
-        )
+        with VerticalScroll(id="rp-preview-scroll"):
+            yield Markdown(
+                "No report yet — fill the form and click Generate.",
+                id="rp-preview",
+            )
 
     # --- Lifecycle ----------------------------------------------------
 
     def on_mount(self) -> None:
-        # Username suggester from user_graph (only useful for user reports).
+        self._update_form_visibility("full")
         self.watch(self.app, "user_graph",
                    lambda _v: self._refresh_user_suggester())
 
@@ -103,9 +104,16 @@ class ReportsScreen(Vertical):
 
     def on_select_changed(self, event: Select.Changed) -> None:
         if event.select.id == "rp-type":
-            # Update the default output path to match the type
-            new_path = _default_path(str(event.value))
-            self.query_one("#rp-path", Input).value = new_path
+            rtype = str(event.value)
+            self.query_one("#rp-path", Input).value = _default_path(rtype)
+            self._update_form_visibility(rtype)
+
+    def _update_form_visibility(self, rtype: str) -> None:
+        is_user = rtype == "user"
+        is_full = rtype == "full"
+        self.query_one("#rp-days-row").display = not is_user
+        self.query_one("#rp-top-row").display = is_full
+        self.query_one("#rp-user-row").display = is_user
 
     def on_button_pressed(self, event: Button.Pressed) -> None:
         if event.button.id == "rp-generate":

{snowglobe_cli-0.1.9 → snowglobe_cli-0.2.1}/snowglobe/tui/screens/tune.py

@@ -17,7 +17,7 @@ from rich.syntax import Syntax
 
 from textual import work
 from textual.app import ComposeResult
-from textual.containers import Horizontal, Vertical
+from textual.containers import Horizontal, Vertical, VerticalScroll
 from textual.widgets import (
     Button, DataTable, Input, Markdown, Static, TabbedContent, TabPane, Tree,
 )
@@ -46,13 +46,15 @@ class TuneScreen(Vertical):
         with Horizontal(id="tu-body"):
             with Vertical(id="tu-sql-pane", classes="panel"):
                 yield Static("SQL", classes="panel-title")
-                yield Static("Run an analysis to see SQL here.",
-                             id="tu-sql", classes="sql-pane")
+                with VerticalScroll(id="tu-sql-scroll"):
+                    yield Static("Run an analysis to see SQL here.",
+                                 id="tu-sql", classes="sql-pane")
             with Vertical(id="tu-analysis-pane", classes="panel"):
                 yield Static("Analysis", classes="panel-title")
                 with TabbedContent(initial="tu-tab-heuristics"):
                     with TabPane("Heuristics", id="tu-tab-heuristics"):
-                        yield Static("Pending.", id="tu-heuristics")
+                        with VerticalScroll(id="tu-heuristics-scroll"):
+                            yield Static("Pending.", id="tu-heuristics")
                     with TabPane("Insights", id="tu-tab-insights"):
                         yield DataTable(id="tu-insights-table",
                                         cursor_type="row", zebra_stripes=True)
@@ -65,11 +67,12 @@ class TuneScreen(Vertical):
                         with Horizontal(classes="actions-row"):
                             yield Button("Generate AI suggestion (Cortex)",
                                          id="tu-ai-run", variant="warning")
-                        yield Markdown(
-                            "AI suggestions will appear here after you Analyse "
-                            "and then click Generate.",
-                            id="tu-ai",
-                        )
+                        with VerticalScroll(id="tu-ai-scroll"):
+                            yield Markdown(
+                                "AI suggestions will appear here after you Analyse "
+                                "and then click Generate.",
+                                id="tu-ai",
+                            )
 
     # --- Events -------------------------------------------------------
 

{snowglobe_cli-0.1.9 → snowglobe_cli-0.2.1}/snowglobe/tui/styles.tcss

@@ -365,16 +365,22 @@ Tree > .tree--label-highlighted {
     margin-right: 1;
 }
 
-#tu-sql {
+#tu-sql-scroll {
     height: 1fr;
+}
+
+#tu-sql {
     background: $background;
     padding: 1;
 }
 
+#tu-heuristics-scroll {
+    height: 1fr;
+}
+
 #tu-heuristics {
     padding: 1;
     color: $foreground;
-    height: 1fr;
 }
 
 #tu-insights-table,
@@ -388,11 +394,14 @@ Tree > .tree--label-highlighted {
     height: 1fr;
 }
 
+#tu-ai-scroll {
+    height: 1fr;
+}
+
 #tu-ai {
     background: $background;
     color: $foreground;
     padding: 1;
-    height: 1fr;
 }
 
 /* --- Reports screen --- */
@@ -402,8 +411,11 @@ Tree > .tree--label-highlighted {
     margin-bottom: 1;
 }
 
-#rp-preview {
+#rp-preview-scroll {
     height: 1fr;
+}
+
+#rp-preview {
     background: $background;
     border: round $primary;
     padding: 1;

{snowglobe_cli-0.1.9 → snowglobe_cli-0.2.1}/snowglobe_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowglobe-cli
-Version: 0.1.9
+Version: 0.2.1
 Summary: Explainable cost and access visibility for Snowflake
 Author-email: Jaryd Thornton <jaryd90@gmail.com>
 License-Expression: Apache-2.0