Coverage for src/markdown_exec/formatters/pycon.py: 100.00%

1"""Formatter for executing `pycon` code.""" 

2 

3from __future__ import annotations 

4 

5from typing import TYPE_CHECKING, Any 

6 

7from markdown_exec.formatters.base import base_format 

8from markdown_exec.formatters.python import _run_python 

9from markdown_exec.logger import get_logger 

10 

11if TYPE_CHECKING: 

12 from markupsafe import Markup 

13 

14logger = get_logger(__name__) 

15 

16 

17def _transform_source(code: str) -> tuple[str, str]: 

18 python_lines = [] 

19 pycon_lines = [] 

20 for line in code.split("\n"): 

21 if line.startswith((">>> ", "... ")): 

22 pycon_lines.append(line) 

23 python_lines.append(line[4:]) 

24 python_code = "\n".join(python_lines) 

25 return python_code, "\n".join(pycon_lines) 

26 

27 

28def _format_pycon(**kwargs: Any) -> Markup: 

29 return base_format(language="pycon", run=_run_python, transform_source=_transform_source, **kwargs)