Use ipython[2,3] for notebooks when nbconvert version >= 5.0

Chilipp · Feb 6, 2017 · b576a96 · b576a96
1 parent a05badb
commit b576a96
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/sphinx_nbexamples/__init__.py b/sphinx_nbexamples/__init__.py
@@ -47,8 +47,12 @@
 
 logger = logging.getLogger(__name__)
 
-code_blocks = re.compile(r'\.\. code:: python\n(?s)(.+?)(?=\n\S+|$)')
-inner_code_blocks = re.compile(r'(?<=.. code:: python\n)(?s)(.+?)(?=\n\S+|$)')
+if nbconvert.__version__ < '5.0':
+    code_blocks = re.compile(r'\.\. code:: python\n(?s)(.+?)(?=\n\S+|$)')
+    inner_code_blocks = re.compile(r'(?<=.. code:: python\n)(?s)(.+?)(?=\n\S+|$)')
+else:
+    code_blocks = re.compile(r'\.\. code:: ipython\d\n(?s)(.+?)(?=\n\S+|$)')
+    inner_code_blocks = re.compile(r'(?<=.. code:: ipython\d\n)(?s)(.+?)(?=\n\S+|$)')
 magic_patt = re.compile(r'(?m)^(\s+)(%.*\n)')