MDS/openide
1
0
Fork 0
mirror of https://gitflic.ru/project/openide/openide.git synced 2026-09-27 10:03:11 +07:00
Code Issues Packages Projects Releases Wiki Activity

Extracted getMetaClassExpression()

Browse Source
This commit is contained in:
Andrey Vlasovskikh
2014-01-21 00:29:20 +04:00
parent 48b90c2f22
commit 1ee1c255cb
1 changed files with 31 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/src/com/jetbrains/python/psi/impl/PyClassImpl.java
+31 -37
View File
@@ -1170,49 +1170,43 @@ public class PyClassImpl extends PyPresentableElementImpl<PyClassStub> implement
@Nullable
@Override
public PyClassLikeType getMetaClassType(@NotNull TypeEvalContext context) {
final LanguageLevel level = LanguageLevel.forElement(this);
if (level.isAtLeast(LanguageLevel.PYTHON30)) {
// TODO: This requires switching from stubs to AST
final PyExpression[] superClassExpressions = getSuperClassExpressions();
for (PyExpression superClassExpression : superClassExpressions) {
if (superClassExpression instanceof PyKeywordArgument) {
final PyKeywordArgument argument = (PyKeywordArgument)superClassExpression;
if (PyNames.METACLASS.equals(argument.getKeyword())) {
final PyClassLikeType type = getClassLikeType(argument, context);
if (type != null) {
return type;
}
}
}
}
}
else {
final PyTargetExpression metaClassAttribute = findClassAttribute(PyNames.DUNDER_METACLASS, false);
if (metaClassAttribute != null) {
final PyClassLikeType type = getClassLikeType(metaClassAttribute, context);
if (type != null) {
return type;
}
}
final PsiFile containingFile = getContainingFile();
if (containingFile instanceof PyFile) {
final PsiElement element = ((PyFile)containingFile).getElementNamed(PyNames.DUNDER_METACLASS);
if (element instanceof PyTypedElement) {
final PyClassLikeType type = getClassLikeType((PyTypedElement)element, context);
if (type != null) {
return type;
}
}
final PyExpression expression = getMetaClassExpression();
if (expression != null) {
final PyType type = context.getType(expression);
if (type instanceof PyClassLikeType) {
return (PyClassLikeType)type;
}
}
return null;
}
@Nullable
private static PyClassLikeType getClassLikeType(@NotNull PyTypedElement element, @NotNull TypeEvalContext context) {
final PyType type = context.getType(element);
if (type instanceof PyClassLikeType) {
return (PyClassLikeType)type;
public PyExpression getMetaClassExpression() {
final LanguageLevel level = LanguageLevel.forElement(this);
if (level.isAtLeast(LanguageLevel.PYTHON30)) {
// Requires AST access
for (PyExpression expression : getSuperClassExpressions()) {
if (expression instanceof PyKeywordArgument) {
final PyKeywordArgument argument = (PyKeywordArgument)expression;
if (PyNames.METACLASS.equals(argument.getKeyword())) {
return argument.getValueExpression();
}
}
}
}
else {
final PyTargetExpression attribute = findClassAttribute(PyNames.DUNDER_METACLASS, false);
if (attribute != null) {
return attribute;
}
final PsiFile file = getContainingFile();
if (file instanceof PyFile) {
final PyFile pyFile = (PyFile)file;
final PsiElement element = pyFile.getElementNamed(PyNames.DUNDER_METACLASS);
if (element instanceof PyExpression) {
return (PyExpression)element;
}
}
}
return null;
}