dataframe-textual 0.3.2__py3-none-any.whl → 1.5.0__py3-none-any.whl

dataframe_textual/__init__.py

@@ -2,7 +2,7 @@
 
 from .data_frame_help_panel import DataFrameHelpPanel
 from .data_frame_table import DataFrameTable, History
-from .data_frame_viewer import DataFrameViewer, _load_dataframe
+from .data_frame_viewer import DataFrameViewer
 from .table_screen import FrequencyScreen, RowDetailScreen, TableScreen
 from .yes_no_screen import (
     ConfirmScreen,
@@ -31,5 +31,4 @@ __all__ = [
     "FilterScreen",
     "FreezeScreen",
     "OpenFileScreen",
-    "_load_dataframe",
 ]

dataframe_textual/__main__.py

@@ -4,43 +4,91 @@ import argparse
 import sys
 from pathlib import Path
 
+from .common import SUPPORTED_FORMATS, load_dataframe
 from .data_frame_viewer import DataFrameViewer
 
 
-def main():
-    """Run the DataFrame Viewer application."""
+def cli() -> argparse.Namespace:
+    """Parse command-line arguments.
+
+    Determines input files or stdin and validates file existence
+    """
     parser = argparse.ArgumentParser(
-        description="Interactive CSV/Excel viewer for the terminal (Textual version)",
+        prog="dv",
+        description="Interactive terminal based viewer/editor for tabular data (e.g., CSV/Excel).",
         formatter_class=argparse.RawDescriptionHelpFormatter,
         epilog="Examples:\n"
-        "  dataframe-viewer data.csv\n"
-        "  dataframe-viewer file1.csv file2.csv file3.csv\n"
-        "  dataframe-viewer data.xlsx  (opens all sheets in tabs)\n"
-        "  cat data.csv | dataframe-viewer\n",
+        "  %(prog)s data.csv\n"
+        "  %(prog)s file1.csv file2.csv file3.csv\n"
+        "  %(prog)s data.xlsx  (opens each sheet in separate tab)\n"
+        "  cat data.csv | %(prog)s --format csv\n",
+    )
+    parser.add_argument("files", nargs="*", help="Files to view (or read from stdin)")
+    parser.add_argument(
+        "-f",
+        "--format",
+        choices=SUPPORTED_FORMATS,
+        help="Specify the format of the input files (csv, excel, tsv etc.)",
+    )
+    parser.add_argument(
+        "-H",
+        "--no-header",
+        action="store_true",
+        help="Specify that input files have no header row when reading CSV/TSV",
+    )
+    parser.add_argument(
+        "-I", "--no-inferrence", action="store_true", help="Do not infer data types when reading CSV/TSV"
     )
+    parser.add_argument("-E", "--ignore-errors", action="store_true", help="Ignore errors when reading CSV/TSV")
     parser.add_argument(
-        "files", nargs="*", help="CSV or Excel files to view (or read from stdin)"
+        "-c", "--comment-prefix", nargs="?", const="#", help="Comment lines are skipped when reading CSV/TSV"
     )
+    parser.add_argument(
+        "-q", "--quote-char", nargs="?", const=None, default='"', help="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"
+    )
+    parser.add_argument("-u", "--null", nargs="+", help="Values to interpret as null values when reading CSV/TSV")
 
     args = parser.parse_args()
-    filenames = []
+    if args.files is None:
+        args.files = []
 
     # Check if reading from stdin (pipe or redirect)
     if not sys.stdin.isatty():
-        filenames = ["-"]
-    elif args.files:
+        args.files.append("-")
+    else:
         # Validate all files exist
         for filename in args.files:
             if not Path(filename).exists():
                 print(f"File not found: {filename}")
                 sys.exit(1)
-        filenames = args.files
 
-    if not filenames:
+    if not args.files:
         parser.print_help()
         sys.exit(1)
 
-    app = DataFrameViewer(*filenames)
+    return args
+
+
+def main() -> None:
+    """Run the DataFrame Viewer application."""
+    args = cli()
+    sources = load_dataframe(
+        args.files,
+        file_format=args.format,
+        has_header=not args.no_header,
+        infer_schema=not args.no_inferrence,
+        comment_prefix=args.comment_prefix,
+        quote_char=args.quote_char,
+        skip_lines=args.skip_lines,
+        skip_rows_after_header=args.skip_rows_after_header,
+        null_values=args.null,
+        ignore_errors=args.ignore_errors,
+    )
+    app = DataFrameViewer(*sources)
     app.run()