dataframe-textual 2.6.0__py3-none-any.whl → 2.7.0__py3-none-any.whl

dataframe_textual/__main__.py

@@ -51,15 +51,34 @@ def cli() -> argparse.Namespace:
     )
     parser.add_argument("-E", "--ignore-errors", action="store_true", help="Ignore errors when reading CSV/TSV")
     parser.add_argument(
-        "-c", "--comment-prefix", nargs="?", const="#", help="Comment lines are skipped when reading CSV/TSV"
+        "-c",
+        "--comment-prefix",
+        metavar="PREFIX",
+        nargs="?",
+        const="#",
+        help="Comment lines starting with `PREFIX` are skipped when reading CSV/TSV",
     )
     parser.add_argument(
-        "-q", "--quote-char", nargs="?", const=None, default='"', help="Quote character for reading CSV/TSV"
+        "-q",
+        "--quote-char",
+        metavar="C",
+        nargs="?",
+        const=None,
+        default='"',
+        help="Use `C` as quote character for reading CSV/TSV",
     )
-    parser.add_argument("-l", "--skip-lines", type=int, default=0, help="Skip lines when reading CSV/TSV")
     parser.add_argument(
-        "-a", "--skip-rows-after-header", type=int, default=0, help="Skip rows after header when reading CSV/TSV"
+        "-L", "--skip-lines", metavar="N", type=int, default=0, help="Skip first N lines when reading CSV/TSV"
     )
+    parser.add_argument(
+        "-A",
+        "--skip-rows-after-header",
+        metavar="N",
+        type=int,
+        default=0,
+        help="Skip N rows after header when reading CSV/TSV",
+    )
+    parser.add_argument("-N", "--n-rows", metavar="N", type=int, help="Stop after reading N rows from CSV/TSV")
     parser.add_argument("-n", "--null", nargs="+", help="Values to interpret as null values when reading CSV/TSV")
 
     args = parser.parse_args()
@@ -98,6 +117,7 @@ def main() -> None:
         null_values=args.null,
         ignore_errors=args.ignore_errors,
         truncate_ragged_lines=args.truncate_ragged_lines,
+        n_rows=args.n_rows,
     )
     app = DataFrameViewer(*sources)
     app.run()

dataframe_textual/common.py

@@ -485,6 +485,7 @@ def load_dataframe(
     null_values: list[str] | None = None,
     ignore_errors: bool = False,
     truncate_ragged_lines: bool = False,
+    n_rows: int | None = None,
 ) -> list[Source]:
     """Load DataFrames from file specifications.
 
@@ -502,6 +503,8 @@ def load_dataframe(
         skip_rows_after_header: Number of rows to skip after header. Defaults to 0.
         null_values: List of values to interpret as null when reading CSV/TSV files. Defaults to None.
         ignore_errors: Whether to ignore errors when reading CSV/TSV files. Defaults to False.
+        truncate_ragged_lines: Whether to truncate ragged lines when reading CSV/TSV files. Defaults to False.
+        n_rows: Number of rows to read from CSV/TSV files. Defaults to None (read all rows).
 
     Returns:
         List of `Source` objects.
@@ -551,6 +554,7 @@ def load_dataframe(
                 null_values=null_values,
                 ignore_errors=ignore_errors,
                 truncate_ragged_lines=truncate_ragged_lines,
+                n_rows=n_rows,
             )
         )
 
@@ -635,6 +639,7 @@ def load_file(
     null_values: list[str] | None = None,
     ignore_errors: bool = False,
     truncate_ragged_lines: bool = False,
+    n_rows: int | None = None,
 ) -> list[Source]:
     """Load a single file.
 
@@ -660,6 +665,8 @@ def load_file(
         schema_overrides: Optional dictionary of column name to Polars data type to override inferred schema.
         null_values: List of values to interpret as null when reading CSV/TSV files. Defaults to None.
         ignore_errors: Whether to ignore errors when reading CSV/TSV files.
+        truncate_ragged_lines: Whether to truncate ragged lines when reading CSV/TSV files. Defaults to False.
+        n_rows: Number of rows to read from CSV/TSV files. Defaults to None (read all rows).
 
     Returns:
         List of `Source` objects.
@@ -684,6 +691,7 @@ def load_file(
             null_values=null_values,
             ignore_errors=ignore_errors,
             truncate_ragged_lines=truncate_ragged_lines,
+            n_rows=n_rows,
         )
         data.append(Source(lf, filename, filepath.stem))
     elif file_format in ("xlsx", "xls"):
@@ -740,6 +748,8 @@ def load_file(
             schema_overrides=schema_overrides,
             null_values=null_values,
             ignore_errors=ignore_errors,
+            truncate_ragged_lines=truncate_ragged_lines,
+            n_rows=n_rows,
         )
 
     return data

dataframe_textual/data_frame_table.py

@@ -1755,13 +1755,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
@@ -2198,9 +2199,8 @@ class DataFrameTable(DataTable):
 
             # Also update the view if applicable
             if self.df_view is not None:
-                self.df_view = self.df_view.with_columns(
-                    pl.when(pl.col(col_name) == value).then(pl.lit(None)).otherwise(pl.col(col_name)).alias(col_name)
-                )
+                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()
@@ -2997,10 +2997,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()
@@ -3009,7 +3009,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(

{dataframe_textual-2.6.0.dist-info → dataframe_textual-2.7.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dataframe-textual
-Version: 2.6.0
+Version: 2.7.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
@@ -184,14 +184,15 @@ options:
   -t, --truncate-ragged-lines
                         Truncate ragged lines when reading CSV/TSV
   -E, --ignore-errors   Ignore errors when reading CSV/TSV
-  -c, --comment-prefix [COMMENT_PREFIX]
-                        Comment lines are skipped when reading CSV/TSV
-  -q, --quote-char [QUOTE_CHAR]
-                        Quote character for reading CSV/TSV
-  -l, --skip-lines SKIP_LINES
-                        Skip lines when reading CSV/TSV
-  -a, --skip-rows-after-header SKIP_ROWS_AFTER_HEADER
-                        Skip rows after header when reading CSV/TSV
+  -c, --comment-prefix [PREFIX]
+                        Comment lines starting with `PREFIX` are skipped when reading CSV/TSV
+  -q, --quote-char [C]
+                        Use `C` as quote character for reading CSV/TSV
+  -L, --skip-lines N
+                        Skip first N lines when reading CSV/TSV
+  -A, --skip-rows-after-header N
+                        Skip N rows after header when reading CSV/TSV
+  -N, --n-rows N        Stop after reading N rows from CSV/TSV
   -n, --null NULL [NULL ...]
                         Values to interpret as null values when reading CSV/TSV
 ```

{dataframe_textual-2.6.0.dist-info → dataframe_textual-2.7.0.dist-info}/RECORD

@@ -1,14 +1,14 @@
 dataframe_textual/__init__.py,sha256=E53fW1spQRA4jW9grxSqPEmoe9zofzr6twdveMbt_W8,1310
-dataframe_textual/__main__.py,sha256=xXeUA2EqVhufPkTbvv6MOCt3_ESHBH3PsCE--07a0ww,3613
-dataframe_textual/common.py,sha256=WwCUnG5MReUz6yfjHPL1mrwgDddCCJmIbqk_hrRduYU,28461
+dataframe_textual/__main__.py,sha256=tJ6FjjV25ZQzaMdqD5XcDVRZfj8l6kgGvXyrn975rjo,3999
+dataframe_textual/common.py,sha256=CNRdHP3N1li2dy9OsTiW-zfpzf8zcrt2fW8mmYY-YVA,29073
 dataframe_textual/data_frame_help_panel.py,sha256=UEtj64XsVRdtLzuwOaITfoEQUkAfwFuvpr5Npip5WHs,3381
-dataframe_textual/data_frame_table.py,sha256=DwAOhuXeu3ZW82GTE8pYsAI5QXjqZ-IulwaRJLGJSC8,147733
+dataframe_textual/data_frame_table.py,sha256=IjsFc7aJJMAhfdf-898cTCMeDBvtnnDeRRV4cIdZQXY,147806
 dataframe_textual/data_frame_viewer.py,sha256=_VwbCcRBgdTcrZmgS2mRwIJ-cFxOeJ55twDFvQUHMfk,28723
 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.6.0.dist-info/METADATA,sha256=Acv3XFju3na9yBqWynLq_4nQ0jXnU0U1dO-U78rzZCE,29494
-dataframe_textual-2.6.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-dataframe_textual-2.6.0.dist-info/entry_points.txt,sha256=R_GoooOxcq6ab4RaHiVoZ4zrZJ-phMcGmlL2rwqncW8,107
-dataframe_textual-2.6.0.dist-info/licenses/LICENSE,sha256=AVTg0gk1X-LHI-nnHlAMDQetrwuDZK4eypgSMDO46Yc,1069
-dataframe_textual-2.6.0.dist-info/RECORD,,
+dataframe_textual-2.7.0.dist-info/METADATA,sha256=x5HmSx1U443HzBmNdKt0W3raGzcVTXFR4uaJXbKqKbE,29554
+dataframe_textual-2.7.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+dataframe_textual-2.7.0.dist-info/entry_points.txt,sha256=R_GoooOxcq6ab4RaHiVoZ4zrZJ-phMcGmlL2rwqncW8,107
+dataframe_textual-2.7.0.dist-info/licenses/LICENSE,sha256=AVTg0gk1X-LHI-nnHlAMDQetrwuDZK4eypgSMDO46Yc,1069
+dataframe_textual-2.7.0.dist-info/RECORD,,