Current Dir: /snap/core22/2955/usr/lib/python3.11/lib2to3/fixes/
[FILE] fix_apply.py [ edit | delete | rename ]
[FILE] fix_asserts.py [ edit | delete | rename ]
[FILE] fix_basestring.py [ edit | delete | rename ]
[FILE] fix_buffer.py [ edit | delete | rename ]
[FILE] fix_dict.py [ edit | delete | rename ]
[FILE] fix_except.py [ edit | delete | rename ]
[FILE] fix_exec.py [ edit | delete | rename ]
[FILE] fix_execfile.py [ edit | delete | rename ]
[FILE] fix_exitfunc.py [ edit | delete | rename ]
[FILE] fix_filter.py [ edit | delete | rename ]
[FILE] fix_funcattrs.py [ edit | delete | rename ]
[FILE] fix_future.py [ edit | delete | rename ]
[FILE] fix_getcwdu.py [ edit | delete | rename ]
[FILE] fix_has_key.py [ edit | delete | rename ]
[FILE] fix_idioms.py [ edit | delete | rename ]
[FILE] fix_import.py [ edit | delete | rename ]
[FILE] fix_imports.py [ edit | delete | rename ]
[FILE] fix_imports2.py [ edit | delete | rename ]
[FILE] fix_input.py [ edit | delete | rename ]
[FILE] fix_intern.py [ edit | delete | rename ]
[FILE] fix_isinstance.py [ edit | delete | rename ]
[FILE] fix_itertools.py [ edit | delete | rename ]
[FILE] fix_itertools_imports.py [ edit | delete | rename ]
[FILE] fix_long.py [ edit | delete | rename ]
[FILE] fix_map.py [ edit | delete | rename ]
[FILE] fix_metaclass.py [ edit | delete | rename ]
[FILE] fix_methodattrs.py [ edit | delete | rename ]
[FILE] fix_ne.py [ edit | delete | rename ]
[FILE] fix_next.py [ edit | delete | rename ]
[FILE] fix_nonzero.py [ edit | delete | rename ]
[FILE] fix_numliterals.py [ edit | delete | rename ]
[FILE] fix_operator.py [ edit | delete | rename ]
[FILE] fix_paren.py [ edit | delete | rename ]
[FILE] fix_print.py [ edit | delete | rename ]
[FILE] fix_raise.py [ edit | delete | rename ]
[FILE] fix_raw_input.py [ edit | delete | rename ]
[FILE] fix_reduce.py [ edit | delete | rename ]
[FILE] fix_reload.py [ edit | delete | rename ]
[FILE] fix_renames.py [ edit | delete | rename ]
[FILE] fix_repr.py [ edit | delete | rename ]
[FILE] fix_set_literal.py [ edit | delete | rename ]
[FILE] fix_standarderror.py [ edit | delete | rename ]
[FILE] fix_sys_exc.py [ edit | delete | rename ]
[FILE] fix_throw.py [ edit | delete | rename ]
[FILE] fix_tuple_params.py [ edit | delete | rename ]
[FILE] fix_types.py [ edit | delete | rename ]
[FILE] fix_unicode.py [ edit | delete | rename ]
[FILE] fix_urllib.py [ edit | delete | rename ]
[FILE] fix_ws_comma.py [ edit | delete | rename ]
[FILE] fix_xrange.py [ edit | delete | rename ]
[FILE] fix_xreadlines.py [ edit | delete | rename ]
[FILE] fix_zip.py [ edit | delete | rename ]
[FILE] __init__.py [ edit | delete | rename ]
Viewing: /snap/core22/2955/usr/lib/python3.11/lib2to3/fixes/fix_itertools.py
""" Fixer for itertools.(imap|ifilter|izip) --> (map|filter|zip) and
itertools.ifilterfalse --> itertools.filterfalse (bugs 2360-2363)
imports from itertools are fixed in fix_itertools_import.py
If itertools is imported as something else (ie: import itertools as it;
it.izip(spam, eggs)) method calls will not get fixed.
"""
# Local imports
from .. import fixer_base
from ..fixer_util import Name
class FixItertools(fixer_base.BaseFix):
BM_compatible = True
it_funcs = "('imap'|'ifilter'|'izip'|'izip_longest'|'ifilterfalse')"
PATTERN = """
power< it='itertools'
trailer<
dot='.' func=%(it_funcs)s > trailer< '(' [any] ')' > >
|
power< func=%(it_funcs)s trailer< '(' [any] ')' > >
""" %(locals())
# Needs to be run after fix_(map|zip|filter)
run_order = 6
def transform(self, node, results):
prefix = None
func = results['func'][0]
if ('it' in results and
func.value not in ('ifilterfalse', 'izip_longest')):
dot, it = (results['dot'], results['it'])
# Remove the 'itertools'
prefix = it.prefix
it.remove()
# Replace the node which contains ('.', 'function') with the
# function (to be consistent with the second part of the pattern)
dot.remove()
func.parent.replace(func)
prefix = prefix or func.prefix
func.replace(Name(func.value[1:], prefix=prefix))