alef 0.115.7

Opinionated polyglot binding generator for Rust libraries
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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105

use crate::backends::pyo3::type_map::python_type;
use crate::codegen::error_gen::{
    ExceptionNaming, exception_families, pyo3_error_attribute_fields, python_exception_name,
};
use crate::codegen::naming::public_field_name;
use crate::core::config::Language;
use crate::core::ir::{ApiSurface, ErrorVariant};

pub(super) fn gen_error_stubs(api: &ApiSurface) -> anyhow::Result<Vec<String>> {
    let mut seen = std::collections::HashSet::new();
    let mut lines = Vec::new();
    for family in exception_families(api, ExceptionNaming::Python)? {
        let error = &family.error;
        if seen.insert(error.name.clone()) {
            lines.push(render_exception(
                &error.name,
                family.parent.as_deref().unwrap_or("Exception"),
                None,
            ));
        }
        for variant in &error.variants {
            let name = python_exception_name(&variant.name, &error.name);
            if seen.insert(name.clone()) {
                lines.push(render_exception(&name, &error.name, Some(variant)));
            }
        }
    }
    if !lines.is_empty() {
        lines.push(String::new());
    }
    Ok(lines)
}

fn render_exception(name: &str, base: &str, variant: Option<&ErrorVariant>) -> String {
    let fields = variant.map(pyo3_error_attribute_fields).unwrap_or_default();
    let fields = fields
        .iter()
        .map(|field| {
            let mut field_type = python_type(&field.ty);
            if field.optional && !field_type.contains("| None") {
                field_type.push_str(" | None");
            }
            crate::alef_context! {
                name => public_field_name(Language::Python, &field.name, None),
                python_type => field_type,
            }
        })
        .collect::<Vec<_>>();
    crate::backends::pyo3::template_env::render(
        "exception_variant_stub.jinja",
        crate::alef_context! { name => name, base => base, fields => fields },
    )
    .trim_end()
    .to_string()
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::core::ir::TypeRef;

    #[test]
    fn should_declare_named_error_attributes_in_python_stubs() {
        use crate::core::ir::{ErrorDef, ErrorVariant, FieldDef};
        let api = ApiSurface {
            errors: vec![ErrorDef {
                name: "RequestError".to_string(),
                variants: vec![ErrorVariant {
                    name: "Rejected".to_string(),
                    fields: vec![
                        FieldDef {
                            name: "status".to_string(),
                            ty: TypeRef::Primitive(crate::core::ir::PrimitiveType::U16),
                            optional: true,
                            ..Default::default()
                        },
                        FieldDef {
                            name: "body".to_string(),
                            ty: TypeRef::String,
                            core_wrapper: crate::core::ir::CoreWrapper::StaticStr,
                            optional: true,
                            ..Default::default()
                        },
                    ],
                    ..Default::default()
                }],
                exception_base: None,
                rust_path: "sample::RequestError".to_string(),
                original_rust_path: String::new(),
                doc: String::new(),
                methods: vec![],
                binding_excluded: false,
                binding_exclusion_reason: None,
                version: Default::default(),
            }],
            ..Default::default()
        };
        let stub = gen_error_stubs(&api).expect("valid stub").join("\n");
        assert!(
            stub.contains("class RejectedError(RequestError):\n    status: int | None"),
            "{stub}"
        );
        assert!(stub.contains("    body: str | None"), "{stub}");
    }
}