def split_code_at_show()

in _ext/sphinx_plotly_directive.py [0:0]

• 16 lines of code
• 7 McCabe index (conditional complexity)

def split_code_at_show(text):
    """Split code at plt.show()."""
    parts = []
    is_doctest = contains_doctest(text)

    part = []
    for line in text.split("\n"):
        if (not is_doctest and line.strip() == "plt.show()") or (
            is_doctest and line.strip() == ">>> plt.show()"
        ):
            part.append(line)
            parts.append("\n".join(part))
            part = []
        else:
            part.append(line)
    if "\n".join(part).strip():
        parts.append("\n".join(part))
    return parts