@@ -24,7 +24,9 @@ impl EmbedPython {
24
24
}
25
25
}
26
26
27
- fn add_whitespace ( & mut self , span : Span , line : usize , column : usize ) -> Result < ( ) , TokenStream > {
27
+ fn add_whitespace ( & mut self , span : Span ) -> Result < ( ) , TokenStream > {
28
+ let line = span. line ( ) ;
29
+ let column = span. column ( ) ;
28
30
if line > self . line {
29
31
while line > self . line {
30
32
self . python . push ( '\n' ) ;
@@ -52,7 +54,7 @@ impl EmbedPython {
52
54
53
55
while let Some ( token) = tokens. next ( ) {
54
56
let span = token. span ( ) ;
55
- self . add_whitespace ( span, span . line ( ) , span . column ( ) ) ?;
57
+ self . add_whitespace ( span) ?;
56
58
57
59
match & token {
58
60
TokenTree :: Group ( x) => {
@@ -62,11 +64,11 @@ impl EmbedPython {
62
64
Delimiter :: Bracket => ( "[" , "]" ) ,
63
65
Delimiter :: None => ( "" , "" ) ,
64
66
} ;
67
+ self . add_whitespace ( x. span_open ( ) ) ?;
65
68
self . python . push_str ( start) ;
66
69
self . column += start. len ( ) ;
67
70
self . add ( x. stream ( ) ) ?;
68
- let end_span = token. span ( ) . end ( ) ;
69
- self . add_whitespace ( span, end_span. line ( ) , end_span. column ( ) . saturating_sub ( end. len ( ) ) ) ?;
71
+ self . add_whitespace ( x. span_close ( ) ) ?;
70
72
self . python . push_str ( end) ;
71
73
self . column += end. len ( ) ;
72
74
}
0 commit comments