# Copyright 2008 Armin Ronacher.
|
# Licensed to PSF under a Contributor Agreement.
|
|
"""Fixer for reduce().
|
|
Makes sure reduce() is imported from the functools module if reduce is
|
used in that module.
|
"""
|
|
from lib2to3 import fixer_base
|
from lib2to3.fixer_util import touch_import
|
|
|
|
class FixReduce(fixer_base.BaseFix):
|
|
BM_compatible = True
|
order = "pre"
|
|
PATTERN = """
|
power< 'reduce'
|
trailer< '('
|
arglist< (
|
(not(argument<any '=' any>) any ','
|
not(argument<any '=' any>) any) |
|
(not(argument<any '=' any>) any ','
|
not(argument<any '=' any>) any ','
|
not(argument<any '=' any>) any)
|
) >
|
')' >
|
>
|
"""
|
|
def transform(self, node, results):
|
touch_import(u'functools', u'reduce', node)
|
