Skip to main content

scope_engine/language/
python.rs

1use super::{LanguageAdapter, LanguageQueries};
2use tree_sitter::Language;
3
4pub struct PythonAdapter;
5
6impl PythonAdapter {
7    #[must_use]
8    pub const fn new() -> Self {
9        Self
10    }
11}
12
13impl Default for PythonAdapter {
14    fn default() -> Self {
15        Self::new()
16    }
17}
18
19impl LanguageAdapter for PythonAdapter {
20    fn language_name(&self) -> &'static str {
21        "python"
22    }
23
24    fn extensions(&self) -> &[&'static str] {
25        &["py"]
26    }
27
28    fn language(&self) -> Language {
29        tree_sitter_python::LANGUAGE.into()
30    }
31
32    fn queries(&self) -> LanguageQueries {
33        LanguageQueries {
34            definitions: r"
35                (function_definition name: (identifier) @name) @def
36                (class_definition name: (identifier) @name) @def
37                (decorated_definition (function_definition name: (identifier) @name)) @def
38                (decorated_definition (class_definition name: (identifier) @name)) @def
39            ",
40            references: r"
41                (call function: (identifier) @ref) @call
42                (call function: (attribute attribute: (identifier) @ref)) @call
43                (identifier) @ref
44            ",
45        }
46    }
47}