aboutsummaryrefslogtreecommitdiffstats
path: root/queries/python/textobjects.scm
diff options
context:
space:
mode:
Diffstat (limited to 'queries/python/textobjects.scm')
-rw-r--r--queries/python/textobjects.scm12
1 files changed, 10 insertions, 2 deletions
diff --git a/queries/python/textobjects.scm b/queries/python/textobjects.scm
index 13b1921ed..9f0d41233 100644
--- a/queries/python/textobjects.scm
+++ b/queries/python/textobjects.scm
@@ -2,9 +2,15 @@
(function_definition
body: (block)? @function.inner) @function.outer
+(decorated_definition
+ (function_definition) @function.outer) @function.outer.start
+
(class_definition
body: (block)? @class.inner) @class.outer
+(decorated_definition
+ (class_definition) @class.outer) @class.outer.start
+
(while_statement
body: (block)? @loop.inner) @loop.outer
@@ -12,8 +18,10 @@
body: (block)? @loop.inner) @loop.outer
(if_statement
- consequence: (block)? @conditional.inner
- alternative: (_ (block) @conditional.inner)?) @conditional.outer
+ alternative: (_ (_) @conditional.inner)?) @conditional.outer
+
+(if_statement
+ consequence: (block)? @conditional.inner)
(if_statement
condition: (_) @conditional.inner)