Implemented vim.current.range

- Add arguments range_start and range_end to methods python_execute and
  python_execute_file
- Add test for python_execute with range

Author: henadzit

neovim/api/nvim.py

@@ -199,6 +199,7 @@ class Current(object):
 
     def __init__(self, session):
         self._session = session
+        self.range = None
 
     @property
     def line(self):

neovim/plugins/script_host.py

@@ -28,15 +28,22 @@ def __init__(self, nvim):
         sys.modules['vim'] = nvim.with_hook(LegacyEvalHook())
         if IS_PYTHON3:
             sys.modules['vim'] = sys.modules['vim'].with_hook(DecodeHook(encoding=nvim.options['encoding'].decode('ascii')))
+        self.legacy_vim = sys.modules['vim']
 
-    def python_execute(self, script):
+    def python_execute(self, script, range_start, range_end):
+        self._set_current_range(range_start, range_end)
         exec(script, self.module.__dict__)
 
-    def python_execute_file(self, file_path):
+    def python_execute_file(self, file_path, range_start, range_end):
+        self._set_current_range(range_start, range_end)
         with open(file_path) as f:
             script = compile(f.read(), file_path, 'exec')
             exec(script, self.module.__dict__)
 
+    def _set_current_range(self, range_start, range_end):
+        self.legacy_vim.current.range = \
+                self.legacy_vim.current.buffer.range(range_start, range_end)
+
     def python_do_range(self, start, stop, code):
         nvim = self.nvim
         start -= 1

test/test_script_host.py

@@ -44,6 +44,14 @@ def test_python_nested_commands():
     eq(vim.vars['set_by_nested_python'], 555)
 
 
+@with_setup(setup=host_setup, teardown=host_teardown)
+def test_python_command_with_range():
+    vim.feedkeys('iline1\nline2\nline3\nline4\033')
+    vim.feedkeys('ggjvj:python vim.vars["range"] = vim.current.range[:]\n')
+    vim.eval('1') # wait for the keys to be processed
+    eq(vim.vars['range'], ['line2', 'line3'])
+
+
 @with_setup(setup=host_setup, teardown=host_teardown)
 def test_pyfile():
     fname = 'pyfile.py'