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

Migrate UnresolvedReferenceAddSelfQuickFix to ModCommand

Browse Source 
PY-65297

GitOrigin-RevId: 4265fda32fa1fe27fe04e7cdea98840cc90c7eee
This commit is contained in:
Georgii Ustinov
2024-01-05 09:10:51 +00:00
committed by intellij-monorepo-bot
parent 007259a141
commit 4d2e4a3197
1 changed files with 10 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/python-psi-impl/src/com/jetbrains/python/inspections/quickfix/UnresolvedReferenceAddSelfQuickFix.java
+10 -9
View File
@@ -2,9 +2,10 @@
package com.jetbrains.python.inspections.quickfix;
import com.intellij.codeInsight.intention.HighPriorityAction;
import com.intellij.codeInspection.LocalQuickFix;
import com.intellij.codeInspection.ProblemDescriptor;
import com.intellij.modcommand.ModPsiUpdater;
import com.intellij.modcommand.PsiUpdateModCommandQuickFix;
import com.intellij.openapi.project.Project;
import com.intellij.psi.PsiElement;
import com.jetbrains.python.PyPsiBundle;
import com.jetbrains.python.psi.LanguageLevel;
import com.jetbrains.python.psi.PyElementGenerator;
@@ -19,7 +20,7 @@ import static com.jetbrains.python.psi.PyUtil.as;
*
* QuickFix to add self to unresolved reference
*/
public class UnresolvedReferenceAddSelfQuickFix implements LocalQuickFix, HighPriorityAction {
public class UnresolvedReferenceAddSelfQuickFix extends PsiUpdateModCommandQuickFix implements HighPriorityAction {
private final String myQualifier;
private final String myAttributeName;
@@ -41,12 +42,12 @@ public class UnresolvedReferenceAddSelfQuickFix implements LocalQuickFix, HighPr
}
@Override
public void applyFix(@NotNull Project project, @NotNull ProblemDescriptor descriptor) {
final PyReferenceExpression element = as(descriptor.getPsiElement(), PyReferenceExpression.class);
if (element == null) return;
public void applyFix(@NotNull Project project, @NotNull PsiElement element, @NotNull ModPsiUpdater updater) {
final PyReferenceExpression reference = as(element, PyReferenceExpression.class);
if (reference == null) return;
PyElementGenerator elementGenerator = PyElementGenerator.getInstance(project);
PyExpression expression = elementGenerator.createExpressionFromText(LanguageLevel.forElement(element),
myQualifier + "." + element.getText());
element.replace(expression);
PyExpression expression = elementGenerator.createExpressionFromText(LanguageLevel.forElement(reference),
myQualifier + "." + reference.getText());
reference.replace(expression);
}
}