Annotate AST trees with source code positions
Project description
The asttokens module annotates Python abstract syntax trees (ASTs) with the positions of tokens and text in the source code that generated them.
It makes it possible for tools that work with logical AST nodes to find the particular text that resulted in those nodes, for example for automated refactoring or highlighting.
Installation
asttokens is available on PyPI: https://pypi-hypernode.com/pypi/asttokens/:
pip install asttokens
The code is on GitHub: https://github.com/gristlabs/asttokens.
The API Reference is here: http://asttokens.readthedocs.io/en/latest/api-index.html.
Usage
ASTTokens works with both Python2 and Python3.
ASTTokens can annotate both trees built by ast, AND those built by astroid.
Here’s an example:
import asttokens, ast
source = "Robot('blue').walk(steps=10*n)"
atok = asttokens.ASTTokens(source, parse=True)Once the tree has been marked, nodes get .first_token, .last_token attributes, and the ASTTokens object offers helpful methods:
attr_node = next(n for n in ast.walk(atok.tree) if isinstance(n, ast.Attribute))
print(atok.get_text(attr_node))
start, end = attr_node.last_token.startpos, attr_node.last_token.endpos
print(atok.text[:start] + 'RUN' + atok.text[end:])Which produces this output:
Robot('blue').walk
Robot('blue').RUN(steps=10*n)The ASTTokens object also offers methods to walk and search the list of tokens that make up the code (or a particular AST node), which is more useful and powerful than dealing with the text directly.
Tests
Tests are in the tests/ subdirectory. To run all tests, run:
nosetests
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
File details
Details for the file asttokens-2.0.1.tar.gz.
File metadata
- Download URL: asttokens-2.0.1.tar.gz
- Upload date:
- Size: 38.9 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/1.12.1 pkginfo/1.5.0.1 requests/2.22.0 setuptools/41.0.1 requests-toolbelt/0.8.0 tqdm/4.29.1 CPython/3.7.4
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 8ba21b5ef63dc0bbae49ccd50badef00469c4f17dbc5ef6db8add900a2f991f7 |
|
| MD5 | 616b735e7c48ae13627751e6de3583b2 |
|
| BLAKE2b-256 | 40ab92a2cd544c1b9d53b00c955a7a6f4a00890dd74b7e0b279ae43e4dc07c68 |
File details
Details for the file asttokens-2.0.1-py2.py3-none-any.whl.
File metadata
- Download URL: asttokens-2.0.1-py2.py3-none-any.whl
- Upload date:
- Size: 20.1 kB
- Tags: Python 2, Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/1.12.1 pkginfo/1.5.0.1 requests/2.22.0 setuptools/41.0.1 requests-toolbelt/0.8.0 tqdm/4.29.1 CPython/3.7.4
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | e777af3c2c87f3dfc7961e0f935ff901d4dade8770ba428631c343cc973bf957 |
|
| MD5 | f5453fdef684b0c66f9bb542d969ae43 |
|
| BLAKE2b-256 | d6ef18538c0bf7878902e866d65aeb2a7e57fd520133931722b6a63fb2b24567 |
