Function python_code

tests/__init__.py:91–123

Format Python source code, given as a string, for use with ``exec``. Use it like so:: exec(python_code(''' async def foo(): pass ''')) This allows to use newer syntactic constructs that'd cause SyntaxError on older Python versions.

(source)

# Utility functions

def python_code(source):
    """Format Python source code, given as a string, for use with ``exec``.

    Use it like so::

        exec(python_code('''
            async def foo():
                pass
        '''))

    This allows to use newer syntactic constructs that'd cause SyntaxError
    on older Python versions.
    """
    # (Whitespace shenanigans adapted from sphinx.utils.prepare_docstring)

    # remove excess whitespace from source code lines which will be there
    # if the code was given as indented, multiline string
    lines = source.expandtabs().splitlines()
    margin = sys.maxsize
    for line in lines:
        code_len = len(line.strip())
        if code_len > 0:
            indent = len(line) - code_len
            margin = min(margin, indent)
    if margin < sys.maxsize:
        for i in range(len(lines)):
            lines[i] = lines[i][margin:]

    # ensure there is an empty line at the end
    if lines and lines[-1]:
        lines.append('')

    return os.linesep.join(lines)