Need help with pasta?
Click the “chat” button below for chat support from the developer who created it, or find similar developers for support.

About the developer

google
237 Stars 30 Forks Apache License 2.0 156 Commits 6 Opened issues

Description

Library to refactor python code through AST manipulation.

Services available

!
?

Need anything else?

Contributors list

pasta: Python AST Augmentation

This is still a work-in-progress; there is much more to do. Existing functionality may not be perfect.

Mission

Enable python source code refactoring through AST modifications.

Sample use cases:

  • Facilitate moving or renaming python modules by rewriting import statements.
  • Refactor code to enforce a certain style, such as reordering function definitions.
  • Safely migrate code from one API to another.

Design Goals

  • Symmetry: Given any input source, it should hold that
    pasta.dump(pasta.parse(src)) == src
    .
  • Mutability: Any changes made in the AST are reflected in the code generated from it.
  • Standardization: The syntax tree parsed by pasta will not introduce new nodes or structure that the user must learn.

Python Version Support

Supports python

2.7
and up to
3.8
.

Dependencies

pasta
depends on
six
.

Basic Usage

import pasta
tree = pasta.parse(source_code)

... Augment contents of tree ...

source_code = pasta.dump(tree)

Built-in Augmentations

Pasta includes some common augmentations out-of-the-box. These can be used as building blocks for more complex refactoring actions.

There will be more of these basic augmentations added over time. Stay tuned!

Rename an imported name

Rewrites references to an imported name, module or package. For some more examples, see

pasta/augment/rename_test.py
.

# Rewrite references from one module to another
rename.rename_external(tree, 'pkg.subpkg.module', 'pkg.other_module')

Rewrite references from one package to another

rename.rename_external(tree, 'pkg.subpkg', 'pkg.other_pkg')

Rewrite references to an imported name in another module

rename.rename_external(tree, 'pkg.module.Query', 'pkg.module.ExecuteQuery')

Known issues and limitations

  • Changing the indentation level of a block of code is not supported. This is not an issue for renames, but would cause problems for refactors like extracting a method.

  • pasta
    works under the assumption that the python version that the code is written for and the version used to run pasta are the same. This is because pasta relies on
    ast.parse
  • Some python features are not fully supported, including

    global
    .

Developing

This project uses

setuptools
to facilitate testing and packaging.

# Run all tests
python setup.py test

Run a single test suite

python setup.py test -s pasta.base.annotate_test.suite

Disclaimer

This is not an official Google product.

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.