nbcat 0.9.6__py3-none-any.whl → 0.11.0__py3-none-any.whl

nbcat/__init__.py

@@ -1 +1 @@
-__version__ = "0.9.6"
+__version__ = "0.11.0"

nbcat/image.py

@@ -0,0 +1,31 @@
+import base64
+import shutil
+from io import BytesIO
+from platform import system
+
+from PIL import Image as PilImage
+from rich.console import Console, ConsoleOptions, RenderResult
+from rich.text import Text
+from timg import METHODS, Renderer
+
+
+class Image:
+    def __init__(self, image: str, method: str = "a24h"):
+        img = BytesIO(base64.b64decode(image.replace("\n", "")))
+        self.image = PilImage.open(img)
+        self.method = method if system() != "Windows" else "ascii"
+
+    def __rich_console__(self, console: Console, options: ConsoleOptions) -> RenderResult:
+        renderer = Renderer()
+        renderer.load_image(self.image)
+        width = shutil.get_terminal_size()[0] - 1
+        if self.method == "sixel":
+            width = width * 6
+
+        renderer.resize(width)
+
+        if self.method == "sixel":
+            renderer.reduce_colors(16)
+
+        output = renderer.to_string(METHODS[self.method]["class"])
+        yield Text.from_ansi(output, no_wrap=True, end="")

nbcat/main.py

@@ -1,18 +1,17 @@
 import argparse
 import sys
 from pathlib import Path
-from typing import Union
 
 import argcomplete
 import requests
 from argcomplete.completers import FilesCompleter
 from pydantic import ValidationError
 from rich import box
-from rich.console import Console, RenderableType
+from rich.console import Console, Group, RenderableType
+from rich.padding import Padding
 from rich.panel import Panel
 from rich.pretty import Pretty
 from rich.syntax import Syntax
-from rich.table import Table
 from rich.text import Text
 
 from . import __version__
@@ -22,6 +21,7 @@ from .exceptions import (
     NotebookNotFoundError,
     UnsupportedNotebookTypeError,
 )
+from .image import Image
 from .markdown import Markdown
 from .schemas import Cell, Notebook
 
@@ -68,7 +68,7 @@ def read_notebook(fp: str, debug: bool = False) -> Notebook:
         raise InvalidNotebookFormatError(f"Invalid notebook: {e}")
 
 
-def render_cell(cell: Cell) -> list[tuple[Union[str, None], RenderableType]]:
+def render_cell(cell: Cell) -> RenderableType:
     """
     Render the content of a notebook cell for display.
 
@@ -85,16 +85,18 @@ def render_cell(cell: Cell) -> list[tuple[Union[str, None], RenderableType]]:
     """
 
     def _render_markdown(input: str) -> Markdown:
-        return Markdown(input)
+        return Markdown(input, code_theme="ansi_dark")
 
-    def _render_code(input: str) -> Panel:
-        return Panel(Syntax(input, "python", theme="ansi_dark"), box=box.SQUARE)
+    def _render_code(input: str, language: str = "python") -> Panel:
+        return Panel(
+            Syntax(input, language, line_numbers=True, theme="ansi_dark", dedent=True), padding=0
+        )
 
     def _render_raw(input: str) -> Text:
         return Text(input)
 
-    def _render_image(input: str) -> None:
-        return None
+    def _render_image(input: str) -> Image:
+        return Image(input)
 
     def _render_json(input: str) -> Pretty:
         return Pretty(input)
@@ -110,31 +112,21 @@ def render_cell(cell: Cell) -> list[tuple[Union[str, None], RenderableType]]:
         OutputCellType.JSON: _render_json,
     }
 
-    rows: list[tuple[Union[str, None], RenderableType]] = []
+    rows: list[RenderableType] = []
     renderer = RENDERERS.get(cell.cell_type)
     source = renderer(cell.input) if renderer else None
     if source:
-        label = f"[green][{cell.execution_count}][/]" if cell.execution_count else None
-        rows.append(
-            (
-                label,
-                source,
-            )
-        )
+        rows.append(Padding(source, (1, 0)))
+        if not cell.outputs:
+            return source
 
     for o in cell.outputs:
         if o.output:
             renderer = RENDERERS.get(o.output.output_type)
             output = renderer(o.output.text) if renderer else None
             if output:
-                label = f"[blue][{o.execution_count}][/]" if o.execution_count else None
-                rows.append(
-                    (
-                        label,
-                        output,
-                    )
-                )
-    return rows
+                rows.append(Panel(output, style="italic", box=box.MINIMAL))
+    return Group(*rows)
 
 
 def print_notebook(nb: Notebook):
@@ -148,15 +140,17 @@ def print_notebook(nb: Notebook):
         console.print("[bold red]Notebook contains no cells.")
         return
 
-    layout = Table.grid(padding=1)
-    layout.add_column(no_wrap=True, width=6)
-    layout.add_column()
-
     for cell in nb.cells:
-        for label, content in render_cell(cell):
-            layout.add_row(label, content)
-
-    console.print(layout)
+        rendered = render_cell(cell)
+        if isinstance(rendered, Group):
+            out = Panel(
+                rendered,
+                title=f"[green][{cell.execution_count}][/]" if cell.execution_count else None,
+                title_align="left",
+            )
+        else:
+            out = Padding(rendered, (1, 0))
+        console.print(out)
 
 
 def main():

nbcat/schemas.py

@@ -34,7 +34,7 @@ class DisplayDataOutput(BaseOutput):
     def output(self) -> Union[CellOutput, None]:
         data_type_map = {
             "text/html": OutputCellType.HTML,
-            "text/png": OutputCellType.IMAGE,
+            "image/png": OutputCellType.IMAGE,
             "text/plain": OutputCellType.PLAIN,
             "application/vnd.raw.v1+json": OutputCellType.JSON,
         }

{nbcat-0.9.6.dist-info → nbcat-0.11.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nbcat
-Version: 0.9.6
+Version: 0.11.0
 Summary: cat for jupyter notebooks
 Project-URL: Homepage, https://github.com/akopdev/nbcat
 Project-URL: Repository, https://github.com/akopdev/nbcat
@@ -33,6 +33,7 @@ Requires-Dist: argcomplete
 Requires-Dist: pydantic
 Requires-Dist: requests
 Requires-Dist: rich
+Requires-Dist: timg
 Provides-Extra: dev
 Requires-Dist: pytest; extra == 'dev'
 Requires-Dist: pytest-cov; extra == 'dev'

nbcat-0.11.0.dist-info/RECORD

@@ -0,0 +1,13 @@
+nbcat/__init__.py,sha256=raMu9XA9JEjvdoTmFqcOw7qhJX24rYDP7XmS59TAO-Q,23
+nbcat/enums.py,sha256=Fn8PIcLl_uY4nQIs1EUvmKTwfhNUIZgmhRFiCSJk9wk,411
+nbcat/exceptions.py,sha256=Ho7LQz9K70VtIMDNtAwuAtGmb-lFKxGxSj7MN3-EpDA,321
+nbcat/image.py,sha256=hGZZK5mtij7ckWAvQwzim3thqGC92pLW5ZU5CYbRv_Q,995
+nbcat/main.py,sha256=vchFzwcvPGmv4pg5AHRaOOsZh83ynmxkTqIKl0z_Sys,5749
+nbcat/markdown.py,sha256=K6yL9rY3uTxMPsRIJnxqNHmCX2Qc-f3my8eiHxTLfic,1835
+nbcat/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nbcat/schemas.py,sha256=a7GoKgPTHgun199-J-sZq-ahkdQwvyRaCdQVg1gC798,3135
+nbcat-0.11.0.dist-info/METADATA,sha256=jmAcfrystY8f3iRq_lm6g7PdhRbPntAfcYsr0sx1TsQ,4103
+nbcat-0.11.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+nbcat-0.11.0.dist-info/entry_points.txt,sha256=io_GRDsecAkYuCZALsjyea3VBq91VCoSznqlZEAJshY,42
+nbcat-0.11.0.dist-info/licenses/LICENSE,sha256=7GjUnahXdd5opdvlpJdb1BisLbiXt2iOFhzIUduhdkE,1072
+nbcat-0.11.0.dist-info/RECORD,,

nbcat-0.9.6.dist-info/RECORD

@@ -1,12 +0,0 @@
-nbcat/__init__.py,sha256=IgVHjr-TeioZYLJSkvpT80LLGi6U3ONzR1cfYfd5XNQ,22
-nbcat/enums.py,sha256=Fn8PIcLl_uY4nQIs1EUvmKTwfhNUIZgmhRFiCSJk9wk,411
-nbcat/exceptions.py,sha256=Ho7LQz9K70VtIMDNtAwuAtGmb-lFKxGxSj7MN3-EpDA,321
-nbcat/main.py,sha256=7RYjHvWmKWoD9LFI1dopKvp_wvqWa2Rk9AYZ_RsrIaw,5805
-nbcat/markdown.py,sha256=K6yL9rY3uTxMPsRIJnxqNHmCX2Qc-f3my8eiHxTLfic,1835
-nbcat/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nbcat/schemas.py,sha256=fjSbdXa4pHKRV1Z-vUESkYZywkojanCvSu8s7rc9Xkw,3134
-nbcat-0.9.6.dist-info/METADATA,sha256=v-8f9HAlHG2t-Mw36CZAIGcMRTeCWiyt0BzZMlS164w,4082
-nbcat-0.9.6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-nbcat-0.9.6.dist-info/entry_points.txt,sha256=io_GRDsecAkYuCZALsjyea3VBq91VCoSznqlZEAJshY,42
-nbcat-0.9.6.dist-info/licenses/LICENSE,sha256=7GjUnahXdd5opdvlpJdb1BisLbiXt2iOFhzIUduhdkE,1072
-nbcat-0.9.6.dist-info/RECORD,,