summaryrefslogtreecommitdiffstats
path: root/Lib/lib2to3/fixes/fix_itertools.py
diff options
context:
space:
mode:
authorBenjamin Peterson <benjamin@python.org>2011-03-08 04:50:37 (GMT)
committerBenjamin Peterson <benjamin@python.org>2011-03-08 04:50:37 (GMT)
commit7988ba33d9e700f42c353be40b33a0d7366ad648 (patch)
treee60ea12800bb75022063830cd80450668d000822 /Lib/lib2to3/fixes/fix_itertools.py
parent4bcfb92168aeb358479f9149461ac23983a50232 (diff)
downloadcpython-7988ba33d9e700f42c353be40b33a0d7366ad648.zip
cpython-7988ba33d9e700f42c353be40b33a0d7366ad648.tar.gz
cpython-7988ba33d9e700f42c353be40b33a0d7366ad648.tar.bz2
transform izip_longest #11424
Diffstat (limited to 'Lib/lib2to3/fixes/fix_itertools.py')
-rw-r--r--Lib/lib2to3/fixes/fix_itertools.py5
1 files changed, 3 insertions, 2 deletions
diff --git a/Lib/lib2to3/fixes/fix_itertools.py b/Lib/lib2to3/fixes/fix_itertools.py
index 80790bf..63346b9 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