Add common.rethrow_uncaught

jedi/common.py

@@ -1,8 +1,10 @@
 """ A universal module with functions / classes without dependencies. """
+import sys
 import contextlib
+import functools
 import tokenize
 
-from _compatibility import next
+from _compatibility import next, reraise
 import debug
 import settings
 
@@ -33,6 +35,17 @@ class MultiLevelAttributeError(Exception):
         return 'Original:\n\n' + ''.join(tb)
 
 
+def rethrow_uncaught(func):
+    @functools.wraps(func)
+    def wrapper(*args, **kwds):
+        try:
+            return func(*args, **kwds)
+        except AttributeError:
+            exc_info = sys.exc_info()
+            reraise(MultiLevelAttributeError, exc_info[2])
+    return wrapper
+
+
 class PushBackIterator(object):
     def __init__(self, iterator):
         self.pushes = []