dataframe-textual 2.2.3__py3-none-any.whl → 2.3.0__py3-none-any.whl

dataframe_textual/data_frame_table.py

@@ -1,5 +1,6 @@
 """DataFrameTable widget for displaying and interacting with Polars DataFrames."""
 
+import io
 import sys
 from collections import defaultdict, deque
 from dataclasses import dataclass
@@ -140,6 +141,7 @@ class DataFrameTable(DataTable):
         - **Ctrl+F** - 📜 Page down
         - **Ctrl+B** - 📜 Page up
         - **PgUp/PgDn** - 📜 Page up/down
+        - **&** - 📌 Mark current row as header
 
         ## ♻️ Undo/Redo/Reset
         - **u** - ↩️ Undo last action
@@ -248,6 +250,7 @@ class DataFrameTable(DataTable):
         ("comma", "toggle_thousand_separator", "Toggle thousand separator"),  # `,`
         ("underscore", "expand_column", "Expand column to full width"),  # `_`
         ("circumflex_accent", "toggle_rid", "Toggle internal row index"),  # `^`
+        ("ampersand", "set_cursor_row_as_header", "Set cursor row as the new header row"),  # `&`
         # Copy
         ("c", "copy_cell", "Copy cell to clipboard"),
         ("ctrl+c", "copy_column", "Copy column to clipboard"),
@@ -725,6 +728,10 @@ class DataFrameTable(DataTable):
         """Toggle the internal row index column visibility."""
         self.do_toggle_rid()
 
+    def action_set_cursor_row_as_header(self) -> None:
+        """Set cursor row as the new header row."""
+        self.do_set_cursor_row_as_header()
+
     def action_show_hidden_rows_columns(self) -> None:
         """Show all hidden rows/columns."""
         self.do_show_hidden_rows_columns()
@@ -1791,6 +1798,48 @@ class DataFrameTable(DataTable):
         # Recreate table for display
         self.setup_table()
 
+    def do_set_cursor_row_as_header(self) -> None:
+        """Set cursor row as the new header row."""
+        ridx = self.cursor_row_idx
+
+        # Get the new header values
+        new_header = list(self.df.row(ridx))
+        new_header[-1] = RID  # Ensure last column remains RID
+
+        # Handle duplicate column names by appending suffixes
+        seen = {}
+        for i, col in enumerate(new_header):
+            if col in seen:
+                seen[col] += 1
+                new_header[i] = f"{col}_{seen[col]}"
+            else:
+                seen[col] = 0
+
+        # Create a mapping of old column names to new column names
+        col_rename_map = {old_col: str(new_col) for old_col, new_col in zip(self.df.columns, new_header)}
+
+        # Add to history
+        self.add_history(f"Set row [$success]{ridx + 1}[/] as header", dirty=False)
+
+        # Rename columns in the dataframe
+        self.df = self.df.slice(ridx + 1).rename(col_rename_map)
+
+        # Write to string buffer
+        buffer = io.StringIO()
+        self.df.write_csv(buffer)
+
+        # Re-read with inferred schema to reset dtypes
+        buffer.seek(0)
+        self.df = pl.read_csv(buffer)
+
+        # Recreate table for display
+        self.setup_table()
+
+        # Move cursor to first column
+        self.move_cursor(row=ridx, column=0)
+
+        # self.notify(f"Set row [$success]{ridx + 1}[/] as header", title="Set Row as Header")
+
     def do_show_hidden_rows_columns(self) -> None:
         """Show all hidden rows/columns by recreating the table."""
         if not self.hidden_columns and self.df_view is None:

{dataframe_textual-2.2.3.dist-info → dataframe_textual-2.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dataframe-textual
-Version: 2.2.3
+Version: 2.3.0
 Summary: Interactive terminal viewer/editor for tabular data
 Project-URL: Homepage, https://github.com/need47/dataframe-textual
 Project-URL: Repository, https://github.com/need47/dataframe-textual.git
@@ -302,6 +302,7 @@ zcat compressed_data.csv.gz | dv -f csv
 | `_` (underscore) | Toggle column full width |
 | `z` | Freeze rows and columns |
 | `,` | Toggle thousand separator for numeric display |
+| `&` | Set current row as the new header row | 
 | `h` | Hide current column |
 | `H` | Show all hidden rows/columns |
 
@@ -696,7 +697,7 @@ Press `,` to toggle thousand separator formatting for numeric data:
 
 ### 15. Save File
 
-Press `Ctrl+S` to save filtered, edited, or sorted data back to file
+Press `Ctrl+S` to save filtered, edited, or sorted data back to file. The output format is automatically determined by the file extension, making it easy to convert between different formats (e.g., CSV to TSV).
 
 ### 16. Undo/Redo/Reset
 

{dataframe_textual-2.2.3.dist-info → dataframe_textual-2.3.0.dist-info}/RECORD

@@ -2,13 +2,13 @@ dataframe_textual/__init__.py,sha256=E53fW1spQRA4jW9grxSqPEmoe9zofzr6twdveMbt_W8
 dataframe_textual/__main__.py,sha256=xXeUA2EqVhufPkTbvv6MOCt3_ESHBH3PsCE--07a0ww,3613
 dataframe_textual/common.py,sha256=dF6trD4C60cybtOwAaeILAKXdBh6nltnkyBrh2twXo8,28632
 dataframe_textual/data_frame_help_panel.py,sha256=UEtj64XsVRdtLzuwOaITfoEQUkAfwFuvpr5Npip5WHs,3381
-dataframe_textual/data_frame_table.py,sha256=OPi9TCW_Xjj3AZJhp6bIwe538xkpC4TSU0aRlLGgMWc,149673
+dataframe_textual/data_frame_table.py,sha256=vvXgM-BDHKLCHzue3JJ5O7OKIlMj3w8WyJCVe4eAoeE,151405
 dataframe_textual/data_frame_viewer.py,sha256=aUjIk9BWYKyMG87PirFxR79iNLzkEcZ-I5XVnXwDEnU,23284
 dataframe_textual/sql_screen.py,sha256=P3j1Fv45NIKEYo9adb7NPod54FaU-djFIvCUMMHbvjY,7534
 dataframe_textual/table_screen.py,sha256=XPzJI6FXjwnxtQSMTmluygwkYM-0-Lx3v9o-MuL6bMg,19071
 dataframe_textual/yes_no_screen.py,sha256=NI7Zt3rETDWYiT5CH_FDy7sIWkZ7d7LquaZZbX79b2g,26400
-dataframe_textual-2.2.3.dist-info/METADATA,sha256=wvRLdCrQkWk-P53o-7cdSLqbdLCUvaJgLXZAXqvNi4o,29160
-dataframe_textual-2.2.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-dataframe_textual-2.2.3.dist-info/entry_points.txt,sha256=R_GoooOxcq6ab4RaHiVoZ4zrZJ-phMcGmlL2rwqncW8,107
-dataframe_textual-2.2.3.dist-info/licenses/LICENSE,sha256=AVTg0gk1X-LHI-nnHlAMDQetrwuDZK4eypgSMDO46Yc,1069
-dataframe_textual-2.2.3.dist-info/RECORD,,
+dataframe_textual-2.3.0.dist-info/METADATA,sha256=-PeWQmScG38YjVTf_grKTWx7F8tAPGiJTmczB2RMoRo,29351
+dataframe_textual-2.3.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+dataframe_textual-2.3.0.dist-info/entry_points.txt,sha256=R_GoooOxcq6ab4RaHiVoZ4zrZJ-phMcGmlL2rwqncW8,107
+dataframe_textual-2.3.0.dist-info/licenses/LICENSE,sha256=AVTg0gk1X-LHI-nnHlAMDQetrwuDZK4eypgSMDO46Yc,1069
+dataframe_textual-2.3.0.dist-info/RECORD,,