Skip to content

Commit 7d8a798

Browse files
committed
Simplify code using proc_macro::Group::span_{open,close}.
1 parent 1b5007e commit 7d8a798

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

macros/src/embed_python.rs

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,9 @@ impl EmbedPython {
2424
}
2525
}
2626

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();
2830
if line > self.line {
2931
while line > self.line {
3032
self.python.push('\n');
@@ -52,7 +54,7 @@ impl EmbedPython {
5254

5355
while let Some(token) = tokens.next() {
5456
let span = token.span();
55-
self.add_whitespace(span, span.line(), span.column())?;
57+
self.add_whitespace(span)?;
5658

5759
match &token {
5860
TokenTree::Group(x) => {
@@ -62,11 +64,11 @@ impl EmbedPython {
6264
Delimiter::Bracket => ("[", "]"),
6365
Delimiter::None => ("", ""),
6466
};
67+
self.add_whitespace(x.span_open())?;
6568
self.python.push_str(start);
6669
self.column += start.len();
6770
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())?;
7072
self.python.push_str(end);
7173
self.column += end.len();
7274
}

0 commit comments

Comments
 (0)