Function analyze_call

plugins/ast/python/src/lib.rs:499–561

Inspect a field's right-hand-side call for ORM hints, flags, relations, and type.

(expression: &Expression)

/// Inspect a field's right-hand-side call for ORM hints, flags, relations, and type.
fn analyze_call(expression: &Expression) -> Option<CallInfo> {
    let Expression::Call(CallExpression {
        func, arguments, ..
    }) = expression
    else {
        return None;
    };

    let leaf = call_leaf(func)?;
    let module = call_module(func);
    let mut info = CallInfo::default();

    if module.as_deref() == Some("models") {
        info.orm = Some("django");
    } else if matches!(leaf.as_str(), "Column" | "mapped_column" | "relationship") {
        info.orm = Some("sqlalchemy");
    }

    // Relationship / foreign-key constructors yield a relation, not a scalar column.
    const RELATIONS: &[&str] = &[
        "relationship",
        "ForeignKey",
        "ManyToManyField",
        "OneToOneField",
    ];

    if RELATIONS.contains(&leaf.as_str()) {
        info.relation = arguments.args.first().and_then(relation_target);

        if leaf != "relationship" {
            info.orm = Some("django");
        }

        return Some(info);
    }

    // Flags from truthy boolean keywords (primary_key/unique/nullable/null).
    for keyword in &arguments.keywords {
        let Some(arg) = keyword.arg.as_ref().map(Identifier::as_str) else {
            continue;
        };

        if !matches!(&keyword.value, Expression::BooleanLiteral(literal) if literal.value) {
            continue;
        }

        match arg {
            "primary_key" => info.flags.push("pk".to_string()),
            "unique" => info.flags.push("unique".to_string()),
            "nullable" | "null" => info.flags.push("nullable".to_string()),
            _ => {}
        }
    }

    // Column type: Django field leaf, or SQLAlchemy `Column(Type, ...)` first positional.
    if module.as_deref() == Some("models") {
        info.ctor_type = Some(django_type(&leaf));
    } else if leaf == "Column" || leaf == "mapped_column" {