Support Sage notebooks

docs/CHANGELOG.md

@@ -4,6 +4,9 @@ Jupytext ChangeLog
 1.10.1 (2021-02-??)
 -------------------
 
+**Added**
+- Sage notebooks are supported. They can be converted to `.sage` and `.md` files and back. Thanks to Lars Franke for suggesting this! ([#727](https://github.com/mwouts/jupytext/issues/727))
+
 
 1.10.0 (2021-02-04)
 -------------------

docs/languages.md

@@ -21,6 +21,7 @@ Jupytext works with notebooks in any of the following languages:
 - R
 - Robot Framework
 - Rust/Evxcr
+- Sage
 - Scala
 - Scheme
 - Script of Script

jupytext/formats.py

@@ -766,6 +766,11 @@ def auto_ext_from_metadata(metadata):
     """Script extension from notebook metadata"""
     auto_ext = metadata.get("language_info", {}).get("file_extension")
 
+    # Sage notebooks have ".py" as the associated extension in "language_info",
+    # so we change it to ".sage" in that case, see #727
+    if auto_ext == ".py" and metadata.get("kernelspec", {}).get("language") == "sage":
+        auto_ext = ".sage"
+
     if auto_ext is None:
         language = metadata.get("kernelspec", {}).get("language") or metadata.get(
             "jupytext", {}

jupytext/languages.py

@@ -57,6 +57,7 @@
     ".sos": {"language": "sos", "comment": "#"},
     ".java": {"language": "java", "comment": "//"},
     ".groovy": {"language": "groovy", "comment": "//"},
+    ".sage": {"language": "sage", "comment": "#"},
 }
 
 _COMMENT_CHARS = [

tests/notebooks/ipynb_sage/sage_print_hello.ipynb

@@ -0,0 +1,34 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "print(\"Hello world\")"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "SageMath 9.2",
+   "language": "sage",
+   "name": "sagemath"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.9.1"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 4
+}

tests/notebooks/mirror/ipynb_to_Rmd/sage_print_hello.Rmd

@@ -0,0 +1,11 @@
+---
+jupyter:
+  kernelspec:
+    display_name: SageMath 9.2
+    language: sage
+    name: sagemath
+---
+
+```{sage}
+print("Hello world")
+```

tests/notebooks/mirror/ipynb_to_hydrogen/sage_print_hello.sage

@@ -0,0 +1,10 @@
+# ---
+# jupyter:
+#   kernelspec:
+#     display_name: SageMath 9.2
+#     language: sage
+#     name: sagemath
+# ---
+
+# %%
+print("Hello world")

tests/notebooks/mirror/ipynb_to_md/sage_print_hello.md

@@ -0,0 +1,11 @@
+---
+jupyter:
+  kernelspec:
+    display_name: SageMath 9.2
+    language: sage
+    name: sagemath
+---
+
+```sage
+print("Hello world")
+```

tests/notebooks/mirror/ipynb_to_myst/sage_print_hello.md

@@ -0,0 +1,10 @@
+---
+kernelspec:
+  display_name: SageMath 9.2
+  language: sage
+  name: sagemath
+---
+
+```{code-cell} ipython3
+print("Hello world")
+```

tests/notebooks/mirror/ipynb_to_percent/sage_print_hello.sage

@@ -0,0 +1,10 @@
+# ---
+# jupyter:
+#   kernelspec:
+#     display_name: SageMath 9.2
+#     language: sage
+#     name: sagemath
+# ---
+
+# %%
+print("Hello world")

tests/notebooks/mirror/ipynb_to_script/sage_print_hello.sage

@@ -0,0 +1,9 @@
+# ---
+# jupyter:
+#   kernelspec:
+#     display_name: SageMath 9.2
+#     language: sage
+#     name: sagemath
+# ---
+
+print("Hello world")

tests/test_auto_ext.py

@@ -39,6 +39,10 @@ def test_auto_from_kernelspecs_works(nb_file):
     elif expected_ext == ".fs":
         expected_ext = ".fsx"
     auto_ext = auto_ext_from_metadata(nb.metadata)
+    if auto_ext == ".sage":
+        pytest.xfail(
+            "Sage notebooks have Python in their language_info metadata, see #727"
+        )
     assert auto_ext == expected_ext