ꦫꦣꦺꦤ꧀ꦄꦤ꧀ꦠꦱꦺꦤ

PATH: usr / lib64 / python3.6 / lib2to3 / fixes

"""Fixer for operator functions.

operator.isCallable(obj)       -> hasattr(obj, '__call__')
operator.sequenceIncludes(obj) -> operator.contains(obj)
operator.isSequenceType(obj)   -> isinstance(obj, collections.Sequence)
operator.isMappingType(obj)    -> isinstance(obj, collections.Mapping)
operator.isNumberType(obj)     -> isinstance(obj, numbers.Number)
operator.repeat(obj, n)        -> operator.mul(obj, n)
operator.irepeat(obj, n)       -> operator.imul(obj, n)
"""

import collections

# Local imports
from lib2to3 import fixer_base
from lib2to3.fixer_util import Call, Name, String, touch_import

def invocation(s):
    def dec(f):
        f.invocation = s
        return f
    return dec

class FixOperator(fixer_base.BaseFix):
    BM_compatible = True
    order = "pre"

methods = """
              method=('isCallable'|'sequenceIncludes'
                     |'isSequenceType'|'isMappingType'|'isNumberType'
                     |'repeat'|'irepeat')
              """
    obj = "'(' obj=any ')'"
    PATTERN = """
              power< module='operator'
                trailer< '.' %(methods)s > trailer< %(obj)s > >
              |
              power< %(methods)s trailer< %(obj)s > >
              """ % dict(methods=methods, obj=obj)

def transform(self, node, results):
        method = self._check_method(node, results)
        if method is not None:
            return method(node, results)

@invocation("operator.contains(%s)")
    def _sequenceIncludes(self, node, results):
        return self._handle_rename(node, results, "contains")

@invocation("hasattr(%s, '__call__')")
    def _isCallable(self, node, results):
        obj = results["obj"]
        args = [obj.clone(), String(", "), String("'__call__'")]
        return Call(Name("hasattr"), args, prefix=node.prefix)

@invocation("operator.mul(%s)")
    def _repeat(self, node, results):
        return self._handle_rename(node, results, "mul")

@invocation("operator.imul(%s)")
    def _irepeat(self, node, results):
        return self._handle_rename(node, results, "imul")

@invocation("isinstance(%s, collections.Sequence)")
    def _isSequenceType(self, node, results):
        return self._handle_type2abc(node, results, "collections", "Sequence")

@invocation("isinstance(%s, collections.Mapping)")
    def _isMappingType(self, node, results):
        return self._handle_type2abc(node, results, "collections", "Mapping")

@invocation("isinstance(%s, numbers.Number)")
    def _isNumberType(self, node, results):
        return self._handle_type2abc(node, results, "numbers", "Number")

def _handle_rename(self, node, results, name):
        method = results["method"][0]
        method.value = name
        method.changed()

def _handle_type2abc(self, node, results, module, abc):
        touch_import(None, module, node)
        obj = results["obj"]
        args = [obj.clone(), String(", " + ".".join([module, abc]))]
        return Call(Name("isinstance"), args, prefix=node.prefix)

def _check_method(self, node, results):
        method = getattr(self, "_" + results["method"][0].value)
        if isinstance(method, collections.Callable):
            if "module" in results:
                return method
            else:
                sub = (str(results["obj"]),)
                invocation_str = method.invocation % sub
                self.warning(node, "You should use '%s' here." % invocation_str)
        return None

[-] fix_urllib.py [edit]
[-] fix_standarderror.py [edit]
[-] fix_imports.py [edit]
[-] fix_getcwdu.py [edit]
[-] fix_nonzero.py [edit]
[-] fix_itertools.py [edit]
[-] fix_xrange.py [edit]
[-] fix_metaclass.py [edit]
[-] fix_methodattrs.py [edit]
[+] ..
[-] fix_ne.py [edit]
[-] fix_repr.py [edit]
[-] fix_zip.py [edit]
[-] fix_exec.py [edit]
[-] fix_print.py [edit]
[-] fix_paren.py [edit]
[-] fix_imports2.py [edit]
[-] fix_apply.py [edit]
[-] fix_ws_comma.py [edit]
[-] fix_dict.py [edit]
[-] fix_buffer.py [edit]
[-] fix_intern.py [edit]
[-] fix_next.py [edit]
[-] fix_itertools_imports.py [edit]
[-] fix_throw.py [edit]
[+] __pycache__
[-] fix_long.py [edit]
[-] fix_raw_input.py [edit]
[-] fix_future.py [edit]
[-] fix_exitfunc.py [edit]
[-] fix_renames.py [edit]
[-] fix_input.py [edit]
[-] fix_reduce.py [edit]
[-] fix_operator.py [edit]
[-] fix_numliterals.py [edit]
[-] fix_isinstance.py [edit]
[-] fix_basestring.py [edit]
[-] fix_filter.py [edit]
[-] fix_raise.py [edit]
[-] fix_types.py [edit]
[-] fix_funcattrs.py [edit]
[-] fix_set_literal.py [edit]
[-] __init__.py [edit]
[-] fix_execfile.py [edit]
[-] fix_unicode.py [edit]
[-] fix_reload.py [edit]
[-] fix_import.py [edit]
[-] fix_sys_exc.py [edit]
[-] fix_asserts.py [edit]
[-] fix_map.py [edit]
[-] fix_tuple_params.py [edit]
[-] fix_xreadlines.py [edit]
[-] fix_except.py [edit]
[-] fix_has_key.py [edit]
[-] fix_idioms.py [edit]