Automatic support for ipython3 code blocks

Author: mgeier

src/sphinx_codeautolink/extension/block.py

@@ -53,6 +53,12 @@ def clean_pycon(source: str) -> Tuple[str, str]:
     return source, '\n'.join(clean_lines)
 
 
+def clean_ipython(source: str) -> Tuple[str, str]:
+    """Clean up IPython syntax to pure Python."""
+    from IPython.core.inputtransformer2 import TransformerManager
+    return source, TransformerManager().transform_cell(source)
+
+
 class CodeBlockAnalyser(nodes.SparseNodeVisitor):
     """Transform literal blocks of Python with links to reference documentation."""
 
@@ -70,7 +76,10 @@ def __init__(
         relative_path = Path(self.document['source']).relative_to(source_dir)
         self.current_document = str(relative_path.with_suffix(''))
         self.global_preface = global_preface
-        self.custom_blocks = {'pycon': clean_pycon}
+        self.custom_blocks = {
+            'pycon': clean_pycon,
+            'ipython3': clean_ipython,
+        }
         self.custom_blocks.update(custom_blocks)
         self.valid_blocks = ('py', 'python') + tuple(self.custom_blocks.keys())
         self.title_stack = []
@@ -230,7 +239,7 @@ def link_html(
     text = document.read_text('utf-8')
     soup = BeautifulSoup(text, 'html.parser')
 
-    block_types = {'python', 'pycon'} | set(custom_blocks.keys())
+    block_types = {'python', 'pycon', 'ipython3'} | set(custom_blocks.keys())
     classes = [f'highlight-{t}' for t in block_types] + ['doctest']
     classes += search_css_classes