Fix documentation class and function text objects don't work in normal mode

lieryan · lieryan · commit c47e6dbc55ab · 2021-10-01T10:17:48.000+10:00
diff --git a/doc/pymode.txt b/doc/pymode.txt
@@ -209,10 +209,10 @@ Key   Command
 ]]    Jump to next class or function  (normal, visual, operator modes)
 [M    Jump to previous class or method (normal, visual, operator modes)
 ]M    Jump to next class or method (normal, visual, operator modes)
-aC    Select a class. Ex: vaC, daC, yaC, caC (normal, operator modes)
-iC    Select inner class. Ex: viC, diC, yiC, ciC (normal, operator modes)
-aM    Select a function or method. Ex: vaM, daM, yaM, caM (normal, operator modes)
-iM    Select inner function or method. Ex: viM, diM, yiM, ciM (normal, operator modes)
+aC    Select a class. Ex: vaC, daC, yaC, caC (operator modes)
+iC    Select inner class. Ex: viC, diC, yiC, ciC (operator modes)
+aM    Select a function or method. Ex: vaM, daM, yaM, caM (operator modes)
+iM    Select inner function or method. Ex: viM, diM, yiM, ciM (operator modes)
 ====  ============================
 
 Enable pymode-motion                                          *'g:pymode_motion'*
diff --git a/pymode/rope.py b/pymode/rope.py
@@ -924,13 +924,17 @@ def _insert_import(name, module, ctx):
 @env.catch_exceptions
 def select_logical_line():
     source, offset = env.get_offset_params()
+    count = int(env.var('v:count1'))
 
     lines = codeanalyze.SourceLinesAdapter(source)
-    lineno = lines.get_line_number(offset)
+    start_line = lines.get_line_number(offset)
     line_finder = codeanalyze.LogicalLineFinder(lines)
-    start, end = line_finder.logical_line_in(lineno)
 
-    env.select_line(start, end)
+    start_lineno, _ = line_finder.logical_line_in(start_line)
+    for _, (_, end_lineno) in zip(range(count), line_finder.generate_regions(start_line)):
+        pass
+
+    env.select_line(start_lineno, end_lineno)
 
 
 # Monkey patch Rope
