diff --git a/python/src/com/jetbrains/python/psi/PyClass.java b/python/src/com/jetbrains/python/psi/PyClass.java index d5c6b7dfafde..47899fcca03d 100644 --- a/python/src/com/jetbrains/python/psi/PyClass.java +++ b/python/src/com/jetbrains/python/psi/PyClass.java @@ -2,7 +2,7 @@ package com.jetbrains.python.psi; import com.intellij.lang.ASTNode; import com.intellij.psi.PsiElement; -import com.intellij.psi.PsiNamedElement; +import com.intellij.psi.PsiNameIdentifierOwner; import com.intellij.psi.StubBasedPsiElement; import com.intellij.util.ArrayFactory; import com.intellij.util.Processor; @@ -18,7 +18,7 @@ import java.util.List; * Represents a class declaration in source. */ public interface PyClass extends - PsiNamedElement, PyStatement, NameDefiner, PyDocStringOwner, StubBasedPsiElement, ScopeOwner, PyDecoratable + PsiNameIdentifierOwner, PyStatement, NameDefiner, PyDocStringOwner, StubBasedPsiElement, ScopeOwner, PyDecoratable { ArrayFactory ARRAY_FACTORY = new ArrayFactory() { @Override diff --git a/python/src/com/jetbrains/python/psi/impl/PyClassImpl.java b/python/src/com/jetbrains/python/psi/impl/PyClassImpl.java index c74ce30c4d9c..570e8a46b34f 100644 --- a/python/src/com/jetbrains/python/psi/impl/PyClassImpl.java +++ b/python/src/com/jetbrains/python/psi/impl/PyClassImpl.java @@ -85,6 +85,11 @@ public class PyClassImpl extends PyPresentableElementImpl implement } } + public PsiElement getNameIdentifier() { + final ASTNode nameNode = getNameNode(); + return nameNode != null ? nameNode.getPsi() : null; + } + public ASTNode getNameNode() { return getNode().findChildByType(PyTokenTypes.IDENTIFIER); }