1

我有这样的python代码具有这种结构

def main:

    ''' comment '''
    if True:
        print "do"
    print "done

此代码与交互模式不兼容（例如，如果我在交互会话中复制/粘贴它）。为此，它需要：

def main:
    ''' comment '''
    if True:
        print "do"

    print "done"

否则交互模式会在缩进问题上中断。

你知道用 generate_token / unkenize 链转换代码的简单方法吗？我有点迷失在 NL / NEWLINE / INDENT / DEDENT 语义中。

我发现这个脚本来删除删除注释/文档字符串的 Python 注释/文档字符串。它看起来非常适合我的问题，但它无法将其整理出来以在复杂代码上获得干净的输出。

4

1 回答 1

0

我能想到的最好的（解决了我的问题）

def _python_interactive_indent(self, code):
    prev_toktype = tokenize.INDENT
    first_line = None
    last_lineno = -1
    last_col = 0

    output = ''

    tokgen = tokenize.generate_tokens(StringIO.StringIO(code).readline)
    indent = 0
    hasNL = False
    prefixed = False
    for toktype, ttext, (slineno, scol), (elineno, ecol), ltext in tokgen:
        done = False
        if toktype == tokenize.INDENT:
            indent = indent + 1
        if toktype == tokenize.DEDENT:
            indent = indent - 1
        if slineno > last_lineno:
            last_col = 0
        if not done and toktype == tokenize.NL:
            hasNL = True
            done = True
        if not done and toktype == tokenize.COMMENT:
            done = True
        if not done and toktype == tokenize.STRING and prev_toktype == tokenize.INDENT:
            done = True
        if not done and hasNL and toktype != tokenize.DEDENT and toktype != tokenize.INDENT:
            hasNL = False
            output = output + ("    " * indent) + '\n'
            output += "    " * indent
            prefixed = True
        if not done:
            if not prefixed and scol > last_col:
                output += (" " * (scol - last_col))
            output += (ttext)
        prefixed = False
        prev_toktype = toktype
        last_col = ecol
        last_lineno = elineno
    return output
于 2014-01-20T20:56:11.597 回答