Parse a Python AST and delete stuff from it

deletestuffast.py

import ast

class RemoveMethod(ast.NodeTransformer):
    '''
    Removes all occurences of a method from the AST
    Example:
    ---
    tree = RemoveMethod('bar').visit(tree)

    x = foo.bar()           => x = foo
    y = meow('hi').bar()    => y = meow('hi')
    '''
    def __init__(self, func):
        self.func = func

    def visit_Call(self, node):
        transform = node
        try:
            if node.func.attr == self.func:
                if not len(node.args):
                    transform = node.func.value
        except AttributeError:
            pass
        return ast.copy_location(transform, node)

tree = RemoveMethod('bar').visit(tree)