From 3638d5149de9ebed971649674944f93267b25e1c Mon Sep 17 00:00:00 2001
From: Dave Halter <davidhalter88@gmail.com>
Date: Mon, 6 Oct 2014 17:37:34 +0200
Subject: [PATCH] Change time_cache, to also host the star_import_cache.

---
 jedi/cache.py            | 42 ++++++++++++++--------------------------
 jedi/evaluate/imports.py |  2 +-
 2 files changed, 15 insertions(+), 29 deletions(-)

diff --git a/jedi/cache.py b/jedi/cache.py
index 0069b45c..7315b79c 100644
--- a/jedi/cache.py
+++ b/jedi/cache.py
@@ -31,7 +31,7 @@ from jedi import settings
 from jedi import common
 from jedi import debug
 
-_time_caches = []
+_time_caches = {}
 
 _star_import_cache = {}
 
@@ -57,12 +57,13 @@ def clear_time_caches(delete_all=False):
     global _time_caches
 
     if delete_all:
-        _time_caches[:] = []
+        for cache in _time_caches.values():
+            cache.clear()
         _star_import_cache.clear()
         parser_cache.clear()
     else:
         # normally just kill the expired entries, not all
-        for tc in _time_caches:
+        for tc in _time_caches.values():
             # check time_cache for expired entries
             for key, (t, value) in list(tc.items()):
                 if t < time.time():
@@ -71,14 +72,16 @@ def clear_time_caches(delete_all=False):
 
 
 def time_cache(time_add_setting):
-    """ This decorator works as follows: Call it with a setting and after that
+    """
+    s
+    This decorator works as follows: Call it with a setting and after that
     use the function with a callable that returns the key.
     But: This function is only called if the key is not available. After a
     certain amount of time (`time_add_setting`) the cache is invalid.
     """
     def _temp(key_func):
         dct = {}
-        _time_caches.append(dct)
+        _time_caches[time_add_setting] = dct
 
         def wrapper(*args, **kwargs):
             generator = key_func(*args, **kwargs)
@@ -165,37 +168,21 @@ def memoize(func):
 
 
 def cache_star_import(func):
+    @time_cache("star_import_cache_validity")
     def wrapper(evaluator, scope, *args, **kwargs):
-        with common.ignored(KeyError):
-            start_time, modules = _star_import_cache[scope]
-            if start_time + settings.star_import_cache_validity > time.time():
-                return modules
-        # cache is too old and therefore invalid or not available
-        _invalidate_star_import_cache_module(scope)
-        modules = func(evaluator, scope, *args, **kwargs)
-        _star_import_cache[scope] = time.time(), modules
-
-        return modules
+        yield scope  # The cache key
+        yield func(evaluator, scope, *args, **kwargs)
     return wrapper
 
 
 def _invalidate_star_import_cache_module(module, only_main=False):
     """ Important if some new modules are being reparsed """
     try:
-        t, modules = _star_import_cache[module]
+        t, modules = _time_caches['star_import_cache_validity'][module]
     except KeyError:
         pass
     else:
-        del _star_import_cache[module]
-        for m in modules:
-            _invalidate_star_import_cache_module(m, only_main=True)
-
-    if not only_main:
-        # We need a list here because otherwise the list is being changed
-        # during the iteration in py3k: iteritems -> items.
-        for key, (t, modules) in list(_star_import_cache.items()):
-            if module in modules:
-                _invalidate_star_import_cache_module(key)
+        del _time_caches['star_import_cache_validity'][module]
 
 
 def invalidate_star_import_cache(path):
@@ -203,10 +190,9 @@ def invalidate_star_import_cache(path):
     try:
         parser_cache_item = parser_cache[path]
     except KeyError:
-        return False
+        pass
     else:
         _invalidate_star_import_cache_module(parser_cache_item.parser.module)
-        return True
 
 
 def load_parser(path, name):
diff --git a/jedi/evaluate/imports.py b/jedi/evaluate/imports.py
index 04f11ac5..a1ceb330 100644
--- a/jedi/evaluate/imports.py
+++ b/jedi/evaluate/imports.py
@@ -377,7 +377,7 @@ class _Importer(object):
                     pr.Name(FakeSubModule, 'flask_' + str(part), part.parent, pos),
                 ) + orig_path[3:]
                 return self._real_follow_file_system()
-            except ModuleNotFound as e:
+            except ModuleNotFound:
                 self.import_path = (
                     pr.Name(FakeSubModule, 'flaskext', part.parent, pos),
                 ) + orig_path[2:]