From 27ff928ed3a2a9f38ae7fcbce9c6fbab9a9155b7 Mon Sep 17 00:00:00 2001
From: fitermay <fiterman@gmail.com>
Date: Fri, 25 Aug 2017 04:55:25 +0300
Subject: [PATCH] PY-25794, PY-25491: prevent PyImportElementImpl iterateNames
 from unstubbing files

(cherry picked from commit 4b85d35)
---
 .../python/psi/impl/PyImportElementImpl.java  | 19 +++++++------------
 1 file changed, 7 insertions(+), 12 deletions(-)

diff --git a/python/src/com/jetbrains/python/psi/impl/PyImportElementImpl.java b/python/src/com/jetbrains/python/psi/impl/PyImportElementImpl.java
index 7e2b47f9d61a..6197c3c72128 100644
--- a/python/src/com/jetbrains/python/psi/impl/PyImportElementImpl.java
+++ b/python/src/com/jetbrains/python/psi/impl/PyImportElementImpl.java
@@ -24,7 +24,6 @@ import com.intellij.psi.PsiFile;
 import com.intellij.psi.stubs.IStubElementType;
 import com.intellij.psi.util.PsiTreeUtil;
 import com.intellij.psi.util.QualifiedName;
-import com.intellij.util.containers.EmptyIterable;
 import com.jetbrains.python.PyElementTypes;
 import com.jetbrains.python.PythonDialectsTokenSetProvider;
 import com.jetbrains.python.psi.*;
@@ -180,19 +179,15 @@ public class PyImportElementImpl extends PyBaseElementImpl<PyImportElementStub>
     };
   }
 
+  @Override
+  public String getName() {
+    return getVisibleName();
+  }
+
   @NotNull
   public Iterable<PyElement> iterateNames() {
-    PyElement ret = getAsNameElement();
-    if (ret == null) {
-      final PyReferenceExpression importReference = getImportReferenceExpression();
-      if (importReference != null) {
-        ret = PyPsiUtils.getFirstQualifier(importReference);
-      }
-    }
-    if (ret == null) {
-      return EmptyIterable.getInstance();
-    }
-    return Collections.singleton(ret);
+    final String visibleName = getVisibleName();
+    return visibleName != null ? Collections.singletonList(this) : Collections.emptyList();
   }
 
   @NotNull