aboutsummaryrefslogtreecommitdiffstats
path: root/queries
diff options
context:
space:
mode:
Diffstat (limited to 'queries')
-rw-r--r--queries/javascript/locals.scm2
-rw-r--r--queries/python/locals.scm8
2 files changed, 9 insertions, 1 deletions
diff --git a/queries/javascript/locals.scm b/queries/javascript/locals.scm
index d56000d5a..9a10db86d 100644
--- a/queries/javascript/locals.scm
+++ b/queries/javascript/locals.scm
@@ -29,7 +29,7 @@
name: (identifier) @definition)
(import_specifier
- (identifier) @definition)
+ (identifier) @definition.import)
; References
;------------
diff --git a/queries/python/locals.scm b/queries/python/locals.scm
index bcf1b1572..1deca18c9 100644
--- a/queries/python/locals.scm
+++ b/queries/python/locals.scm
@@ -8,6 +8,14 @@
left: (expression_list
(identifier) @definition.associated))))) @scope
+; Imports
+(aliased_import
+ alias: (identifier) @definition.import)
+(import_statement
+ name: (dotted_name ((identifier) @definition.import)))
+(import_from_statement
+ name: (dotted_name ((identifier) @definition.import)))
+
; Function with parameters, defines parameters
(parameters
(identifier) @definition.var)