feat(python): Scope type hints

README.md

@@ -1434,6 +1434,7 @@ Language scopes:
           - globals:              Global, i.e. module-level variables
           - variable-identifiers: Identifiers for variables (left-hand side of
             assignments)
+          - types:                Types in type hints
 
       --python-query <PYTHON_QUERY>
           Scope Python code using a custom tree-sitter query.

src/scoping/langs/python.rs

@@ -47,6 +47,8 @@ pub enum PreparedPythonQuery {
     Globals,
     /// Identifiers for variables (left-hand side of assignments).
     VariableIdentifiers,
+    /// Types in type hints.
+    Types,
 }
 
 impl From<PreparedPythonQuery> for TSQuery {
@@ -162,6 +164,7 @@ impl From<PreparedPythonQuery> for TSQuery {
                 PreparedPythonQuery::VariableIdentifiers => {
                     "(assignment left: (identifier) @identifier)"
                 }
+                PreparedPythonQuery::Types => "(type) @type",
             },
         )
         .expect("Prepared queries to be valid")

tests/langs/mod.rs

@@ -157,6 +157,11 @@ impl InScopeLinePart {
     include_str!("python/base.py"),
     Python::new(CodeQuery::Prepared(PreparedPythonQuery::VariableIdentifiers)),
 )]
+#[case(
+    "base.py_types",
+    include_str!("python/base.py"),
+    Python::new(CodeQuery::Prepared(PreparedPythonQuery::Types)),
+)]
 #[case(
     "base.ts_strings",
     include_str!("typescript/base.ts"),

tests/langs/snapshots/r#mod__langs__base.py_types.snap

@@ -0,0 +1,19 @@
+---
+source: tests/langs/mod.rs
+expression: inscope_parts
+---
+- n: 16
+  l: "test_var: int = 10\n"
+  m: "          ^^^       "
+- n: 63
+  l: "    def class_method(cls) -> None:\n"
+  m: "                             ^^^^   "
+- n: 69
+  l: "    def instance_method(self) -> None:\n"
+  m: "                                 ^^^^   "
+- n: 76
+  l: "    def static_method() -> None:\n"
+  m: "                           ^^^^   "
+- n: 112
+  l: "def exception_handling(x) -> None:\n"
+  m: "                             ^^^^   "