fixes #24

Author: kwinkunks

CHANGES.md

@@ -1,5 +1,10 @@
 # Changelog
 
+## 0.1.6 — 13 Jul 2022
+
+- Notebook kernel is changed to match the current course.
+
+
 ## 0.1.5 — 18 May 2022
 
 - Improved the appearance of feedback messages, and added emojis.

kosu/customize.py

@@ -110,6 +110,19 @@ def hide_toolbar(notebook):
     return notebook
 
 
+def change_kernel(notebook, kernel):
+    """
+    Vanillafy the kernelspec.
+    """
+    new_kernelspec = {
+        "display_name": f"{kernel}",
+        "language": "python",
+        "name": f"{kernel}",
+    }
+    notebook['metadata']['kernelspec'].update(new_kernelspec)
+    return notebook
+
+
 def process_notebook(infile,
                      outfile,
                      clear_input=False,  # Don't touch the input file.
@@ -120,6 +133,7 @@ def process_notebook(infile,
                      hidecode=True,
                      demo=False,  # If demo, remove exercises and enable demos.
                      data_url_stem=None,
+                     kernel=None,
                     ):
     """
     Loads an 'ipynb' file as a dict and performs cleaning tasks
@@ -147,6 +161,9 @@ def process_notebook(infile,
     if hidecode:
         notebook = hide_code(notebook)
 
+    if kernel is not None:
+        notebook = change_kernel(notebook, kernel)
+
     notebook = hide_toolbar(notebook)
 
     text = json.dumps(notebook)

kosu/kosu.py

@@ -307,7 +307,7 @@ def build_course(course, clean, zip, upload, clobber):
     if upload:
         success = upload_zip(zipped)
         if success:
-            click.secho(f"⬆️ Uploaded {zipped}", fg="green")
+            click.secho(f"⬆️  Uploaded {zipped}", fg="green")
             link  = f"https://{KOSU['s3-bucket']}.s3.amazonaws.com/"
             link += f"{course}.zip"
             click.secho(f"🔗 File link: {link}", fg="green")
@@ -342,13 +342,14 @@ def build_notebooks(path, config):
     all_items = [f for items in config['curriculum'].values() for f in items]
     notebooks = list(filter(lambda item: '.ipynb' in item, all_items))
     notebooks += config.get('extras', list())
+    kernel = config.get('environment', config['course']).lower()
     images_to_copy = []
     data_urls_to_check = []
     click.secho('📔 Processing notebooks ', fg="cyan", nl=False)
     for notebook in notebooks:
         infile = pathlib.Path(KOSU['notebooks-source']) / notebook
         outfile = nb_path / notebook
-        images, data_urls = process_notebook(infile, outfile)
+        images, data_urls = process_notebook(infile, outfile, kernel=kernel)
         images_to_copy.extend(images)
         data_urls_to_check.extend(data_urls)
         shutil.copyfile(infile, m_path / notebook)
@@ -359,7 +360,7 @@ def build_notebooks(path, config):
     for notebook in notebooks:
         infile = pathlib.Path(KOSU['notebooks-source']) / notebook
         outfile = demo_path / notebook
-        images, data_urls = process_notebook(infile, outfile, demo=True)
+        images, data_urls = process_notebook(infile, outfile, demo=True, kernel=kernel)
         images_to_copy.extend(images)
         data_urls_to_check.extend(data_urls)
         shutil.copyfile(infile, m_path / notebook)