nbcat 0.8.1__tar.gz → 0.9.0__tar.gz

{nbcat-0.8.1 → nbcat-0.9.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nbcat
-Version: 0.8.1
+Version: 0.9.0
 Summary: cat for jupyter notebooks
 Project-URL: Homepage, https://github.com/akopdev/nbcat
 Project-URL: Repository, https://github.com/akopdev/nbcat
@@ -29,6 +29,7 @@ License: MIT License
         SOFTWARE.
 License-File: LICENSE
 Requires-Python: >=3.10
+Requires-Dist: argcomplete
 Requires-Dist: pydantic
 Requires-Dist: requests
 Requires-Dist: rich

{nbcat-0.8.1 → nbcat-0.9.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "nbcat"
-version = "0.8.1"
+version = "0.9.0"
 description = "cat for jupyter notebooks"
 authors = [
     { name = "Akop Kesheshyan", email = "devnull@akop.dev" }
@@ -12,6 +12,7 @@ license = {file = "LICENSE"}
 readme = "README.md"
 requires-python = ">=3.10"
 dependencies = [
+    "argcomplete",
     "requests",
     "pydantic",
     "rich",

nbcat-0.9.0/src/nbcat/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.9.0"

{nbcat-0.8.1 → nbcat-0.9.0}/src/nbcat/main.py

@@ -2,7 +2,9 @@ import argparse
 import sys
 from pathlib import Path
 
+import argcomplete
 import requests
+from argcomplete.completers import FilesCompleter
 from pydantic import ValidationError
 from rich import box
 from rich.console import Console, RenderableType
@@ -148,7 +150,9 @@ def main():
         description="cat for Jupyter Notebooks",
         argument_default=argparse.SUPPRESS,
     )
-    parser.add_argument("file", help="Path or URL to a .ipynb notebook", type=str)
+    parser.add_argument(
+        "file", help="Path or URL to a .ipynb notebook", type=str
+    ).completer = FilesCompleter()
     parser.add_argument(
         "--version",
         help="print version information and quite",
@@ -157,6 +161,7 @@ def main():
     )
 
     try:
+        argcomplete.autocomplete(parser)
         args = parser.parse_args()
         notebook = read_notebook(args.file)
         print_notebook(notebook)

{nbcat-0.8.1 → nbcat-0.9.0}/src/nbcat/schemas.py

@@ -12,7 +12,6 @@ class BaseOutput(BaseModel):
 
 
 class StreamOutput(BaseOutput):
-    name: str
     text: list[str] | str
 
     @computed_field

nbcat-0.9.0/tests/test_render_cell.py

@@ -0,0 +1,67 @@
+import pytest
+from rich.markdown import Markdown
+from rich.panel import Panel
+from rich.text import Text
+
+from nbcat.main import render_cell
+from nbcat.schemas import Cell, StreamOutput
+
+
+@pytest.mark.parametrize(
+    "cell_type,source,expected",
+    [
+        ("markdown", "# Heading", Markdown),
+        ("code", "print('Hello')", Panel),
+        ("raw", "Raw content", Text),
+        ("heading", "Heading text", Text),
+    ],
+)
+def test_render_cell_input_rendering(cell_type: str, source: str, expected):
+    cell = Cell(cell_type=cell_type, source=source, execution_count=42, outputs=[])
+    rendered = render_cell(cell)
+
+    assert len(rendered) == 1
+    label, content = rendered[0]
+    assert label == "[green][42][/]:"
+    assert isinstance(content, expected)
+
+
+def test_render_cell_with_outputs():
+    output_1 = StreamOutput(text=["First output"], execution_count=7, output_type="stream")
+    output_2 = StreamOutput(text=["Second output"], output_type="stream")
+
+    cell = Cell(
+        cell_type="code",
+        source="print('Hello')",
+        execution_count=None,
+        outputs=[output_1, output_2],
+    )
+
+    rendered = render_cell(cell)
+
+    print(rendered)
+    assert len(rendered) == 3
+    assert rendered[0][0] is None
+    assert isinstance(rendered[0][1], Panel)
+
+    assert rendered[1][0] == "[blue][7][/]:"
+    assert isinstance(rendered[1][1], str)
+
+    assert rendered[2][0] is None
+    assert isinstance(rendered[2][1], str)
+
+
+def test_render_cell_skips_empty_outputs():
+    output = StreamOutput(text="", execution_count=99, output_type="stream")
+    cell = Cell(
+        cell_type="raw",
+        source="Raw input",
+        execution_count=1,
+        outputs=[output],
+    )
+
+    rendered = render_cell(cell)
+
+    assert len(rendered) == 1  # Only source input is rendered
+    assert rendered[0][0] == "[green][1][/]:"
+    assert isinstance(rendered[0][1], Text)