Merge pull request #77 from equalsraf/tb-decode-hook-enc

Python3 fixes: functools.partial and encoding

Author: tarruda

neovim/plugin/decorators.py

@@ -1,8 +1,10 @@
 """Decorators used by python host plugin system."""
+
 import inspect
+import logging
 
+from ..compat import IS_PYTHON3
 
-import logging
 logger = logging.getLogger(__name__)
 debug, info, warn = (logger.debug, logger.info, logger.warn,)
 __all__ = ('plugin', 'rpc_export', 'command', 'autocmd', 'function',
@@ -20,8 +22,11 @@ def plugin(cls):
     # decorated functions have a bound Nvim instance as first argument.
     # For methods in a plugin-decorated class this is not required, because
     # the class initializer will already receive the nvim object.
-    for _, fn in inspect.getmembers(cls, inspect.ismethod):
-        if hasattr(fn, '_nvim_bind'):
+    predicate = lambda fn: hasattr(fn, '_nvim_bind')
+    for _, fn in inspect.getmembers(cls, predicate):
+        if IS_PYTHON3:
+            fn._nvim_bind = False
+        else:
             fn.im_func._nvim_bind = False
     return cls
 

neovim/plugin/host.py

@@ -37,6 +37,8 @@ def __init__(self, nvim):
             'shutdown': self.shutdown
         }
         self._nvim_encoding = nvim.options['encoding']
+        if IS_PYTHON3 and isinstance(self._nvim_encoding, bytes):
+            self._nvim_encoding = self._nvim_encoding.decode('ascii')
 
     def start(self, plugins):
         """Start listening for msgpack-rpc requests and notifications."""
@@ -51,6 +53,8 @@ def shutdown(self):
 
     def _on_request(self, name, args):
         """Handle a msgpack-rpc request."""
+        if IS_PYTHON3 and isinstance(name, bytes):
+            name = name.decode(self._nvim_encoding)
         handler = self._request_handlers.get(name, None)
         if not handler:
             msg = 'no request handler registered for "%s"' % name
@@ -64,6 +68,8 @@ def _on_request(self, name, args):
 
     def _on_notification(self, name, args):
         """Handle a msgpack-rpc notification."""
+        if IS_PYTHON3 and isinstance(name, bytes):
+            name = name.decode(self._nvim_encoding)
         handler = self._notification_handlers.get(name, None)
         if not handler:
             warn('no notification handler registered for "%s"', name)