File Explorer

1# Copyright 2008 Armin Ronacher.2# Licensed to PSF under a Contributor Agreement.3 4"""Fixer that cleans up a tuple argument to isinstance after the tokens5in it were fixed.  This is mainly used to remove double occurrences of6tokens as a leftover of the long -> int / unicode -> str conversion.7 8eg.  isinstance(x, (int, long)) -> isinstance(x, (int, int))9       -> isinstance(x, int)10"""11 12from .. import fixer_base13from ..fixer_util import token14 15 16class FixIsinstance(fixer_base.BaseFix):17    BM_compatible = True18    PATTERN = """19    power<20        'isinstance'21        trailer< '(' arglist< any ',' atom< '('22            args=testlist_gexp< any+ >23        ')' > > ')' >24    >25    """26 27    run_order = 628 29    def transform(self, node, results):30        names_inserted = set()31        testlist = results["args"]32        args = testlist.children33        new_args = []34        iterator = enumerate(args)35        for idx, arg in iterator:36            if arg.type == token.NAME and arg.value in names_inserted:37                if idx < len(args) - 1 and args[idx + 1].type == token.COMMA:38                    next(iterator)39                    continue40            else:41                new_args.append(arg)42                if arg.type == token.NAME:43                    names_inserted.add(arg.value)44        if new_args and new_args[-1].type == token.COMMA:45            del new_args[-1]46        if len(new_args) == 1:47            atom = testlist.parent48            new_args[0].prefix = atom.prefix49            atom.replace(new_args[0])50        else:51            args[:] = new_args52            node.changed()53