python_ast/ast/tree/
tuple.rs

1use proc_macro2::TokenStream;
2use pyo3::{Borrowed, FromPyObject, PyAny, PyResult};
3use quote::quote;
4use serde::{Deserialize, Serialize};
5
6use crate::{
7    CodeGen, CodeGenContext, ExprType, PythonOptions, SymbolTableScopes,
8    Node, impl_node_with_positions, extract_list
9};
10
11#[derive(Clone, Debug, Serialize, Deserialize, PartialEq)]
12pub struct Tuple {
13    pub elts: Vec<ExprType>,
14    pub lineno: Option<usize>,
15    pub col_offset: Option<usize>,
16    pub end_lineno: Option<usize>,
17    pub end_col_offset: Option<usize>,
18}
19
20impl<'a, 'py> FromPyObject<'a, 'py> for Tuple {
21    type Error = pyo3::PyErr;
22    fn extract(ob: Borrowed<'a, 'py, PyAny>) -> PyResult<Self> {
23        let elts: Vec<ExprType> = extract_list(&ob, "elts", "tuple elements")?;
24        
25        Ok(Tuple {
26            elts,
27            lineno: ob.lineno(),
28            col_offset: ob.col_offset(),
29            end_lineno: ob.end_lineno(),
30            end_col_offset: ob.end_col_offset(),
31        })
32    }
33}
34
35impl_node_with_positions!(Tuple { lineno, col_offset, end_lineno, end_col_offset });
36
37impl CodeGen for Tuple {
38    type Context = CodeGenContext;
39    type Options = PythonOptions;
40    type SymbolTable = SymbolTableScopes;
41
42    fn to_rust(
43        self,
44        ctx: Self::Context,
45        options: Self::Options,
46        symbols: Self::SymbolTable,
47    ) -> Result<TokenStream, Box<dyn std::error::Error>> {
48        let elements: Result<Vec<_>, _> = self.elts
49            .into_iter()
50            .map(|elt| elt.to_rust(ctx.clone(), options.clone(), symbols.clone()))
51            .collect();
52        
53        let elements = elements?;
54        
55        // Generate appropriate tuple syntax based on element count
56        match elements.len() {
57            0 => Ok(quote! { () }),
58            1 => {
59                let element = &elements[0];
60                Ok(quote! { (#element,) }) // Single element tuple needs trailing comma
61            },
62            _ => Ok(quote! { (#(#elements),*) }),
63        }
64    }
65}
66
67#[cfg(test)]
68mod tests {
69    use super::*;
70    use crate::create_parse_test;
71
72    create_parse_test!(test_empty_tuple, "()", "tuple_test.py");
73    create_parse_test!(test_single_tuple, "(1,)", "tuple_test.py");
74    create_parse_test!(test_simple_tuple, "(1, 2, 3)", "tuple_test.py");
75    create_parse_test!(test_tuple_with_variables, "(x, y, z)", "tuple_test.py");
76}
python_ast/ast/tree/tuple.rs

python_ast/ast/tree/
tuple.rs
