PyClass implements PsiNameIdentifierOwner

This commit is contained in:
Dmitry Jemerov
2011-02-21 13:17:42 +01:00
parent d598c1fefa
commit 4326688ef2
2 changed files with 7 additions and 2 deletions
@@ -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<PyClassStub>, ScopeOwner, PyDecoratable
PsiNameIdentifierOwner, PyStatement, NameDefiner, PyDocStringOwner, StubBasedPsiElement<PyClassStub>, ScopeOwner, PyDecoratable
{
ArrayFactory<PyClass> ARRAY_FACTORY = new ArrayFactory<PyClass>() {
@Override
@@ -85,6 +85,11 @@ public class PyClassImpl extends PyPresentableElementImpl<PyClassStub> implement
}
}
public PsiElement getNameIdentifier() {
final ASTNode nameNode = getNameNode();
return nameNode != null ? nameNode.getPsi() : null;
}
public ASTNode getNameNode() {
return getNode().findChildByType(PyTokenTypes.IDENTIFIER);
}