spydecy_python/
type_extractor.rs

1//! Type hint extraction from Python AST
2//!
3//! This module extracts type annotations from Python code and converts them
4//! to Spydecy's type system.
5
6use crate::parser::PythonAST;
7use anyhow::Result;
8use spydecy_hir::types::Type;
9
10/// Extract type hints from Python AST
11///
12/// # Errors
13///
14/// Returns an error if type hints cannot be extracted
15pub fn extract_type_hints(ast: &PythonAST) -> Result<Vec<(String, Type)>> {
16    let mut type_hints = Vec::new();
17
18    // Walk the AST and extract type annotations
19    extract_type_hints_recursive(ast, &mut type_hints)?;
20
21    Ok(type_hints)
22}
23
24#[allow(clippy::unnecessary_wraps)]
25fn extract_type_hints_recursive(
26    _ast: &PythonAST,
27    _type_hints: &mut Vec<(String, Type)>,
28) -> Result<()> {
29    // Implementation will come in Sprint 2
30    Ok(())
31}
32
33#[cfg(test)]
34mod tests {
35    use super::*;
36
37    #[test]
38    fn test_extract_type_hints() {
39        let ast = PythonAST::new("Module".to_string());
40        let result = extract_type_hints(&ast);
41        assert!(result.is_ok());
42    }
43}
spydecy_python/type_extractor.rs

spydecy_python/
type_extractor.rs
