File tree Expand file tree Collapse file tree 1 file changed +9
-3
lines changed Expand file tree Collapse file tree 1 file changed +9
-3
lines changed Original file line number Diff line number Diff line change 8
8
9
9
10
10
class FixClassicDivision (fixer_base .BaseFix ):
11
- _accept_type = token .SLASH
11
+ PATTERN = '''
12
+ '/=' | '/'
13
+ '''
12
14
13
15
def start_tree (self , tree , name ):
14
16
super (FixClassicDivision , self ).start_tree (tree , name )
15
17
self .skip = "division" in tree .future_features
16
18
17
19
def match (self , node ):
18
- return node .value == "/"
20
+ return node .value in ( '/' , '/=' )
19
21
20
22
def transform (self , node , results ):
21
23
if self .skip :
22
24
return
23
25
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 )
You can’t perform that action at this time.
0 commit comments