dataframe-textual 2.4.1__py3-none-any.whl → 2.9.1__py3-none-any.whl

dataframe_textual/data_frame_table.py

@@ -16,7 +16,7 @@ from textual.coordinate import Coordinate
 from textual.events import Click
 from textual.reactive import reactive
 from textual.render import measure
-from textual.widgets import DataTable, TabPane
+from textual.widgets import DataTable
 from textual.widgets._data_table import (
     CellDoesNotExist,
     CellKey,
@@ -35,7 +35,6 @@ from .common import (
     NULL_DISPLAY,
     RID,
     SUBSCRIPT_DIGITS,
-    SUPPORTED_FORMATS,
     DtypeConfig,
     format_row,
     get_next_item,
@@ -56,7 +55,6 @@ from .yes_no_screen import (
     FindReplaceScreen,
     FreezeScreen,
     RenameColumnScreen,
-    SaveFileScreen,
     SearchScreen,
 )
 
@@ -163,10 +161,21 @@ class DataFrameTable(DataTable):
         - **,** - 🔢 Toggle thousand separator for numeric display
         - **K** - 🔄 Cycle cursor (cell → row → column → cell)
 
-        ## ↕️ Sorting
-        - **[** - 🔼 Sort column ascending
-        - **]** - 🔽 Sort column descending
-        - *(Multi-column sort supported)*
+        ## ✏️ Editing
+        - **Double-click** - ✍️ Edit cell or rename column header
+        - **e** - ✍️ Edit current cell
+        - **E** - 📊 Edit entire column with expression
+        - **a** - ➕ Add empty column after current
+        - **A** - ➕ Add column with name and optional expression
+        - **@** - 🔗 Add a new link column from template
+        - **x** - ❌ Delete current row
+        - **X** - ❌ Delete row and those below
+        - **Ctrl+X** - ❌ Delete row and those above
+        - **delete** - ❌ Clear current cell (set to NULL)
+        - **Shift+Delete** - ❌ Clear current column (set matching cells to NULL)
+        - **-** - ❌ Delete current column
+        - **d** - 📋 Duplicate current column
+        - **D** - 📋 Duplicate current row
 
         ## ✅ Row Selection
         - **\\\\** - ✅ Select rows with cell matches or those matching cursor value in current column
@@ -190,28 +199,15 @@ class DataFrameTable(DataTable):
         - *(Supports case-insensitive & whole-word matching)*
 
         ## 👁️ View & Filter
-        - **"** - 📍 Filter selected rows (removes others)
-        - **v** - 👁️ View selected rows (hides others)
-        - **V** - 🔧 View selected rows matching expression (hides others)
+        - **"** - 📍 Filter selected rows (others removed)
+        - **.** - 👁️ View rows with non-null values in current column (others hidden)
+        - **v** - 👁️ View selected rows (others hidden)
+        - **V** - 🔧 View selected rows matching expression (others hidden)
 
-        ## 🔍 SQL Interface
-        - **l** - 💬 Open simple SQL interface (select columns & where clause)
-        - **L** - 🔎 Open advanced SQL interface (full SQL queries)
-
-        ## ✏️ Editing
-        - **Double-click** - ✍️ Edit cell or rename column header
-        - **e** - ✍️ Edit current cell
-        - **E** - 📊 Edit entire column with expression
-        - **a** - ➕ Add empty column after current
-        - **A** - ➕ Add column with name and optional expression
-        - **@** - 🔗 Add a new link column from template
-        - **x** - ❌ Delete current row
-        - **X** - ❌ Delete row and those below
-        - **Ctrl+X** - ❌ Delete row and those above
-        - **delete** - ❌ Clear current cell (set to NULL)
-        - **-** - ❌ Delete current column
-        - **d** - 📋 Duplicate current column
-        - **D** - 📋 Duplicate current row
+        ## ↕️ Sorting
+        - **[** - 🔼 Sort column ascending
+        - **]** - 🔽 Sort column descending
+        - *(Multi-column sort supported)*
 
         ## 🎯 Reorder
         - **Shift+↑↓** - ⬆️⬇️ Move row up/down
@@ -223,11 +219,14 @@ class DataFrameTable(DataTable):
         - **!** - ✅ Cast column to boolean
         - **$** - 📝 Cast column to string
 
-        ## 💾 Copy & Save
+        ## 💾 Copy
         - **c** - 📋 Copy cell to clipboard
         - **Ctrl+c** - 📊 Copy column to clipboard
         - **Ctrl+r** - 📝 Copy row to clipboard (tab-separated)
-        - **Ctrl+s** - 💾 Save current tab to file
+
+        ## 🔍 SQL Interface
+        - **l** - 💬 Open simple SQL interface (select columns & where clause)
+        - **L** - 🔎 Open advanced SQL interface (full SQL queries)
     """).strip()
 
     # fmt: off
@@ -255,8 +254,6 @@ class DataFrameTable(DataTable):
         ("c", "copy_cell", "Copy cell to clipboard"),
         ("ctrl+c", "copy_column", "Copy column to clipboard"),
         ("ctrl+r", "copy_row", "Copy row to clipboard"),
-        # Save
-        ("ctrl+s", "save_to_file", "Save to file"),
         # Metadata, Detail, Frequency, and Statistics
         ("m", "metadata_shape", "Show metadata for row count and column count"),
         ("M", "metadata_column", "Show metadata for column"),
@@ -268,6 +265,7 @@ class DataFrameTable(DataTable):
         ("left_square_bracket", "sort_ascending", "Sort ascending"),  # `[`
         ("right_square_bracket", "sort_descending", "Sort descending"),  # `]`
         # View & Filter
+        ("full_stop", "view_rows_non_null", "View rows with non-null values in current column"),
         ("v", "view_rows", "View selected rows"),
         ("V", "view_rows_expr", "View selected rows matching expression"),
         ("quotation_mark", "filter_rows", "Filter selected rows"),  # `"`
@@ -290,6 +288,7 @@ class DataFrameTable(DataTable):
         ("R", "replace_global", "Replace global"),  # `Shift+R`
         # Delete
         ("delete", "clear_cell", "Clear cell"),
+        ("shift+delete", "clear_column", "Clear cells in current column that match cursor value"),  # `Shift+Delete`
         ("minus", "delete_column", "Delete column"),  # `-`
         ("x", "delete_row", "Delete row"),
         ("X", "delete_row_and_below", "Delete row and those below"),
@@ -744,10 +743,6 @@ class DataFrameTable(DataTable):
         """Sort by current column in descending order."""
         self.do_sort_by_column(descending=True)
 
-    def action_save_to_file(self) -> None:
-        """Save the current dataframe to a file."""
-        self.do_save_to_file()
-
     def action_show_frequency(self) -> None:
         """Show frequency distribution for the current column."""
         self.do_show_frequency()
@@ -768,6 +763,10 @@ class DataFrameTable(DataTable):
         """Show metadata for the current column."""
         self.do_metadata_column()
 
+    def action_view_rows_non_null(self) -> None:
+        """View rows with non-null values in the current column."""
+        self.do_view_rows_non_null()
+
     def action_view_rows(self) -> None:
         """View rows by current cell value."""
         self.do_view_rows()
@@ -804,6 +803,10 @@ class DataFrameTable(DataTable):
         """Clear the current cell (set to None)."""
         self.do_clear_cell()
 
+    def action_clear_column(self) -> None:
+        """Clear cells in the current column that match the cursor value."""
+        self.do_clear_column()
+
     def action_select_row(self) -> None:
         """Select rows with cursor value in the current column."""
         self.do_select_row()
@@ -1758,13 +1761,14 @@ class DataFrameTable(DataTable):
             max_width = label_width
 
             # Scan through all loaded rows that are visible to find max width
-            for row_idx in range(self.loaded_rows):
-                cell_value = str(self.df.item(row_idx, col_idx))
-                cell_width = measure(self.app.console, cell_value, 1)
+            for row_start, row_end in self.loaded_ranges:
+                for row_idx in range(row_start, row_end):
+                    cell_value = str(self.df.item(row_idx, col_idx))
+                    cell_width = measure(self.app.console, cell_value, 1)
 
-                if cell_width > max_width:
-                    need_expand = True
-                max_width = max(max_width, cell_width)
+                    if cell_width > max_width:
+                        need_expand = True
+                        max_width = cell_width
 
             if not need_expand:
                 return
@@ -2052,25 +2056,10 @@ class DataFrameTable(DataTable):
             # Also update the view if applicable
             # Update the value of col_name in df_view using the value of col_name from df based on RID mapping between them
             if self.df_view is not None:
-                # Get updated column from df for rows that exist in df_view
-                col_updated = f"^_{col_name}_^"
-                col_exists = "^_exists_^"
-                lf_updated = self.df.lazy().select(
-                    RID, pl.col(col_name).alias(col_updated), pl.lit(True).alias(col_exists)
-                )
-                # Join and use when/then/otherwise to handle all updates including NULLs
-                self.df_view = (
-                    self.df_view.lazy()
-                    .join(lf_updated, on=RID, how="left")
-                    .with_columns(
-                        pl.when(pl.col(col_exists))
-                        .then(pl.col(col_updated))
-                        .otherwise(pl.col(col_name))
-                        .alias(col_name)
-                    )
-                    .drop(col_updated, col_exists)
-                    .collect()
-                )
+                # Get updated column from df
+                lf_updated = self.df.lazy().select(RID, pl.col(col_name))
+                # Update df_view by joining on RID
+                self.df_view = self.df_view.lazy().update(lf_updated, on=RID, include_nulls=True).collect()
         except Exception as e:
             self.notify(
                 f"Error applying expression: [$error]{term}[/] to column [$accent]{col_name}[/]",
@@ -2156,18 +2145,26 @@ class DataFrameTable(DataTable):
 
         # Update the cell to None in the dataframe
         try:
-            self.df = self.df.with_columns(
-                pl.when(pl.arange(0, len(self.df)) == ridx)
-                .then(pl.lit(None))
-                .otherwise(pl.col(col_name))
-                .alias(col_name)
+            self.df = (
+                self.df.lazy()
+                .with_columns(
+                    pl.when(pl.arange(0, len(self.df)) == ridx)
+                    .then(pl.lit(None))
+                    .otherwise(pl.col(col_name))
+                    .alias(col_name)
+                )
+                .collect()
             )
 
             # Also update the view if applicable
             if self.df_view is not None:
                 ridx_view = self.df.item(ridx, self.df.columns.index(RID))
-                self.df_view = self.df_view.with_columns(
-                    pl.when(pl.col(RID) == ridx_view).then(pl.lit(None)).otherwise(pl.col(col_name)).alias(col_name)
+                self.df_view = (
+                    self.df_view.lazy()
+                    .with_columns(
+                        pl.when(pl.col(RID) == ridx_view).then(pl.lit(None)).otherwise(pl.col(col_name)).alias(col_name)
+                    )
+                    .collect()
                 )
 
             # Update the display
@@ -2186,7 +2183,43 @@ class DataFrameTable(DataTable):
                 timeout=10,
             )
             self.log(f"Error clearing cell ({ridx}, {col_name}): {str(e)}")
-            raise e
+
+    def do_clear_column(self) -> None:
+        """Clear the current column by setting all its values to None."""
+        col_idx = self.cursor_column
+        col_name = self.cursor_col_name
+        value = self.cursor_value
+
+        # Add to history
+        self.add_history(f"Cleared column [$success]{col_name}[/]", dirty=True)
+
+        try:
+            # Update the entire column to None in the dataframe
+            self.df = (
+                self.df.lazy()
+                .with_columns(
+                    pl.when(pl.col(col_name) == value).then(pl.lit(None)).otherwise(pl.col(col_name)).alias(col_name)
+                )
+                .collect()
+            )
+
+            # Also update the view if applicable
+            if self.df_view is not None:
+                lf_updated = self.df.lazy().select(RID, pl.col(col_name))
+                self.df_view = self.df_view.lazy().update(lf_updated, on=RID, include_nulls=True).collect()
+
+            # Recreate table for display
+            self.setup_table()
+
+            # Move cursor to the cleared column
+            self.move_cursor(column=col_idx)
+
+            # self.notify(f"Cleared column [$success]{col_name}[/]", title="Clear Column")
+        except Exception as e:
+            self.notify(
+                f"Error clearing column [$error]{col_name}[/]", title="Clear Column", severity="error", timeout=10
+            )
+            self.log(f"Error clearing column `{col_name}`: {str(e)}")
 
     def do_add_column(self, col_name: str = None) -> None:
         """Add acolumn after the current column."""
@@ -2849,7 +2882,7 @@ class DataFrameTable(DataTable):
                 expr = validate_expr(term, self.df.columns, cidx)
             except Exception as e:
                 self.notify(
-                    f"Error validating expression [$error]{term}[/]", title="Search", severity="error", timeout=10
+                    f"Error validating expression [$error]{term}[/]", title="Select Row", severity="error", timeout=10
                 )
                 self.log(f"Error validating expression `{term}`: {str(e)}")
                 return
@@ -2875,7 +2908,7 @@ class DataFrameTable(DataTable):
                     expr = pl.col(col_name).cast(pl.String).str.contains(term)
                     self.notify(
                         f"Error converting [$error]{term}[/] to [$accent]{dtype}[/]. Cast to string.",
-                        title="Search",
+                        title="Select Row",
                         severity="warning",
                     )
 
@@ -2887,7 +2920,7 @@ class DataFrameTable(DataTable):
             ok_rids = set(lf.filter(expr).collect()[RID])
         except Exception as e:
             self.notify(
-                f"Error applying search filter `[$error]{term}[/]`", title="Search", severity="error", timeout=10
+                f"Error applying search filter `[$error]{term}[/]`", title="Select Row", severity="error", timeout=10
             )
             self.log(f"Error applying search filter `{term}`: {str(e)}")
             return
@@ -2896,7 +2929,7 @@ class DataFrameTable(DataTable):
         if match_count == 0:
             self.notify(
                 f"No matches found for `[$warning]{term}[/]`. Try [$accent](?i)abc[/] for case-insensitive search.",
-                title="Search",
+                title="Select Row",
                 severity="warning",
             )
             return
@@ -2906,8 +2939,8 @@ class DataFrameTable(DataTable):
         # Add to history
         self.add_history(message)
 
-        # Update selected rows to include new selections
-        self.selected_rows.update(ok_rids)
+        # Update selected rows
+        self.selected_rows = ok_rids
 
         # Show notification immediately, then start highlighting
         self.notify(message, title="Select Row")
@@ -2970,10 +3003,10 @@ class DataFrameTable(DataTable):
             # self.notify("No selections to clear", title="Clear Selections and Matches", severity="warning")
             return
 
-        row_count = len(self.selected_rows | set(self.matches.keys()))
+        # row_count = len(self.selected_rows | set(self.matches.keys()))
 
         # Add to history
-        self.add_history("Cleared all selected rows")
+        self.add_history("Cleared all selections and matches")
 
         # Clear all selections
         self.selected_rows = set()
@@ -2982,7 +3015,7 @@ class DataFrameTable(DataTable):
         # Recreate table for display
         self.setup_table()
 
-        self.notify(f"Cleared selections for [$success]{row_count}[/] rows", title="Clear Selections and Matches")
+        # self.notify(f"Cleared selections for [$success]{row_count}[/] rows", title="Clear Selections and Matches")
 
     # Find & Replace
     def find_matches(
@@ -3015,7 +3048,7 @@ class DataFrameTable(DataTable):
         else:
             columns_to_search = list(enumerate(self.df.columns))
 
-        # Search each column consistently
+        # Handle each column consistently
         for col_idx, col_name in columns_to_search:
             # Build expression based on term type
             if term == NULL:
@@ -3075,8 +3108,9 @@ class DataFrameTable(DataTable):
         cidx = self.cursor_col_idx if scope == "column" else None
 
         # Push the search modal screen
+
         self.app.push_screen(
-            SearchScreen("Find", term, self.df, cidx),
+            SearchScreen("Find" if scope == "column" else "Global Find", term, self.df, cidx),
             callback=self.find if scope == "column" else self.find_global,
         )
 
@@ -3106,10 +3140,9 @@ class DataFrameTable(DataTable):
         # Add to history
         self.add_history(f"Found `[$success]{term}[/]` in column [$accent]{col_name}[/]")
 
-        # Add to matches and count total
+        # Update matches and count total
         match_count = sum(len(cols) for cols in matches.values())
-        for rid, cols in matches.items():
-            self.matches[rid].update(cols)
+        self.matches = matches
 
         self.notify(f"Found [$success]{match_count}[/] matches for `[$accent]{term}[/]`", title="Find")
 
@@ -3140,10 +3173,9 @@ class DataFrameTable(DataTable):
         # Add to history
         self.add_history(f"Found `[$success]{term}[/]` across all columns")
 
-        # Add to matches and count total
+        # Update matches and count total
         match_count = sum(len(cols) for cols in matches.values())
-        for rid, cols in matches.items():
-            self.matches[rid].update(cols)
+        self.matches = matches
 
         self.notify(
             f"Found [$success]{match_count}[/] matches for `[$accent]{term}[/]` across all columns",
@@ -3251,7 +3283,7 @@ class DataFrameTable(DataTable):
         """Open replace screen for current column."""
         # Push the replace modal screen
         self.app.push_screen(
-            FindReplaceScreen(self, title="Find and Replace in Current Column"),
+            FindReplaceScreen(self, title="Find and Replace"),
             callback=self.replace,
         )
 
@@ -3394,18 +3426,23 @@ class DataFrameTable(DataTable):
             # Only applicable to string columns for substring matches
             if dtype == pl.String and not state.match_whole:
                 term_find = f"(?i){state.term_find}" if state.match_nocase else state.term_find
+                new_value = (
+                    pl.lit(None)
+                    if state.term_replace == NULL
+                    else pl.col(col_name).str.replace_all(term_find, state.term_replace)
+                )
                 self.df = self.df.with_columns(
-                    pl.when(mask)
-                    .then(pl.col(col_name).str.replace_all(term_find, state.term_replace))
-                    .otherwise(pl.col(col_name))
-                    .alias(col_name)
+                    pl.when(mask).then(new_value).otherwise(pl.col(col_name)).alias(col_name)
                 )
             else:
-                # Try to convert replacement value to column dtype
-                try:
-                    value = DtypeConfig(dtype).convert(state.term_replace)
-                except Exception:
-                    value = state.term_replace
+                if state.term_replace == NULL:
+                    value = None
+                else:
+                    # Try to convert replacement value to column dtype
+                    try:
+                        value = DtypeConfig(dtype).convert(state.term_replace)
+                    except Exception:
+                        value = state.term_replace
 
                 self.df = self.df.with_columns(
                     pl.when(mask).then(pl.lit(value)).otherwise(pl.col(col_name)).alias(col_name)
@@ -3413,15 +3450,8 @@ class DataFrameTable(DataTable):
 
             # Also update the view if applicable
             if self.df_view is not None:
-                col_updated = f"^_{col_name}_^"
-                lf_updated = self.df.lazy().filter(mask).select(pl.col(col_name).alias(col_updated), pl.col(RID))
-                self.df_view = (
-                    self.df_view.lazy()
-                    .join(lf_updated, on=RID, how="left")
-                    .with_columns(pl.coalesce(pl.col(col_updated), pl.col(col_name)).alias(col_name))
-                    .drop(col_updated)
-                    .collect()
-                )
+                lf_updated = self.df.lazy().filter(mask).select(pl.col(RID), pl.col(col_name))
+                self.df_view = self.df_view.lazy().update(lf_updated, on=RID, include_nulls=True).collect()
 
             state.replaced_occurrence += len(ridxs)
 
@@ -3500,9 +3530,14 @@ class DataFrameTable(DataTable):
             # Only applicable to string columns for substring matches
             if dtype == pl.String and not state.match_whole:
                 term_find = f"(?i){state.term_find}" if state.match_nocase else state.term_find
+                new_value = (
+                    pl.lit(None)
+                    if state.term_replace == NULL
+                    else pl.col(col_name).str.replace_all(term_find, state.term_replace)
+                )
                 self.df = self.df.with_columns(
                     pl.when(pl.arange(0, len(self.df)) == ridx)
-                    .then(pl.col(col_name).str.replace_all(term_find, state.term_replace))
+                    .then(new_value)
                     .otherwise(pl.col(col_name))
                     .alias(col_name)
                 )
@@ -3516,11 +3551,14 @@ class DataFrameTable(DataTable):
                         .alias(col_name)
                     )
             else:
-                # try to convert replacement value to column dtype
-                try:
-                    value = DtypeConfig(dtype).convert(state.term_replace)
-                except Exception:
-                    value = state.term_replace
+                if state.term_replace == NULL:
+                    value = None
+                else:
+                    # try to convert replacement value to column dtype
+                    try:
+                        value = DtypeConfig(dtype).convert(state.term_replace)
+                    except Exception:
+                        value = state.term_replace
 
                 self.df = self.df.with_columns(
                     pl.when(pl.arange(0, len(self.df)) == ridx)
@@ -3548,6 +3586,8 @@ class DataFrameTable(DataTable):
         if not state.done:
             # Get the new value of the current cell after replacement
             new_cell_value = self.df.item(ridx, cidx)
+            if new_cell_value is None:
+                new_cell_value = NULL_DISPLAY
             row_key = str(ridx)
             col_key = col_name
             self.update_cell(
@@ -3568,6 +3608,15 @@ class DataFrameTable(DataTable):
         self.show_next_replace_confirmation()
 
     # View & Filter
+    def do_view_rows_non_null(self) -> None:
+        """View non-null rows based on the cursor column."""
+        cidx = self.cursor_col_idx
+        col_name = self.cursor_col_name
+
+        term = pl.col(col_name).is_not_null()
+
+        self.view_rows((term, cidx, False, True))
+
     def do_view_rows(self) -> None:
         """View rows.
 
@@ -3662,6 +3711,9 @@ class DataFrameTable(DataTable):
 
         expr_str = "boolean list or series" if isinstance(expr, (list, pl.Series)) else str(expr)
 
+        # Add to history
+        self.add_history(f"Viewed rows by expression [$success]{expr_str}[/]")
+
         # Apply the filter expression
         try:
             df_filtered = lf.filter(expr).collect()
@@ -3673,12 +3725,10 @@ class DataFrameTable(DataTable):
 
         matched_count = len(df_filtered)
         if not matched_count:
+            self.histories_undo.pop()  # Remove last history entry
             self.notify(f"No rows match the expression: [$success]{expr}[/]", title="View Rows", severity="warning")
             return
 
-        # Add to history
-        self.add_history(f"Filtered by expression [$success]{expr_str}[/]")
-
         ok_rids = set(df_filtered[RID])
 
         # Create a view of self.df as a copy
@@ -3699,7 +3749,7 @@ class DataFrameTable(DataTable):
         # Recreate table for display
         self.setup_table()
 
-        self.notify(f"Filtered to [$success]{matched_count}[/] matching row(s)", title="View Rows")
+        self.notify(f"Showing [$success]{matched_count}[/] matching row(s)", title="View Rows")
 
     def do_filter_rows(self) -> None:
         """Filter rows.
@@ -3755,7 +3805,7 @@ class DataFrameTable(DataTable):
 
         self.notify(f"{message}. Now showing [$success]{len(self.df)}[/] rows.", title="Filter Rows")
 
-    # Copy & Save
+    # Copy
     def do_copy_to_clipboard(self, content: str, message: str) -> None:
         """Copy content to clipboard using pbcopy (macOS) or xclip (Linux).
 
@@ -3779,132 +3829,6 @@ class DataFrameTable(DataTable):
         except FileNotFoundError:
             self.notify("Error copying to clipboard", title="Copy to Clipboard", severity="error", timeout=10)
 
-    def do_save_to_file(self, all_tabs: bool | None = None, task_after_save: str | None = None) -> None:
-        """Open screen to save file."""
-        self._task_after_save = task_after_save
-        tab_count = len(self.app.tabs)
-        save_all = tab_count > 1 and all_tabs is not False
-
-        filepath = Path(self.filename)
-        if save_all:
-            ext = filepath.suffix.lower()
-            if ext in (".xlsx", ".xls"):
-                filename = self.filename
-            else:
-                filename = "all-tabs.xlsx"
-        else:
-            filename = str(filepath.with_stem(self.tabname))
-
-        self.app.push_screen(
-            SaveFileScreen(filename, save_all=save_all, tab_count=tab_count),
-            callback=self.save_to_file,
-        )
-
-    def save_to_file(self, result) -> None:
-        """Handle result from SaveFileScreen."""
-        if result is None:
-            return
-        filename, save_all, overwrite_prompt = result
-        self._save_all = save_all
-
-        # Check if file exists
-        if overwrite_prompt and Path(filename).exists():
-            self._pending_filename = filename
-            self.app.push_screen(
-                ConfirmScreen("File already exists. Overwrite?"),
-                callback=self.confirm_overwrite,
-            )
-        else:
-            self.save_file(filename)
-
-    def confirm_overwrite(self, should_overwrite: bool) -> None:
-        """Handle result from ConfirmScreen."""
-        if should_overwrite:
-            self.save_file(self._pending_filename)
-        else:
-            # Go back to SaveFileScreen to allow user to enter a different name
-            self.app.push_screen(
-                SaveFileScreen(self._pending_filename, save_all=self._save_all),
-                callback=self.save_to_file,
-            )
-
-    def save_file(self, filename: str) -> None:
-        """Actually save the dataframe to a file."""
-        filepath = Path(filename)
-        ext = filepath.suffix.lower()
-        if ext == ".gz":
-            ext = Path(filename).with_suffix("").suffix.lower()
-
-        fmt = ext.removeprefix(".")
-        if fmt not in SUPPORTED_FORMATS:
-            self.notify(
-                f"Unsupported file format [$success]{fmt}[/]. Use [$accent]CSV[/] as fallback. Supported formats: {', '.join(SUPPORTED_FORMATS)}",
-                title="Save to File",
-                severity="warning",
-            )
-            fmt = "csv"
-
-        df = (self.df if self.df_view is None else self.df_view).select(pl.exclude(RID))
-        try:
-            if fmt == "csv":
-                df.write_csv(filename)
-            elif fmt in ("tsv", "tab"):
-                df.write_csv(filename, separator="\t")
-            elif fmt in ("xlsx", "xls"):
-                self.save_excel(filename)
-            elif fmt == "json":
-                df.write_json(filename)
-            elif fmt == "ndjson":
-                df.write_ndjson(filename)
-            elif fmt == "parquet":
-                df.write_parquet(filename)
-            else:  # Fallback to CSV
-                df.write_csv(filename)
-
-            # Update current filename
-            self.filename = filename
-
-            # Reset dirty flag after save
-            if self._save_all:
-                tabs: dict[TabPane, DataFrameTable] = self.app.tabs
-                for table in tabs.values():
-                    table.dirty = False
-            else:
-                self.dirty = False
-
-            if hasattr(self, "_task_after_save"):
-                if self._task_after_save == "close_tab":
-                    self.app.do_close_tab()
-                elif self._task_after_save == "quit_app":
-                    self.app.exit()
-
-            # From ConfirmScreen callback, so notify accordingly
-            if self._save_all:
-                self.notify(f"Saved all tabs to [$success]{filename}[/]", title="Save to File")
-            else:
-                self.notify(f"Saved current tab to [$success]{filename}[/]", title="Save to File")
-
-        except Exception as e:
-            self.notify(f"Error saving [$error]{filename}[/]", title="Save to File", severity="error", timeout=10)
-            self.log(f"Error saving file `{filename}`: {str(e)}")
-
-    def save_excel(self, filename: str) -> None:
-        """Save to an Excel file."""
-        import xlsxwriter
-
-        if not self._save_all or len(self.app.tabs) == 1:
-            # Single tab - save directly
-            df = (self.df if self.df_view is None else self.df_view).select(pl.exclude(RID))
-            df.write_excel(filename, worksheet=self.tabname)
-        else:
-            # Multiple tabs - use xlsxwriter to create multiple sheets
-            with xlsxwriter.Workbook(filename) as wb:
-                tabs: dict[TabPane, DataFrameTable] = self.app.tabs
-                for table in tabs.values():
-                    worksheet = wb.add_worksheet(table.tabname)
-                    df = (table.df if table.df_view is None else table.df_view).select(pl.exclude(RID))
-                    df.write_excel(workbook=wb, worksheet=worksheet)
-
     # SQL Interface
     def do_simple_sql(self) -> None:
         """Open the SQL interface screen."""