treex-cli 0.1.6__tar.gz → 0.1.8__tar.gz

{treex_cli-0.1.6/treex_cli.egg-info → treex_cli-0.1.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: treex-cli
-Version: 0.1.6
+Version: 0.1.8
 Summary: List directory contents as a tree with file and directory metadata
 Author: Corey Goldberg
 Maintainer: Corey Goldberg
@@ -100,7 +100,7 @@ Install `treex` from [PyPI][pypi-home] using either `pip` or `pipx`:
 
 ```
 $ treex --help
-usage: treex [-h] [-a] [-m] [-q] [-s] [directory]
+usage: treex [-h] [-a] [-m] [-q] [-d] [-s] [directory]
 
 List directory contents as a tree with file and directory metadata.
 
@@ -109,9 +109,10 @@ positional arguments:
 
 options:
   -h, --help      show this help message and exit
-  -a, --all       show all files (including those ignored by Git)
+  -a, --all       show all files, including those ignored by Git
   -m, --modified  show file modification times
   -q, --quiet     don't show file metadata
+  -d, --dirs      show directories only
   -s, --summary   show summary only
 ```
 

{treex_cli-0.1.6 → treex_cli-0.1.8}/README.md

@@ -75,7 +75,7 @@ Install `treex` from [PyPI][pypi-home] using either `pip` or `pipx`:
 
 ```
 $ treex --help
-usage: treex [-h] [-a] [-m] [-q] [-s] [directory]
+usage: treex [-h] [-a] [-m] [-q] [-d] [-s] [directory]
 
 List directory contents as a tree with file and directory metadata.
 
@@ -84,9 +84,10 @@ positional arguments:
 
 options:
   -h, --help      show this help message and exit
-  -a, --all       show all files (including those ignored by Git)
+  -a, --all       show all files, including those ignored by Git
   -m, --modified  show file modification times
   -q, --quiet     don't show file metadata
+  -d, --dirs      show directories only
   -s, --summary   show summary only
 ```
 

{treex_cli-0.1.6 → treex_cli-0.1.8}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "treex-cli"
-version = "0.1.6"
+version = "0.1.8"
 description="List directory contents as a tree with file and directory metadata"
 license = "MIT"
 license-files = ["LICENSE"]
@@ -53,6 +53,7 @@ py-modules = ["treex"]
 [tool.pytest]
 addopts = ["-v"]
 filterwarnings = ["default"]
+tmp_path_retention_policy = "none"
 
 [tool.ruff]
 line-length = 88
@@ -62,7 +63,7 @@ target-version = "py310"
 [tool.ruff.lint]
 select = ["ALL"]
 ignore = [
-    "ANN", "COM812", "D", "DTZ001", "DTZ006",
+    "ANN", "C901", "COM812", "D", "DTZ001", "DTZ006",
     "FBT", "PLR", "S603", "S607", "T201"
 ]
 fixable = ["ALL"]

{treex_cli-0.1.6 → treex_cli-0.1.8}/treex.py

@@ -14,6 +14,7 @@ filesystem is scanned normally without Git filtering.
 
 import argparse
 import subprocess
+import sys
 from contextlib import suppress
 from datetime import datetime
 from pathlib import Path
@@ -151,6 +152,7 @@ def _collect_tree(
     rows,
     show_metadata,
     show_modified,
+    show_files,
 ):
     """Walk the tree once and collect the output rows."""
     entries = _get_tree_entries(path, gitignore)
@@ -158,15 +160,26 @@ def _collect_tree(
         rows.append((prefix + "└── [permission denied]", None))
         return 0
     total_size = 0
-    for index, entry in enumerate(entries):
-        is_last = index == len(entries) - 1
-        connector = "└── " if is_last else "├── "
-        tree_name = prefix + connector + entry.name
+    # When files are hidden, only directories determine which visible
+    # entry is last. Files are still processed for stats and sizes.
+    visible_entries = entries
+    if not show_files:
+        visible_entries = [entry for entry in entries if entry.is_dir()]
+    visible_index = 0
+    for entry in entries:
+        is_visible = show_files or entry.is_dir()
+        if is_visible:
+            is_last = visible_index == len(visible_entries) - 1
+            visible_index += 1
+            connector = "└── " if is_last else "├── "
+            tree_name = prefix + connector + entry.name
+        else:
+            tree_name = None
         if entry.is_dir():
             stats["directories"] += 1
             row_index = len(rows)
             rows.append((tree_name, None))
-            # Indent nested entries, preserving the tree's vertical branch
+            # Indent nested entries, preserving the tree's vertical branch.
             extension = "    " if is_last else "│   "
             dir_size = _collect_tree(
                 entry,
@@ -176,6 +189,7 @@ def _collect_tree(
                 rows,
                 show_metadata,
                 show_modified,
+                show_files,
             )
             total_size += dir_size
             if show_metadata:
@@ -186,25 +200,17 @@ def _collect_tree(
                 stats["files"] += 1
                 stats["total_size"] += size
                 total_size += size
-                timestamp = modified if show_modified else None
-                rows.append((tree_name, (size_text, info, timestamp)))
+                if show_files:
+                    timestamp = modified if show_modified else None
+                    rows.append((tree_name, (size_text, info, timestamp)))
             else:
                 stats["files"] += 1
                 stats["total_size"] = None
-                rows.append((tree_name, None))
+                if show_files:
+                    rows.append((tree_name, None))
     return total_size
 
 
-def print_summary(stats):
-    directories = stats["directories"]
-    files = stats["files"]
-    total_size = stats["total_size"]
-    summary = f"{directories:,} directories • {files:,} files"
-    if total_size:
-        summary += f" • {_format_size(total_size)}"
-    print(summary)
-
-
 def print_tree(
     directory,
     prefix="",
@@ -213,8 +219,9 @@ def print_tree(
     show_metadata=True,
     show_modified=False,
     show_tree=True,
+    show_files=True,
 ):
-    """Print the directory tree with aligned file metadata."""
+    """Print the directory tree with aligned metadata."""
     if stats is None:
         stats = {"directories": 0, "files": 0, "total_size": 0}
     rows = []
@@ -227,18 +234,16 @@ def print_tree(
         rows,
         show_metadata,
         show_modified,
+        show_files,
     )
     if show_tree:
-        # Align file metadata with the longest tree/name
-        max_tree_name_width = max(
-            (len(tree_name) for tree_name, _ in rows),
-            default=0,
-        )
-        for tree_name, file_info_data in rows:
-            if file_info_data is None:
+        # Align metadata with the longest tree/name
+        max_tree_name_width = max((len(tree_name) for tree_name, _ in rows), default=0)
+        for tree_name, metadata in rows:
+            if metadata is None:
                 print(tree_name)
                 continue
-            size_text, info, modified = file_info_data
+            size_text, info, modified = metadata
             output = f"{tree_name:<{max_tree_name_width}}{size_text:>10}"
             if info is not None:
                 output += f"    {info:<10}"
@@ -248,11 +253,24 @@ def print_tree(
     return stats
 
 
+def print_summary(stats):
+    """Print a summary of directory/file counts and total size."""
+    directories = stats["directories"]
+    files = stats["files"]
+    total_size = stats["total_size"]
+    directory_label = "directory" if directories == 1 else "directories"
+    file_label = "file" if files == 1 else "files"
+    summary = f"{directories:,} {directory_label} • {files:,} {file_label}"
+    if total_size is not None:
+        summary += f" • {_format_size(total_size)}"
+    print(summary)
+
+
 def parse_args(argv=None):
     """Parse command-line arguments."""
     parser = argparse.ArgumentParser(
-        description="List directory contents as a tree with "
-        "file and directory metadata.",
+        description="List directory contents as a tree "
+        "with file and directory metadata.",
     )
     parser.add_argument(
         "directory",
@@ -265,7 +283,7 @@ def parse_args(argv=None):
         "-a",
         "--all",
         action="store_true",
-        help="show all files (including those ignored by Git)",
+        help="show all files, including those ignored by Git",
     )
     parser.add_argument(
         "-m",
@@ -279,6 +297,12 @@ def parse_args(argv=None):
         action="store_true",
         help="don't show file metadata",
     )
+    parser.add_argument(
+        "-d",
+        "--dirs",
+        action="store_true",
+        help="show directories only",
+    )
     parser.add_argument(
         "-s",
         "--summary",
@@ -289,28 +313,35 @@ def parse_args(argv=None):
 
 
 def main():
-    args = parse_args()
-    path = args.directory
-    if not path.is_dir():
-        print(f"Not a directory: {path}")
-        return 1
-    # Only initialize Git support if it will be used
-    gitignore = None if args.all else GitIgnore(path)
-    show_tree = not args.summary
-    show_metadata = not args.quiet
-    if show_tree:
-        print(path)
-    stats = print_tree(
-        path,
-        gitignore=gitignore,
-        show_metadata=show_metadata,
-        show_modified=args.modified,
-        show_tree=show_tree,
-    )
-    if show_tree:
-        print()
-    print_summary(stats)
-    return 0
+    try:
+        args = parse_args()
+        path = args.directory
+        if not path.is_dir():
+            print(f"Not a directory: {path}", file=sys.stderr)
+            return 1
+        # Only initialize Git support if it will be used
+        gitignore = None if args.all else GitIgnore(path)
+        show_tree = not args.summary
+        show_metadata = not args.quiet
+        show_files = not args.dirs
+        if show_tree:
+            print(path)
+        stats = print_tree(
+            path,
+            gitignore=gitignore,
+            show_metadata=show_metadata,
+            show_modified=args.modified,
+            show_tree=show_tree,
+            show_files=show_files,
+        )
+        if show_tree:
+            print()
+        print_summary(stats)
+    except KeyboardInterrupt:
+        print("\nInterrupted", file=sys.stderr)
+        return 130
+    else:
+        return 0
 
 
 if __name__ == "__main__":

{treex_cli-0.1.6 → treex_cli-0.1.8/treex_cli.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: treex-cli
-Version: 0.1.6
+Version: 0.1.8
 Summary: List directory contents as a tree with file and directory metadata
 Author: Corey Goldberg
 Maintainer: Corey Goldberg
@@ -100,7 +100,7 @@ Install `treex` from [PyPI][pypi-home] using either `pip` or `pipx`:
 
 ```
 $ treex --help
-usage: treex [-h] [-a] [-m] [-q] [-s] [directory]
+usage: treex [-h] [-a] [-m] [-q] [-d] [-s] [directory]
 
 List directory contents as a tree with file and directory metadata.
 
@@ -109,9 +109,10 @@ positional arguments:
 
 options:
   -h, --help      show this help message and exit
-  -a, --all       show all files (including those ignored by Git)
+  -a, --all       show all files, including those ignored by Git
   -m, --modified  show file modification times
   -q, --quiet     don't show file metadata
+  -d, --dirs      show directories only
   -s, --summary   show summary only
 ```