blob: 8a43882e94bb408209e6ff924a3f362382ae640d (
about) (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
|
"""
Fixer for input.
Does a check for `from builtins import input` before running the lib2to3 fixer.
The fixer will not run when the input is already present.
this:
a = input()
becomes:
from builtins import input
a = eval(input())
and this:
from builtins import input
a = input()
becomes (no change):
from builtins import input
a = input()
"""
import lib2to3.fixes.fix_input
from lib2to3.fixer_util import does_tree_import
class FixInput(lib2to3.fixes.fix_input.FixInput):
def transform(self, node, results):
if does_tree_import('builtins', 'input', node):
return
return super(FixInput, self).transform(node, results)
|
