semvec/pypi-packages huggingface dataset:
astate 0.0.7
Various AST utils
License: apache-2.0

# astate

Various AST utils

To install:	```pip install astate```

# Examples


## re_find_in_body

Recursively find elements in an ast body that satisfy a condition

```python
>>> from astate import re_find_in_body
>>> def foo(x):
...     def bar(y):
...         y * 1
...     return bar(x)
>>>
>>> [x.name for x in re_find_in_body(foo)]
['foo', 'bar']
>>> [x.name for x in re_find_in_body(foo, max_levels=1)]
['foo']
```
    
## remove_docstrings

Remove docstrings from the given Python code.

```python
>>> from astate import remove_docstrings
>>> code = '''
... def foo():
...     """This is a docstring."""
...     print("Hello")
...
... class Bar:
...     """Class docstring."""
...     def baz(self):
...         """Method docstring."""
...         pass
... '''
>>> print(remove_docstrings(code))
def foo():
    print('Hello')
<BLANKLINE>
class Bar:
<BLANKLINE>
    def baz(self):
        pass
```