diff options
| author | Benjamin Peterson <benjamin@python.org> | 2011-03-07 22:50:37 -0600 |
|---|---|---|
| committer | Benjamin Peterson <benjamin@python.org> | 2011-03-07 22:50:37 -0600 |
| commit | 49d7149e6d12ee878fcf30d703150a5d00f3908d (patch) | |
| tree | 14ab9f55ebf37924c78f5a0153848f945ebd012b /Lib/lib2to3/fixes/fix_itertools.py | |
| parent | 3f84b07816fc63dd4996146bdd757f6bc81c54a9 (diff) | |
| download | cpython-git-49d7149e6d12ee878fcf30d703150a5d00f3908d.tar.gz | |
transform izip_longest #11424
Diffstat (limited to 'Lib/lib2to3/fixes/fix_itertools.py')
| -rw-r--r-- | Lib/lib2to3/fixes/fix_itertools.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/Lib/lib2to3/fixes/fix_itertools.py b/Lib/lib2to3/fixes/fix_itertools.py index 80790bff2e..63346b9220 100644 --- a/Lib/lib2to3/fixes/fix_itertools.py +++ b/Lib/lib2to3/fixes/fix_itertools.py @@ -13,7 +13,7 @@ from ..fixer_util import Name class FixItertools(fixer_base.BaseFix): BM_compatible = True - it_funcs = "('imap'|'ifilter'|'izip'|'ifilterfalse')" + it_funcs = "('imap'|'ifilter'|'izip'|'izip_longest'|'ifilterfalse')" PATTERN = """ power< it='itertools' trailer< @@ -28,7 +28,8 @@ class FixItertools(fixer_base.BaseFix): def transform(self, node, results): prefix = None func = results['func'][0] - if 'it' in results and func.value != 'ifilterfalse': + if ('it' in results and + func.value not in ('ifilterfalse', 'izip_longest')): dot, it = (results['dot'], results['it']) # Remove the 'itertools' prefix = it.prefix |