Source code for ideas.examples.function_keyword

"""This module enables someone to use ``function`` as a keyword
equivalent to ``lambda``.
from ideas import import_hook
import token_utils

[docs]def transform_source(source, **_kwargs): """This performs a simple replacement of ``function`` by ``lambda``.""" new_tokens = [] for token in token_utils.tokenize(source): # token_utils allows us to easily replace the string content # of any token if token == "function": token.string = "lambda" new_tokens.append(token) return token_utils.untokenize(new_tokens)
[docs]def add_hook(**_kwargs): """Creates and automatically adds the import hook in sys.meta_path""" hook = import_hook.create_hook( transform_source=transform_source, hook_name=__name__, # optional ) return hook