Hide nbconvert outputs

Author: zph

runbook/cli/commands/create.py

@@ -1,7 +1,7 @@
 from os import path
 
 import click
-from nbconvert.nbconvertapp import NbConvertApp
+from runbook.cli.lib import nbconvert_launch_instance
 from runbook.cli.validators import validate_create_language, validate_template
 
 
@@ -48,7 +48,7 @@ def create(ctx, filename, template, language):
         path.join("runbooks", "binder"),
     ]
 
-    NbConvertApp().launch_instance(argv=argv)
+    nbconvert_launch_instance(argv)
 
     click.echo(
         click.style(

runbook/cli/commands/plan.py

@@ -5,6 +5,7 @@
 from pathlib import Path
 
 import click
+from runbook.cli.lib import nbconvert_launch_instance
 from runbook.cli.validators import validate_plan_params, validate_runbook_file_path
 from runbook.constants import RUNBOOK_METADATA
 
@@ -96,9 +97,8 @@ def plan(ctx, input, embed, identifier="", params={}):
     # TODO: join the unified logic of create and plan
 
     # TODO: hide the nbconvert verbose output?
-    from nbconvert.nbconvertapp import NbConvertApp
 
-    NbConvertApp().launch_instance(argv=argv)
+    nbconvert_launch_instance(argv)
 
     for f in embed:
         shutil.copyfile(src=f, dst=f"{output_folder}/{path.basename(f)}")

runbook/cli/lib.py

@@ -1,5 +1,8 @@
 import hashlib
+import sys
+from io import StringIO
 
+from nbconvert.nbconvertapp import NbConvertApp
 from ulid import ULID
 
 
@@ -14,3 +17,22 @@ def sha256sum(filename):
 # use ULID() directly
 def ts_id(length=10):
     return str(ULID())[0:length]
+
+
+# Suppresses nbconvert output
+def nbconvert_launch_instance(argv):
+    # Temporarily redirect stdout/stderr to suppress nbconvert output
+    stdout = sys.stdout
+    stderr = sys.stderr
+    sys.stdout = StringIO()
+    sys.stderr = StringIO()
+    try:
+        NbConvertApp().launch_instance(argv=argv)
+    except Exception as e:
+        print(sys.stdout.getvalue())
+        print(sys.stderr.getvalue())
+        raise e
+    finally:
+        # Restore stdout/stderr
+        sys.stdout = stdout
+        sys.stderr = stderr