mirror of
https://gitflic.ru/project/openide/openide.git
synced 2026-09-27 10:03:11 +07:00
Use TypeEvalContext.userInitiated() for replace tuple with list quickfix (PY-15558)
The default type evaluation context doesn't allow to perform proper follow assignments chain even for AST-based trees now (which is consistent with stub-based trees). We have to specify the appropriate context manually.
This commit is contained in:
+4
-1
@@ -22,6 +22,7 @@ import com.intellij.psi.PsiElement;
|
||||
import com.jetbrains.python.PyBundle;
|
||||
import com.jetbrains.python.psi.*;
|
||||
import com.jetbrains.python.psi.resolve.PyResolveContext;
|
||||
import com.jetbrains.python.psi.types.TypeEvalContext;
|
||||
import org.jetbrains.annotations.NotNull;
|
||||
|
||||
public class PyReplaceTupleWithListQuickFix implements LocalQuickFix {
|
||||
@@ -46,7 +47,9 @@ public class PyReplaceTupleWithListQuickFix implements LocalQuickFix {
|
||||
PySubscriptionExpression subscriptionExpression = (PySubscriptionExpression)targets[0];
|
||||
if (subscriptionExpression.getOperand() instanceof PyReferenceExpression) {
|
||||
PyReferenceExpression referenceExpression = (PyReferenceExpression)subscriptionExpression.getOperand();
|
||||
element = referenceExpression.followAssignmentsChain(PyResolveContext.defaultContext()).getElement();
|
||||
final TypeEvalContext context = TypeEvalContext.userInitiated(project, element.getContainingFile());
|
||||
final PyResolveContext resolveContext = PyResolveContext.noImplicits().withTypeEvalContext(context);
|
||||
element = referenceExpression.followAssignmentsChain(resolveContext).getElement();
|
||||
if (element instanceof PyParenthesizedExpression) {
|
||||
final PyExpression expression = ((PyParenthesizedExpression)element).getContainedExpression();
|
||||
replaceWithListLiteral(element, (PyTupleExpression)expression);
|
||||
|
||||
Reference in New Issue
Block a user