astimprove.py 文件源码

def _create_astmissing_lines(self) -> List[Optional[Token]]:
        """
        Return a copy of line_tokens containing lines ignored by the AST
        (comments and blanks-only lines)
        """
        lines: List[Optional[Token]] = []
        nl_token = Token(token_module.NEWLINE, '\n', (0, 0), (0, 0), '\n')

        for i, linetokens in enumerate(self._all_lines):
            if len(linetokens) == 1 and linetokens[0].name == 'NL':
                lines.append(nl_token)
            else:
                for token in linetokens:
                    if token.name == 'COMMENT' and \
                            token.rawvalue.lstrip().startswith('#'):
                        lines.append(token)
                        break
                else:
                    lines.append(None)
        assert len(lines) == len(self._all_lines)

        for i, linetokens2 in enumerate(lines):
            if linetokens2:
                self._current_line = i
                break
        else:
            self._current_line = len(lines)
        return lines