pyimportparse 0.2.3

A rust crate to parse python imports
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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88

use itertools::Itertools;
use pretty_assertions::assert_eq;
use serde::{Deserialize, Serialize};
use std::collections::{HashMap, HashSet};
use std::fs;
use walkdir::{DirEntry, WalkDir};

use pyimportparse::parse_imports;

fn main() {
    let mut data = HashMap::new();

    for entry in WalkDir::new("vendor/django/django")
        .into_iter()
        .filter_entry(|e| !is_hidden(e))
    {
        let entry = entry.unwrap();
        if entry.file_type().is_dir() {
            continue;
        }
        if !entry.file_name().to_str().unwrap().ends_with(".py") {
            continue;
        }

        let code = fs::read_to_string(entry.path()).unwrap();
        let imports = parse_imports(&code).unwrap();

        data.insert(
            entry.path().to_str().unwrap().to_owned(),
            imports
                .into_iter()
                .map(|i| SerializableImport {
                    imported_object: i.imported_object,
                    typechecking_only: i.typechecking_only,
                })
                .collect(),
        );
    }

    let imports_data = SerializableImportsData { data };

    let expected_imports_data = fs::read_to_string("vendor/django/imports.json").unwrap();
    let expected_imports_data: SerializableImportsData =
        serde_json::from_str(&expected_imports_data).unwrap();

    assert_eq!(
        expected_imports_data.data.keys().collect::<HashSet<_>>(),
        imports_data.data.keys().collect::<HashSet<_>>()
    );
    for key in expected_imports_data.data.keys() {
        // println!("{}", key);
        assert_eq!(
            expected_imports_data
                .data
                .get(key)
                .unwrap()
                .iter()
                .sorted()
                .collect::<Vec<_>>(),
            imports_data
                .data
                .get(key)
                .unwrap()
                .iter()
                .sorted()
                .collect::<Vec<_>>()
        );
    }
}

fn is_hidden(entry: &DirEntry) -> bool {
    entry
        .file_name()
        .to_str()
        .map(|s| s.starts_with("."))
        .unwrap_or(false)
}

#[derive(Debug, Eq, PartialEq, Serialize, Deserialize)]
struct SerializableImportsData {
    data: HashMap<String, HashSet<SerializableImport>>,
}

#[derive(Debug, Eq, PartialEq, Hash, Ord, PartialOrd, Serialize, Deserialize)]
struct SerializableImport {
    imported_object: String,
    typechecking_only: bool,
}