token-utils

Code on Github

Installation

pip install token-utils

Purpose

The purpose of token-utils is to simplify manipulations of tokens normally obtaind from Python’s tokenize module. One of token-utils’ features is that, unlike Python’s version, the following is always guaranteed:

from token_utils import tokenize, untokenize

source = "Arbitrary Python code here"

assert source == untokenize(tokenize(source))

To get an idea of the simplicity of using token-utils, consider this example from Python’s standard library which substitute Decimals for floats in a string of statements, where we changed the name of the function for greater clarity:

# decimal_py.py
#
# Example taken from Python's tokenize module documentation

from tokenize import tokenize, untokenize, NUMBER, STRING, NAME, OP
from io import BytesIO


def float_to_decimal(s):  # changed name from Python's example
    result = []
    g = tokenize(BytesIO(s.encode("utf-8")).readline)  # tokenize the string
    for toknum, tokval, _, _, _ in g:
        if toknum == NUMBER and "." in tokval:  # replace NUMBER tokens
            result.extend(
                [(NAME, "Decimal"), (OP, "("), (STRING, repr(tokval)), (OP, ")")]
            )
        else:
            result.append((toknum, tokval))
    return untokenize(result).decode("utf-8")

Here’s how you could achieve the same result with token-utils:

# decimal_tok.py

from token_utils import tokenize, untokenize


def float_to_decimal(source):
    tokens = tokenize(source)
    for token in tokens:
        if token.is_float():
            token.string = f"Decimal('{token.string}')"
    return untokenize(tokens)

To do