From df9c9d8dff9beebf47ec3100aa0d9c7d96420985 Mon Sep 17 00:00:00 2001
From: Dave Halter <davidhalter88@gmail.com>
Date: Wed, 1 May 2019 10:47:20 +0200
Subject: [PATCH] Fix a flask issue

---
 jedi/plugins/flask.py | 17 +++++++++--------
 1 file changed, 9 insertions(+), 8 deletions(-)

diff --git a/jedi/plugins/flask.py b/jedi/plugins/flask.py
index c4906c4c..5385b6ba 100644
--- a/jedi/plugins/flask.py
+++ b/jedi/plugins/flask.py
@@ -12,13 +12,14 @@ class FlaskPlugin(BasePlugin):
                 # New style.
                 ipath = (u'flask_' + import_names[2]),
                 context_set = callback(evaluator, ipath, None, sys_path)
-                if not context_set:
-                    context_set = callback(evaluator, (u'flaskext',), None, sys_path)
-                    return callback(
-                        evaluator,
-                        (u'flaskext', import_names[2]),
-                        next(iter(context_set)),
-                        sys_path
-                    )
+                if context_set:
+                    return context_set
+                context_set = callback(evaluator, (u'flaskext',), None, sys_path)
+                return callback(
+                    evaluator,
+                    (u'flaskext', import_names[2]),
+                    next(iter(context_set)),
+                    sys_path
+                )
             return callback(evaluator, import_names, module_context, sys_path)
         return wrapper