termgraph 0.7.6__tar.gz → 0.7.7__tar.gz

{termgraph-0.7.6 → termgraph-0.7.7}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.5
 Name: termgraph
-Version: 0.7.6
+Version: 0.7.7
 Summary: A command-line tool that draws basic graphs in the terminal.
 Project-URL: Homepage, https://github.com/mkaz/termgraph
 Project-URL: Repository, https://github.com/mkaz/termgraph

{termgraph-0.7.6 → termgraph-0.7.7}/pyproject.toml

@@ -1,7 +1,7 @@
 
 [project]
 name = "termgraph"
-version = "0.7.6"
+version = "0.7.7"
 description = "A command-line tool that draws basic graphs in the terminal."
 readme = "README.md"
 license = "MIT"

{termgraph-0.7.6 → termgraph-0.7.7}/termgraph/termgraph.py

@@ -97,6 +97,23 @@ def init_args() -> dict:
 
 def main():
     """Main function."""
+    try:
+        try:
+            _run()
+        finally:
+            # Flush even when argument parsing exits for --help or --version.
+            sys.stdout.flush()
+    except BrokenPipeError:
+        # Avoid another failed flush during interpreter shutdown.
+        devnull = os.open(os.devnull, os.O_WRONLY)
+        try:
+            os.dup2(devnull, sys.stdout.fileno())
+        finally:
+            os.close(devnull)
+        raise SystemExit(1) from None
+
+
+def _run():
     args = init_args()
 
     if args["version"]:
@@ -106,14 +123,11 @@ def main():
     data_obj = Data.from_file(args["filename"], args)
     colors = _extract_colors(data_obj, args)
 
-    try:
-        if args["calendar"]:
-            # calendar_heatmap still uses old interface
-            calendar_heatmap(data_obj.data, data_obj.labels, args)
-        else:
-            chart(data_obj, args, colors)
-    except BrokenPipeError:
-        pass
+    if args["calendar"]:
+        # calendar_heatmap still uses old interface
+        calendar_heatmap(data_obj.data, data_obj.labels, args)
+    else:
+        chart(data_obj, args, colors)
 
 
 def chart(data_obj: Data, args: dict, colors: list) -> None:

termgraph-0.7.7/tests/test_init.py

@@ -0,0 +1,42 @@
+import pytest
+
+from termgraph import termgraph as tg
+
+
+def test_init_args():
+    tg.init_args()
+
+
+@pytest.mark.parametrize(
+    "rows,options",
+    [
+        (1, ["--width", "10"]),
+        (300, ["--width", "1000"]),
+        (1, ["--version"]),
+        (1, ["--help"]),
+    ],
+)
+def test_cli_closed_stdout_pipe(tmp_path, rows, options):
+    """Exercise interpreter shutdown too, not only errors raised inside draw."""
+    import os
+    import subprocess
+    import sys
+
+    # One small row fails at the final flush; many wide rows fail while drawing.
+    data_file = tmp_path / f"pipe-{rows}.dat"
+    data_file.write_text("item 1\n" * rows)
+    read_fd, write_fd = os.pipe()
+    os.close(read_fd)
+    try:
+        result = subprocess.run(
+            [sys.executable, "-m", "termgraph.termgraph", str(data_file), *options],
+            check=False,
+            stdout=write_fd,
+            stderr=subprocess.PIPE,
+            text=True,
+            timeout=15,
+        )
+    finally:
+        os.close(write_fd)
+    assert result.returncode == 1
+    assert result.stderr == ""

termgraph-0.7.6/tests/test_init.py

@@ -1,5 +0,0 @@
-from termgraph import termgraph as tg
-
-
-def test_init_args():
-    tg.init_args()