def fill_template()

in java/ext/javadoctest.py [0:0]

• 14 lines of code
• 8 McCabe index (conditional complexity)

    def fill_template(self, template, code):
        # Detect the special case where cookbook code is already wrapped in a
        # class and just use the code as-is without wrapping it up
        if code.find("public class Example") >= 0:
            return template + code

        # Split input code into imports and not-imports
        lines = code.split("\n")
        code_imports = [l for l in lines if l.startswith("import")]
        code_rest = [l for l in lines if not l.startswith("import")]

        pieces = [
            template,
            "\n".join(code_imports),
            "\n\npublic class Example {\n    public static void main(String[] args) {\n",
            "\n".join(code_rest),
            "    }\n}",
        ]

        return "\n".join(pieces)