PCQRSCANER/venv/Lib/site-packages/libpasteurize/fixes/fix_division.py
2019-12-22 21:51:47 +01:00

29 lines
904 B
Python

u"""
Fixer for division: from __future__ import division if needed
"""
from lib2to3 import fixer_base
from libfuturize.fixer_util import token, future_import
def match_division(node):
u"""
__future__.division redefines the meaning of a single slash for division,
so we match that and only that.
"""
slash = token.SLASH
return node.type == slash and not node.next_sibling.type == slash and \
not node.prev_sibling.type == slash
class FixDivision(fixer_base.BaseFix):
run_order = 4 # this seems to be ignored?
def match(self, node):
u"""
Since the tree needs to be fixed once and only once if and only if it
matches, then we can start discarding matches after we make the first.
"""
return match_division(node)
def transform(self, node, results):
future_import(u"division", node)