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 }; } + @Override + public String getName() { + return getVisibleName(); + } + @NotNull public Iterable 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