Method extract_function

atomic-semantic/src/parsers/python.rs:106–184

Extract a function or method definition.

(
        &self,
        node: &Node,
        source: &str,
        file_path: &str,
        in_class: Option<&str>,
    )

    /// Extract a function or method definition.
    fn extract_function(
        &self,
        node: &Node,
        source: &str,
        file_path: &str,
        in_class: Option<&str>,
    ) -> Option<Entity> {
        let name_node = node.child_by_field_name("name")?;
        let name = self.node_text(&name_node, source);

        // Skip dunder methods that are noise (__module__, __qualname__, etc.)
        // but keep __init__, __str__, __repr__, __eq__, etc.
        if name.starts_with("__") && name.ends_with("__") {
            let keep = [
                "__init__",
                "__new__",
                "__del__",
                "__str__",
                "__repr__",
                "__eq__",
                "__ne__",
                "__lt__",
                "__le__",
                "__gt__",
                "__ge__",
                "__hash__",
                "__bool__",
                "__len__",
                "__iter__",
                "__next__",
                "__contains__",
                "__getitem__",
                "__setitem__",
                "__delitem__",
                "__call__",
                "__enter__",
                "__exit__",
                "__aenter__",
                "__aexit__",
                "__await__",
                "__aiter__",
                "__anext__",
            ];
            if !keep.contains(&name.as_str()) {
                return None;
            }
        }

        let kind = if in_class.is_some() {
            EntityKind::Method
        } else {
            EntityKind::Function
        };

        let line = node.start_position().row as u32 + 1;
        let end_line = node.end_position().row as u32 + 1;

        // Build signature