Add shortcut for getting the number of items

torchbox · Oct 1, 2024 · bac1bd7 · bac1bd7
1 parent bb5201d
commit bac1bd7
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -62,5 +62,6 @@ By default, a pretty table is output, for easy consumption by humans. `--format`
 - `table` (Default)
 - `csv`
 - `json`
+- `count`
 
 Progress output is automatically removed when running non-interactively.
diff --git a/heroku_audit/format.py b/heroku_audit/format.py
@@ -26,12 +26,17 @@ class Format(str, Enum):
     TABLE = "table"
     CSV = "csv"
     JSON = "json"
+    COUNT = "count"
 
 
 FormatOption = Annotated[Format, typer.Option("--format")]
 
 
 def display_data(data: list[dict], display_format: Format) -> None:
+    if display_format == Format.COUNT:
+        print(len(data))
+        return
+
     if not data:
         return