41 lines
1.4 KiB
Python
41 lines
1.4 KiB
Python
"""Fixer for operator.{isCallable,sequenceIncludes}
|
|
|
|
operator.isCallable(obj) -> hasattr(obj, '__call__')
|
|
operator.sequenceIncludes(obj) -> operator.contains(obj)
|
|
"""
|
|
|
|
# Local imports
|
|
from .. import fixer_base
|
|
from ..fixer_util import Call, Name, String
|
|
|
|
class FixOperator(fixer_base.BaseFix):
|
|
|
|
methods = "method=('isCallable'|'sequenceIncludes')"
|
|
func = "'(' func=any ')'"
|
|
PATTERN = """
|
|
power< module='operator'
|
|
trailer< '.' {methods} > trailer< {func} > >
|
|
|
|
|
power< {methods} trailer< {func} > >
|
|
""".format(methods=methods, func=func)
|
|
|
|
def transform(self, node, results):
|
|
method = results["method"][0]
|
|
|
|
if method.value == "sequenceIncludes":
|
|
if "module" not in results:
|
|
# operator may not be in scope, so we can't make a change.
|
|
self.warning(node, "You should use operator.contains here.")
|
|
else:
|
|
method.value = "contains"
|
|
method.changed()
|
|
elif method.value == "isCallable":
|
|
if "module" not in results:
|
|
self.warning(node,
|
|
"You should use hasattr(%s, '__call__') here." %
|
|
results["func"].value)
|
|
else:
|
|
func = results["func"]
|
|
args = [func.clone(), String(", "), String("'__call__'")]
|
|
return Call(Name("hasattr"), args, prefix=node.prefix)
|