diff --git a/python/python-ast/src/com/jetbrains/python/ast/PyAstImportElement.java b/python/python-ast/src/com/jetbrains/python/ast/PyAstImportElement.java index 4e22f6e298f0..87be14a6efa6 100644 --- a/python/python-ast/src/com/jetbrains/python/ast/PyAstImportElement.java +++ b/python/python-ast/src/com/jetbrains/python/ast/PyAstImportElement.java @@ -14,6 +14,11 @@ import static com.jetbrains.python.ast.PyAstElementKt.findChildByType; @ApiStatus.Experimental public interface PyAstImportElement extends PyAstElement, PyAstImportedNameDefiner { + @Override + default @Nullable String getName() { + return getVisibleName(); + } + default @Nullable PyAstReferenceExpression getImportReferenceExpression() { final ASTNode node = getNode().findChildByType(PythonDialectsTokenSetProvider.getInstance().getReferenceExpressionTokens()); return node == null ? null : (PyAstReferenceExpression) node.getPsi(); diff --git a/python/python-psi-impl/src/com/jetbrains/python/psi/impl/PyImportElementImpl.java b/python/python-psi-impl/src/com/jetbrains/python/psi/impl/PyImportElementImpl.java index 1d471547ac3a..7cc9ae17e9ac 100644 --- a/python/python-psi-impl/src/com/jetbrains/python/psi/impl/PyImportElementImpl.java +++ b/python/python-psi-impl/src/com/jetbrains/python/psi/impl/PyImportElementImpl.java @@ -147,7 +147,7 @@ public class PyImportElementImpl extends PyBaseElementImpl @Override public String getName() { - return getVisibleName(); + return PyImportElement.super.getName(); } @Override