1 """Fixer for __nonzero__ -> __bool__ methods."""

     2 # Author: Collin Winter

     3 

     4 # Local imports

     5 from .. import fixer_base

     6 from ..fixer_util import Name, syms

     7 

     8 class FixNonzero(fixer_base.BaseFix):

     9     PATTERN = """

    10     classdef< 'class' any+ ':'

    11               suite< any*

    12                      funcdef< 'def' name='__nonzero__'

    13                               parameters< '(' NAME ')' > any+ >

    14                      any* > >

    15     """

    16 

    17     def transform(self, node, results):

    18         name = results["name"]

    19         new = Name("__bool__", prefix=name.get_prefix())

    20         name.replace(new)