How to find comments on same line as node? (python)

[cognifloyd]

I'm working on some rust code that uses tree-sitter to parse python code (with https://github.com/tree-sitter/tree-sitter-python).

Are there any convenience methods to facilitate searching for comments on a given row?

I have a node that represents a python string. I need to find a comment on the same line to see if the user has asked my tool to ignore that string. Looking at the playground, it seems that comment nodes are a sibling of some arbitrary parent node of unknown depth.

Here is the rust code I'm working on:
https://github.com/pantsbuild/pants/blob/3632a66bc7918ace25c0aa6f05b4b6a4253d908e/src/rust/engine/dep_inference/src/python/mod.rs#L322-L330

impl Visitor for ImportCollector<'_> {
    // ... snip other functions
    fn visit_string(&mut self, node: tree_sitter::Node) -> ChildBehavior {
        let range = node.range();
        let text: &str = self.string_at(range);
        // do something with the text here if the node does not have the comment.
        ChildBehavior::Ignore
    }

This function is used for checking for the pragma comment in several other places, but it requires me to pass in the node whose sibling could be the comment node. That's problematic because I don't know how many parents I'll have to walk to find the comment node.
https://github.com/pantsbuild/pants/blob/3632a66bc7918ace25c0aa6f05b4b6a4253d908e/src/rust/engine/dep_inference/src/python/mod.rs#L110-L123

impl ImportCollector<'_> {
    // ... snip other functions
    fn is_pragma_ignored(&self, node: tree_sitter::Node) -> bool {
        if let Some(sibling) = node.next_named_sibling() {
            let next_node_range = sibling.range();
            if sibling.kind_id() == KindID::COMMENT
                && node.range().end_point.row == next_node_range.start_point.row
                && self
                    .code_at(next_node_range)
                    .contains("# pants: no-infer-dep")
            {
                return true;
            }
        }
        false
    }

Any suggestions?

[cognifloyd]

This is what I came up with: pantsbuild/pants@36f4b3a

Is there a better way to do this? Some utility in tree-stitter, accessible from rust, that facilitates a search like this?