Function extract_schemas

plugins/ast/python/src/lib.rs:360–487

(body: &[Statement])

// ─── schemas ───

fn extract_schemas(body: &[Statement]) -> Vec<Model> {
    let mut schemas = Vec::new();

    for statement in body {
        let Statement::ClassDef(ClassDef {
            name,
            body,
            arguments,
            ..
        }) = statement
        else {
            continue;
        };

        // Heuristic: a model has at least one base class (Base/BaseModel/models.Model/...).
        let bases: Vec<String> = arguments
            .as_deref()
            .map(|args| args.args.iter().filter_map(dotted).collect())
            .unwrap_or_default();

        if bases.is_empty() {
            continue;
        }

        let mut fields = Vec::new();
        let mut relations = Vec::new();
        let mut orm: Option<&'static str> = None;

        for member in body {
            match member {
                // Annotated: `id: Mapped[int] = mapped_column(primary_key=True)`,
                // `name: str` (pydantic), `email: str = Field(unique=True)` (SQLModel).
                Statement::AnnAssign(AnnotationAssignment {
                    target,
                    annotation,
                    value,
                    ..
                }) => {
                    let Expression::Name(NameExpression { id, .. }) = &**target else {
                        continue;
                    };

                    let field_name = id.to_string();

                    if field_name.starts_with("__") {
                        continue;
                    }

                    let (mut r#type, nullable) = annotation_type(annotation);
                    let mut flags = Vec::new();

                    if let Some(call) = value.as_deref().and_then(analyze_call) {
                        orm = orm.or(call.orm);

                        if let Some(target) = call.relation {
                            relations.push(target);
                            continue;
                        }