Python3 compatibility fixes

- Add Python3 compatibility wrapper from imp.find_module
- Python3 does not have a long type

Author: equalsraf

neovim/compat.py

@@ -1,6 +1,36 @@
 """Code for supporting compatibility across python versions."""
 
 import sys
+from imp import find_module as original_find_module
 
 
 IS_PYTHON3 = sys.version_info >= (3, 0)
+
+
+if IS_PYTHON3:
+    def find_module(fullname, path):
+        """Compatibility wrapper for imp.find_module.
+
+        Automatically decodes arguments of find_module, in Python3
+        they must be Unicode
+        """
+        if isinstance(fullname, bytes):
+            fullname = fullname.decode()
+        if isinstance(path, bytes):
+            path = path.decode()
+        elif isinstance(path, list):
+            newpath = []
+            for element in path:
+                if isinstance(element, bytes):
+                    newpath.append(element.decode())
+                else:
+                    newpath.append(element)
+            path = newpath
+        return original_find_module(fullname, path)
+
+    # There is no 'long' type in Python3 just int
+    long = int
+else:
+    find_module = original_find_module
+
+NUM_TYPES = (int, long, float)

neovim/plugins/plugin_host.py

@@ -3,10 +3,10 @@
 import os
 import os.path
 import sys
-from imp import find_module, load_module
+from imp import load_module
 from traceback import format_exc
 
-from ..compat import IS_PYTHON3
+from ..compat import IS_PYTHON3, find_module
 
 
 logger = logging.getLogger(__name__)

neovim/plugins/script_host.py

@@ -3,7 +3,7 @@
 import sys
 
 from ..api.common import SessionHook
-from ..compat import IS_PYTHON3
+from ..compat import NUM_TYPES, IS_PYTHON3
 
 
 logger = logging.getLogger(__name__)
@@ -98,6 +98,6 @@ def __init__(self):
         super(LegacyEvalHook, self).__init__(from_nvim=self._string_eval)
 
     def _string_eval(self, obj, session, method, kind):
-        if method == 'vim_eval' and isinstance(obj, (int, long, float)):
+        if method == 'vim_eval' and isinstance(obj, NUM_TYPES):
             return str(obj)
         return obj