Function extract_code_blocks

tools/check_code_block_format.py:153–178

(
    file: io.TextIOWrapper,
)

def extract_code_blocks(
    file: io.TextIOWrapper,
) -> Iterator[CodeBlock]:
    lines_iter = iter(enumerate(file))

    while True:
        next_item = next(lines_iter, None)
        if next_item is None:
            break
        lineno, line = next_item
        match = REGEX_CODE_BLOCK.match(line)
        if match:
            indent = match.group('indent')
            indent_size = len(indent)
            code_block_start_lineno = lineno + 1
            code_block_lines = [line]
            indent_regex = re.compile(rf"\s{{{indent_size + 1}}}")
            while True:
                next_item = next(lines_iter, None)
                if next_item is None:
                    break
                lineno, line = next_item
                if not indent_regex.match(line) and not is_blank_line(line):
                    break
                code_block_lines.append(line)
            yield CodeBlock(code_block_lines, code_block_start_lineno)


def check_code_block_indentation(