Skip to content

Commit 1ea5e83

Browse files
author
Larry Huang
committed
fix SLASHEQUAL in fix_classic_division
1 parent de255c6 commit 1ea5e83

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

libmodernize/fixes/fix_classic_division.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,17 +8,23 @@
88

99

1010
class FixClassicDivision(fixer_base.BaseFix):
11-
_accept_type = token.SLASH
11+
PATTERN = '''
12+
'/=' | '/'
13+
'''
1214

1315
def start_tree(self, tree, name):
1416
super(FixClassicDivision, self).start_tree(tree, name)
1517
self.skip = "division" in tree.future_features
1618

1719
def match(self, node):
18-
return node.value == "/"
20+
return node.value in ('/', '/=')
1921

2022
def transform(self, node, results):
2123
if self.skip:
2224
return
2325
libmodernize.add_future(node, u'division')
24-
return pytree.Leaf(token.SLASH, "//", prefix=node.prefix)
26+
27+
if node.value == '/':
28+
return pytree.Leaf(token.DOUBLESLASH, '//', prefix=node.prefix)
29+
else:
30+
return pytree.Leaf(token.DOUBLESLASHEQUAL, '//=', prefix=node.prefix)

0 commit comments

Comments
 (0)