import subprocess

import sys

import glob

slug = sys.argv[1]

def run_command(command, verbose=1):

try:

result = subprocess.run(command, check=True, capture_output=True, text=True)

except subprocess.CalledProcessError as e:

print(f"Error building book: {e.stderr}")

sys.exit(1)

if verbose > 0:

print(f"Output: {result.stdout}")

return result

notebooks = glob.glob("*.ipynb")

if len(notebooks) > 1:

print("More than one .ipynb notebook found --> assuming this is a book")

# build book

command = [

"jb",

"build",

"--config",

"_config.yml",

"--toc",

"_toc.yml",

".",

]

run_command(command)

# copy build outputs to 'html' dir

command = ["cp", "-r", "_build/html", "html"]

run_command(command)

else:

# build single notebook

command = [

"jb",

"build",

"--config",

"_config.yml",

f"{slug}.ipynb",

]

run_command(command)

# copy build outputs to 'html' dir

command = ["cp", "-r", f"_build/_page/{slug}/html", "html"]

run_command(command)