Completed
Pull Request — master (#303)
by Björn
34:59 queued 09:58
created

Nvim.subscribe()   A

Complexity

Conditions 1

Size

Total Lines 3

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 2
CRAP Score 1

Importance

Changes 0
Metric Value
cc 1
dl 0
loc 3
ccs 2
cts 2
cp 1
crap 1
rs 10
c 0
b 0
f 0
1
"""Main Nvim interface."""
2 5
import functools
3 5
import os
4 5
import sys
5
from traceback import format_stack
6 5
7
from msgpack import ExtType
0 ignored issues
show
Configuration introduced by
The import msgpack could not be resolved.

This can be caused by one of the following:

1. Missing Dependencies

This error could indicate a configuration issue of Pylint. Make sure that your libraries are available by adding the necessary commands.

# .scrutinizer.yml
before_commands:
    - sudo pip install abc # Python2
    - sudo pip3 install abc # Python3
Tip: We are currently not using virtualenv to run pylint, when installing your modules make sure to use the command for the correct version.

2. Missing __init__.py files

This error could also result from missing __init__.py files in your module folders. Make sure that you place one file in each sub-folder.

Loading history...
8 5
9
from .buffer import Buffer
10 5
from .common import (Remote, RemoteApi, RemoteMap, RemoteSequence,
11 5
                     decode_if_bytes, walk)
12
from .tabpage import Tabpage
13 5
from .window import Window
14 5
from ..compat import IS_PYTHON3
15 5
from ..util import Version, format_exc_skip
16 5
17
__all__ = ('Nvim')
18 5
19
20
os_chdir = os.chdir
21 5
22
23
class Nvim(object):
24 5
25
    """Class that represents a remote Nvim instance.
26
27
    This class is main entry point to Nvim remote API, it is a wrapper
28
    around Session instances.
29
30
    The constructor of this class must not be called directly. Instead, the
31
    `from_session` class method should be used to create the first instance
32
    from a raw `Session` instance.
33
34
    Subsequent instances for the same session can be created by calling the
35
    `with_decode` instance method to change the decoding behavior or
36
    `SubClass.from_nvim(nvim)` where `SubClass` is a subclass of `Nvim`, which
37
    is useful for having multiple `Nvim` objects that behave differently
38
    without one affecting the other.
39
    """
40
41
    @classmethod
42 5
    def from_session(cls, session):
43
        """Create a new Nvim instance for a Session instance.
44
45
        This method must be called to create the first Nvim instance, since it
46
        queries Nvim metadata for type information and sets a SessionHook for
47
        creating specialized objects from Nvim remote handles.
48
        """
49
        session.error_wrapper = lambda e: NvimError(e[1])
50 5
        channel_id, metadata = session.request(b'vim_get_api_info')
51 5
52
        if IS_PYTHON3:
53 5
            # decode all metadata strings for python3
54
            metadata = walk(decode_if_bytes, metadata)
55 5
56
        types = {
57 5
            metadata['types']['Buffer']['id']: Buffer,
58
            metadata['types']['Window']['id']: Window,
59
            metadata['types']['Tabpage']['id']: Tabpage,
60
        }
61
62
        return cls(session, channel_id, metadata, types)
63 5
64
    @classmethod
65 5
    def from_nvim(cls, nvim):
66
        """Create a new Nvim instance from an existing instance."""
67
        return cls(nvim._session, nvim.channel_id, nvim.metadata,
0 ignored issues
show
Coding Style Best Practice introduced by
It seems like _session was declared protected and should not be accessed from this context.

Prefixing a member variable _ is usually regarded as the equivalent of declaring it with protected visibility that exists in other languages. Consequentially, such a member should only be accessed from the same class or a child class:

class MyParent:
    def __init__(self):
        self._x = 1;
        self.y = 2;

class MyChild(MyParent):
    def some_method(self):
        return self._x    # Ok, since accessed from a child class

class AnotherClass:
    def some_method(self, instance_of_my_child):
        return instance_of_my_child._x   # Would be flagged as AnotherClass is not
                                         # a child class of MyParent
Loading history...
68
                   nvim.types, nvim._decode, nvim._err_cb)
0 ignored issues
show
Coding Style Best Practice introduced by
It seems like _decode was declared protected and should not be accessed from this context.

Prefixing a member variable _ is usually regarded as the equivalent of declaring it with protected visibility that exists in other languages. Consequentially, such a member should only be accessed from the same class or a child class:

class MyParent:
    def __init__(self):
        self._x = 1;
        self.y = 2;

class MyChild(MyParent):
    def some_method(self):
        return self._x    # Ok, since accessed from a child class

class AnotherClass:
    def some_method(self, instance_of_my_child):
        return instance_of_my_child._x   # Would be flagged as AnotherClass is not
                                         # a child class of MyParent
Loading history...
Coding Style Best Practice introduced by
It seems like _err_cb was declared protected and should not be accessed from this context.

Prefixing a member variable _ is usually regarded as the equivalent of declaring it with protected visibility that exists in other languages. Consequentially, such a member should only be accessed from the same class or a child class:

class MyParent:
    def __init__(self):
        self._x = 1;
        self.y = 2;

class MyChild(MyParent):
    def some_method(self):
        return self._x    # Ok, since accessed from a child class

class AnotherClass:
    def some_method(self, instance_of_my_child):
        return instance_of_my_child._x   # Would be flagged as AnotherClass is not
                                         # a child class of MyParent
Loading history...
69
70
    def __init__(self, session, channel_id, metadata, types,
71 5
                 decode=False, err_cb=None):
72
        """Initialize a new Nvim instance. This method is module-private."""
73
        self._session = session
74 5
        self.channel_id = channel_id
75 5
        self.metadata = metadata
76 5
        version = metadata.get("version", {"api_level": 0})
77 5
        self.version = Version(**version)
78 5
        self.types = types
79 5
        self.api = RemoteApi(self, 'nvim_')
80 5
        self.vars = RemoteMap(self, 'nvim_get_var', 'nvim_set_var')
81 5
        self.vvars = RemoteMap(self, 'nvim_get_vvar', None)
82 5
        self.options = RemoteMap(self, 'nvim_get_option', 'nvim_set_option')
83 5
        self.buffers = Buffers(self)
84 5
        self.windows = RemoteSequence(self, 'nvim_list_wins')
85 5
        self.tabpages = RemoteSequence(self, 'nvim_list_tabpages')
86 5
        self.current = Current(self)
87 5
        self.session = CompatibilitySession(self)
88 5
        self.funcs = Funcs(self)
89 5
        self.error = NvimError
90 5
        self._decode = decode
91 5
        self._err_cb = err_cb
92 5
93
    def _from_nvim(self, obj, decode=None):
94 5
        if decode is None:
95 5
            decode = self._decode
96 5
        if type(obj) is ExtType:
97 5
            cls = self.types[obj.code]
98 5
            return cls(self, (obj.code, obj.data))
99 5
        if decode:
100 5
            obj = decode_if_bytes(obj, decode)
101 3
        return obj
102 5
103
    def _to_nvim(self, obj):
0 ignored issues
show
Coding Style introduced by
This method could be written as a function/class method.

If a method does not access any attributes of the class, it could also be implemented as a function or static method. This can help improve readability. For example

class Foo:
    def some_method(self, x, y):
        return x + y;

could be written as

class Foo:
    @classmethod
    def some_method(cls, x, y):
        return x + y;
Loading history...
104 5
        if isinstance(obj, Remote):
105 5
            return ExtType(*obj.code_data)
106 5
        return obj
107 5
108
    def request(self, name, *args, **kwargs):
109 5
        r"""Send an API request or notification to nvim.
110
111
        It is rarely needed to call this function directly, as most API
112
        functions have python wrapper functions. The `api` object can
113
        be also be used to call API functions as methods:
114
115
            vim.api.err_write('ERROR\n', async=True)
116
            vim.current.buffer.api.get_mark('.')
117
118
        is equivalent to
119
120
            vim.request('nvim_err_write', 'ERROR\n', async=True)
121
            vim.request('nvim_buf_get_mark', vim.current.buffer, '.')
122
123
124
        Normally a blocking request will be sent.  If the `async` flag is
125
        present and True, a asynchronous notification is sent instead. This
126
        will never block, and the return value or error is ignored.
127
        """
128
        decode = kwargs.pop('decode', self._decode)
129 5
        args = walk(self._to_nvim, args)
130 5
        res = self._session.request(name, *args, **kwargs)
131 5
        return walk(self._from_nvim, res, decode=decode)
132 5
133
    def next_message(self):
134 5
        """Block until a message(request or notification) is available.
135
136
        If any messages were previously enqueued, return the first in queue.
137
        If not, run the event loop until one is received.
138
        """
139
        msg = self._session.next_message()
140 5
        if msg:
141 5
            return walk(self._from_nvim, msg)
142 5
143
    def run_loop(self, request_cb, notification_cb,
144 5
                 setup_cb=None, err_cb=None):
145
        """Run the event loop to receive requests and notifications from Nvim.
146
147
        This should not be called from a plugin running in the host, which
148
        already runs the loop and dispatches events to plugins.
149
        """
150
        if err_cb is None:
151 5
            err_cb = sys.stderr.write
152 5
        self._err_cb = err_cb
153 5
154
        def filter_request_cb(name, args):
155 5
            name = self._from_nvim(name)
156 5
            args = walk(self._from_nvim, args)
157 5
            try:
158 5
                result = request_cb(name, args)
159 5
            except Exception:
160
                msg = ("error caught in request handler '{} {}'\n{}\n\n"
161
                       .format(name, args, format_exc_skip(1)))
162
                self._err_cb(msg)
163
                raise
164
            return walk(self._to_nvim, result)
165 5
166
        def filter_notification_cb(name, args):
167 5
            name = self._from_nvim(name)
168
            args = walk(self._from_nvim, args)
169
            try:
170
                notification_cb(name, args)
171
            except Exception:
172
                msg = ("error caught in notification handler '{} {}'\n{}\n\n"
173
                       .format(name, args, format_exc_skip(1)))
174
                self._err_cb(msg)
175
                raise
176
177
        self._session.run(filter_request_cb, filter_notification_cb, setup_cb)
178 5
179
    def stop_loop(self):
180 5
        """Stop the event loop being started with `run_loop`."""
181
        self._session.stop()
182 5
183
    def close_loop(self):
184 5
        """Close the event loop being started with `run_loop`."""
185
        self._session.close()
186 5
187
188
    def with_decode(self, decode=True):
189 5
        """Initialize a new Nvim instance."""
190
        return Nvim(self._session, self.channel_id,
191
                    self.metadata, self.types, decode, self._err_cb)
192
193
    def ui_attach(self, width, height, rgb):
194
        """Register as a remote UI.
195
196
        After this method is called, the client will receive redraw
197 5
        notifications.
198
        """
199
        return self.request('ui_attach', width, height, rgb)
200
201 5
    def ui_detach(self):
202
        """Unregister as a remote UI."""
203
        return self.request('ui_detach')
204
205
    def ui_try_resize(self, width, height):
206
        """Notify nvim that the client window has resized.
207
208 5
        If possible, nvim will send a redraw request to resize.
209
        """
210 5
        return self.request('ui_try_resize', width, height)
211
212 5
    def subscribe(self, event):
213
        """Subscribe to a Nvim event."""
214 5
        return self.request('nvim_subscribe', event)
215
216 5
    def unsubscribe(self, event):
217
        """Unsubscribe to a Nvim event."""
218 5
        return self.request('nvim_unsubscribe', event)
219
220 5
    def command(self, string, **kwargs):
221
        """Execute a single ex command."""
222
        return self.request('nvim_command', string, **kwargs)
223
224 5
    def command_output(self, string):
225
        """Execute a single ex command and return the output."""
226 5
        return self.request('nvim_command_output', string)
227
228 5
    def eval(self, string, **kwargs):
229
        """Evaluate a vimscript expression."""
230 5
        return self.request('nvim_eval', string, **kwargs)
231
232 5
    def call(self, name, *args, **kwargs):
233
        """Call a vimscript function."""
234
        return self.request('nvim_call_function', name, args, **kwargs)
235
236
    def strwidth(self, string):
237 5
        """Return the number of display cells `string` occupies.
238
239 5
        Tab is counted as one cell.
240
        """
241
        return self.request('nvim_strwidth', string)
242
243 5
    def list_runtime_paths(self):
244
        """Return a list of paths contained in the 'runtimepath' option."""
245
        return self.request('nvim_list_runtime_paths')
246
247
    def foreach_rtp(self, cb):
248
        """Invoke `cb` for each path in 'runtimepath'.
249
250
        Call the given callable for each path in 'runtimepath' until either
251
        callable returns something but None, the exception is raised or there
252
        are no longer paths. If stopped in case callable returned non-None,
253
        vim.foreach_rtp function returns the value returned by callable.
254
        """
255
        for path in self.request('nvim_list_runtime_paths'):
256
            try:
257
                if cb(path) is not None:
258 5
                    break
259
            except Exception:
0 ignored issues
show
Best Practice introduced by
Catching very general exceptions such as Exception is usually not recommended.

Generally, you would want to handle very specific errors in the exception handler. This ensure that you do not hide other types of errors which should be fixed.

So, unless you specifically plan to handle any error, consider adding a more specific exception.

Loading history...
260 5
                break
261 5
262
    def chdir(self, dir_path):
263 5
        """Run os.chdir, then all appropriate vim stuff."""
264
        os_chdir(dir_path)
265
        return self.request('nvim_set_current_dir', dir_path)
266
267
    def feedkeys(self, keys, options='', escape_csi=True):
268
        """Push `keys` to Nvim user input buffer.
269
270
        Options can be a string with the following character flags:
271
        - 'm': Remap keys. This is default.
272
        - 'n': Do not remap keys.
273
        - 't': Handle keys as if typed; otherwise they are handled as if coming
274 5
               from a mapping. This matters for undo, opening folds, etc.
275
        """
276
        return self.request('nvim_feedkeys', keys, options, escape_csi)
277
278
    def input(self, bytes):
0 ignored issues
show
Bug Best Practice introduced by
This seems to re-define the built-in bytes.

It is generally discouraged to redefine built-ins as this makes code very hard to read.

Loading history...
279
        """Push `bytes` to Nvim low level input buffer.
280
281
        Unlike `feedkeys()`, this uses the lowest level input buffer and the
282 5
        call is not deferred. It returns the number of bytes actually
283
        written(which can be less than what was requested if the buffer is
284 5
        full).
285
        """
286
        return self.request('nvim_input', bytes)
287
288
    def replace_termcodes(self, string, from_part=False, do_lt=True,
289
                          special=True):
290
        r"""Replace any terminal code strings by byte sequences.
291
292
        The returned sequences are Nvim's internal representation of keys,
293
        for example:
294
295
        <esc> -> '\x1b'
296
        <cr>  -> '\r'
297
        <c-l> -> '\x0c'
298
        <up>  -> '\x80ku'
299
300
        The returned sequences can be used as input to `feedkeys`.
301 5
        """
302
        return self.request('nvim_replace_termcodes', string,
303
                            from_part, do_lt, special)
304
305 5
    def out_write(self, msg, **kwargs):
306
        """Print `msg` as a normal message."""
307 2
        return self.request('nvim_out_write', msg, **kwargs)
308
309 5
    def err_write(self, msg, **kwargs):
310
        """Print `msg` as an error message."""
311
        return self.request('nvim_err_write', msg, **kwargs)
312
313
    def quit(self, quit_command='qa!'):
314
        """Send a quit command to Nvim.
315
316
        By default, the quit command is 'qa!' which will make Nvim quit without
317
        saving anything.
318
        """
319
        try:
320
            self.command(quit_command)
321
        except IOError:
0 ignored issues
show
Unused Code introduced by
This except handler seems to be unused and could be removed.

Except handlers which only contain pass and do not have an else clause can usually simply be removed:

try:
    raises_exception()
except:  # Could be removed
    pass
Loading history...
322
            # sending a quit command will raise an IOError because the
323 5
            # connection is closed before a response is received. Safe to
324
            # ignore it.
325
            pass
326
327 5
    def new_highlight_source(self):
328
        """Return new src_id for use with Buffer.add_highlight."""
329
        return self.current.buffer.add_highlight("", 0, src_id=0)
330
331
    def async_call(self, fn, *args, **kwargs):
332
        """Schedule `fn` to be called by the event loop soon.
333
334
        This function is thread-safe, and is the only way code not
335
        on the main thread could interact with nvim api objects.
336
337 5
        This function can also be called in a synchronous
338
        event handler, just before it returns, to defer execution
339 5
        that shouldn't block neovim.
340 5
        """
341 5
        call_point = ''.join(format_stack(None, 5)[:-1])
342
343 1
        def handler():
344
            try:
345
                fn(*args, **kwargs)
346 1
            except Exception as err:
347
                msg = ("error caught while executing async callback:\n"
348 5
                       "{!r}\n{}\n \nthe call was requested at\n{}"
349
                       .format(err, format_exc_skip(1), call_point))
350
                self._err_cb(msg)
351 5
                raise
352
        self._session.threadsafe_call(handler)
353
354
355
class Buffers(object):
356
357
    """Remote NVim buffers.
358
359
    Currently the interface for interacting with remote NVim buffers is the
360
    `nvim_list_bufs` msgpack-rpc function. Most methods fetch the list of
361
    buffers from NVim.
362 5
363
    Conforms to *python-buffers*.
364 5
    """
365
366 5
    def __init__(self, nvim):
367
        """Initialize a Buffers object with Nvim object `nvim`."""
368 5
        self._fetch_buffers = nvim.api.list_bufs
369
370 5
    def __len__(self):
371
        """Return the count of buffers."""
372 5
        return len(self._fetch_buffers())
373 5
374 5
    def __getitem__(self, number):
375
        """Return the Buffer object matching buffer number `number`."""
376
        for b in self._fetch_buffers():
377 5
            if b.number == number:
378
                return b
379 5
        raise KeyError(number)
380
381 5
    def __contains__(self, b):
382
        """Return whether Buffer `b` is a known valid buffer."""
383 5
        return isinstance(b, Buffer) and b.valid
384
385
    def __iter__(self):
386 5
        """Return an iterator over the list of buffers."""
387
        return iter(self._fetch_buffers())
388
389
390 5
class CompatibilitySession(object):
391 5
392
    """Helper class for API compatibility."""
393
394 5
    def __init__(self, nvim):
395
        self.threadsafe_call = nvim.async_call
396
397
398 5
class Current(object):
399 5
400 5
    """Helper class for emulating vim.current from python-vim."""
401
402 5
    def __init__(self, session):
403
        self._session = session
404 5
        self.range = None
405
406 5
    @property
407
    def line(self):
408 5
        return self._session.request('nvim_get_current_line')
409
410 5
    @line.setter
411
    def line(self, line):
412 5
        return self._session.request('nvim_set_current_line', line)
413
414 5
    @property
415
    def buffer(self):
416 5
        return self._session.request('nvim_get_current_buf')
417
418 5
    @buffer.setter
419
    def buffer(self, buffer):
420 5
        return self._session.request('nvim_set_current_buf', buffer)
421
422 5
    @property
423
    def window(self):
424 5
        return self._session.request('nvim_get_current_win')
425
426 5
    @window.setter
427
    def window(self, window):
428 5
        return self._session.request('nvim_set_current_win', window)
429
430 5
    @property
431
    def tabpage(self):
432 5
        return self._session.request('nvim_get_current_tabpage')
433
434
    @tabpage.setter
435 5
    def tabpage(self, tabpage):
436
        return self._session.request('nvim_set_current_tabpage', tabpage)
437
438
439 5
class Funcs(object):
440 5
441
    """Helper class for functional vimscript interface."""
442 5
443 5
    def __init__(self, nvim):
444
        self._nvim = nvim
445
446 5
    def __getattr__(self, name):
447 5
        return functools.partial(self._nvim.call, name)
448
449
450
class NvimError(Exception):
451
    pass
452