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

nbcat/__init__.py

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

nbcat/image.py

@@ -10,19 +10,22 @@ from timg import METHODS, Renderer
 
 
 class Image:
-    def __init__(self, image: str):
+    def __init__(self, image: str, method: str = "a24h"):
         img = BytesIO(base64.b64decode(image.replace("\n", "")))
         self.image = PilImage.open(img)
-
-    @property
-    def method_class(self):
-        # TODO: auto detect terminal to benefit from sixel protocol support
-        method = "a24h" if system() != "Windows" else "ascii"
-        return METHODS[method]["class"]
+        self.method = method if system() != "Windows" else "ascii"
 
     def __rich_console__(self, console: Console, options: ConsoleOptions) -> RenderResult:
-        img = Renderer()
-        img.load_image(self.image)
-        img.resize(shutil.get_terminal_size()[0] - 1)
-        output = img.to_string(self.method_class)
-        yield Text.from_ansi(output)
+        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__
@@ -69,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.
 
@@ -86,15 +85,17 @@ 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:
+    def _render_image(input: str) -> Image:
         return Image(input)
 
     def _render_json(input: str) -> Pretty:
@@ -111,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):
@@ -149,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-0.10.0.dist-info → nbcat-0.11.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nbcat
-Version: 0.10.0
+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

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.10.0.dist-info/RECORD

@@ -1,13 +0,0 @@
-nbcat/__init__.py,sha256=v4zmKjsKOPZbp6BrWoz7iK4ST0sdZdUh9bQSJmluZ5o,23
-nbcat/enums.py,sha256=Fn8PIcLl_uY4nQIs1EUvmKTwfhNUIZgmhRFiCSJk9wk,411
-nbcat/exceptions.py,sha256=Ho7LQz9K70VtIMDNtAwuAtGmb-lFKxGxSj7MN3-EpDA,321
-nbcat/image.py,sha256=xv_IH2JMyzelUvLjVUbubfc8blmRuxqa-FcKXn25LSk,909
-nbcat/main.py,sha256=89qxlA6viuRP6_ugbf7zRE4clH4oX67-1t1EbpABUfA,5838
-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.10.0.dist-info/METADATA,sha256=jAKupKwXwBtwqMmTT96IE1LihLl3GUhrfLBd0dfTAxw,4103
-nbcat-0.10.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-nbcat-0.10.0.dist-info/entry_points.txt,sha256=io_GRDsecAkYuCZALsjyea3VBq91VCoSznqlZEAJshY,42
-nbcat-0.10.0.dist-info/licenses/LICENSE,sha256=7GjUnahXdd5opdvlpJdb1BisLbiXt2iOFhzIUduhdkE,1072
-nbcat-0.10.0.dist-info/RECORD,,