Automatic backup 2025-08-29

Author: snejus

rich_tables/diff.py

@@ -147,6 +147,16 @@ def _(before: HashableDict, after: HashableDict) -> dict[str, str]:
     return data
 
 
+@diff.register
+def _(before: HashableDict, after: None) -> dict[str, str]:
+    return diff(before, HashableDict())
+
+
+@diff.register
+def _(before: None, after: HashableDict) -> dict[str, str]:
+    return diff(HashableDict(), after)
+
+
 def pretty_diff(before: Any, after: Any) -> str:
     """Generate a Rich Text object showing differences between any two Python objects.
 

rich_tables/fields.py

@@ -40,7 +40,7 @@
     from rich.console import RenderableType
 
 
-MATCH_COUNT_HEADER = re.compile(r"duration|(?:_sum$|_count$)")
+MATCH_COUNT_HEADER = re.compile(r"duration|(?:_sum$|_?count$)")
 MAX_BPM_COLOR = (("green", 135), ("yellow", 165), ("red", 400))
 
 

rich_tables/table.py

@@ -84,7 +84,7 @@ def get_args() -> argparse.Namespace:
     parser.add_argument("-v", "--verbose", action="store_true")
     parser.add_argument("-j", "--json", action="store_true", help="output as JSON")
     parser.add_argument(
-        "-s", "--save", action="store_true", help="save the output as HTML"
+        "-o", "--output-file", type=Path, help="save as SVG to provided file"
     )
 
     subparsers = parser.add_subparsers(
@@ -137,23 +137,20 @@ def _draw_data_list(data: list[JSONDict], **__: Any) -> None:
 
 
 @contextmanager
-def handle_save(save: bool) -> Iterator[None]:
-    if save:
-        yield
-
-        if save:
-            with tempfile.NamedTemporaryFile(suffix=".html", delete=False) as file:
-                filename = file.name
-            console.save_html(filename)
-            print(f"Saved output as {filename}", file=sys.stderr)
-    else:
-        yield
+def handle_save(output_file: Path | None) -> Iterator[None]:
+    if output_file:
+        console.record = True
+
+    yield
+
+    if output_file:
+        console.save_svg(str(output_file))
 
 
 def main() -> None:
     args = get_args()
 
-    with handle_save(args.save):
+    with handle_save(args.output_file):
         if args.command == "diff":
             console.print(pretty_diff(args.before, args.after), highlight=False)
         else: