pull up getReference(PyResolveContext) to PyQualifiedReference interface

This commit is contained in:
Dmitry Jemerov
2012-08-28 16:44:53 +02:00
parent cd2a799a6e
commit 6804db3b66
3 changed files with 14 additions and 5 deletions
@@ -1,6 +1,9 @@
package com.jetbrains.python.psi;
import com.intellij.lang.ASTNode;
import com.intellij.psi.PsiPolyVariantReference;
import com.jetbrains.python.psi.resolve.PyResolveContext;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
/**
@@ -27,4 +30,7 @@ public interface PyQualifiedExpression extends PyExpression {
*/
@Nullable
ASTNode getNameElement();
@NotNull
PsiPolyVariantReference getReference(PyResolveContext resolveContext);
}
@@ -30,7 +30,4 @@ public interface PyReferenceExpression extends PyQualifiedExpression, PyReferenc
@NotNull
PsiPolyVariantReference getReference();
@NotNull
PsiPolyVariantReference getReference(PyResolveContext resolveContext);
}
@@ -5,6 +5,7 @@ import com.intellij.openapi.extensions.Extensions;
import com.intellij.openapi.util.Pair;
import com.intellij.openapi.util.Ref;
import com.intellij.psi.PsiElement;
import com.intellij.psi.PsiPolyVariantReference;
import com.intellij.psi.PsiReference;
import com.intellij.psi.ResolveResult;
import com.intellij.psi.search.GlobalSearchScope;
@@ -481,10 +482,15 @@ public class PyTargetExpressionImpl extends PyPresentableElementImpl<PyTargetExp
@NotNull
@Override
public PsiReference getReference() {
return getReference(PyResolveContext.defaultContext());
}
@NotNull
public PsiPolyVariantReference getReference(final PyResolveContext resolveContext) {
if (getQualifier() != null) {
return new PyQualifiedReference(this, PyResolveContext.defaultContext());
return new PyQualifiedReference(this, resolveContext);
}
return new PyTargetReference(this, PyResolveContext.defaultContext());
return new PyTargetReference(this, resolveContext);
}
@NotNull