omnitype 0.1.0

A hybrid type checker for Python and other dynamic languages
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

//! Utility functions and helpers for omnitype.

use std::path::{Path, PathBuf};

use walkdir::WalkDir;

/// Returns an iterator over all Python files in the given directory.
pub fn find_python_files<P: AsRef<Path>>(path: P) -> impl Iterator<Item = PathBuf> {
    WalkDir::new(path)
        .into_iter()
        .filter_map(Result::ok)
        .filter(|entry| {
            let path = entry.path();
            path.is_file() && path.extension().is_some_and(|ext| ext == "py")
        })
        .map(|entry| entry.path().to_path_buf())
}

/// Converts a path to a module name.
/// Returns None if the path ends with a trailing slash (indicating a directory)
/// or has no file stem.
pub fn path_to_module_name(path: &Path) -> Option<String> {
    // Check if the path ends with a separator (indicates a directory)
    if path.to_str()?.ends_with(std::path::MAIN_SEPARATOR) || path.to_str()?.ends_with('/') {
        return None;
    }

    // Get the file stem (without extension)
    path.file_stem()
        .and_then(|s| s.to_str())
        .map(|s| s.to_string())
}

#[cfg(test)]
mod tests {
    use super::*;
    use std::fs::File;
    use tempfile::tempdir;

    #[test]
    fn test_find_python_files() {
        let temp_dir = tempdir().unwrap();
        let dir_path = temp_dir.path();

        // Create test files
        let _ = File::create(dir_path.join("test1.py")).unwrap();
        let _ = File::create(dir_path.join("test2.py")).unwrap();
        let _ = File::create(dir_path.join("not_python.txt")).unwrap();

        // Test finding Python files
        let python_files: Vec<_> = find_python_files(dir_path).collect();
        assert_eq!(python_files.len(), 2);

        // Cleanup
        temp_dir.close().unwrap();
    }

    #[test]
    fn test_path_to_module_name() {
        assert_eq!(
            path_to_module_name(Path::new("/path/to/module.py")),
            Some("module".to_string())
        );
        assert_eq!(path_to_module_name(Path::new("module.py")), Some("module".to_string()));
        assert_eq!(path_to_module_name(Path::new("/path/to/dir/")), None);
    }
}