MDS/openide
1
0
Fork 0
mirror of https://gitflic.ru/project/openide/openide.git synced 2026-01-06 03:21:12 +07:00
Code Issues Packages Projects Releases Wiki Activity

Migrate PyRemoveWithTargetQuickFix and PyReplaceStarByUnpackQuickFix to ModCommand

Browse Source 
PY-65297

GitOrigin-RevId: 1530cfd4c31b48f980953538f3211c41902076bd
This commit is contained in:
Georgii Ustinov
2023-12-18 14:39:57 +02:00
committed by intellij-monorepo-bot
parent 4f7601c1ed
commit 58cf31d9f8
2 changed files with 9 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
python/python-psi-impl/src/com/jetbrains/python/inspections/quickfix/PyRemoveWithTargetQuickFix.java
View File

@@ -15,8 +15,8 @@
*/
package com.jetbrains.python.inspections.quickfix;
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.intellij.psi.util.PsiTreeUtil;
@@ -25,7 +25,7 @@ import com.jetbrains.python.psi.PyExpression;
import com.jetbrains.python.psi.PyWithItem;
import org.jetbrains.annotations.NotNull;
public class PyRemoveWithTargetQuickFix implements LocalQuickFix {
public class PyRemoveWithTargetQuickFix extends PsiUpdateModCommandQuickFix {
@NotNull
@Override
public String getFamilyName() {
@@ -33,8 +33,7 @@ public class PyRemoveWithTargetQuickFix implements LocalQuickFix {
}
@Override
public void applyFix(@NotNull Project project, @NotNull ProblemDescriptor descriptor) {
final PsiElement element = descriptor.getPsiElement();
public void applyFix(@NotNull Project project, @NotNull PsiElement element, @NotNull ModPsiUpdater updater) {
final PyWithItem withItem = PsiTreeUtil.getParentOfType(element, PyWithItem.class);
if (withItem == null) return;
final PyExpression withTarget = withItem.getTarget();

12
python/python-psi-impl/src/com/jetbrains/python/inspections/quickfix/PyReplaceStarByUnpackQuickFix.java
View File

@@ -1,7 +1,7 @@
package com.jetbrains.python.inspections.quickfix;
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.intellij.psi.PsiFile;
@@ -10,7 +10,7 @@ import com.jetbrains.python.psi.PyExpression;
import com.jetbrains.python.psi.PyStarExpression;
import org.jetbrains.annotations.NotNull;
public class PyReplaceStarByUnpackQuickFix implements LocalQuickFix {
public class PyReplaceStarByUnpackQuickFix extends PsiUpdateModCommandQuickFix {
@NotNull
@Override
public String getFamilyName() {
@@ -18,11 +18,9 @@ public class PyReplaceStarByUnpackQuickFix implements LocalQuickFix {
}
@Override
public void applyFix(@NotNull Project project, @NotNull ProblemDescriptor descriptor) {
PsiElement element = descriptor.getPsiElement();
if (!(element instanceof PyStarExpression)) return;
public void applyFix(@NotNull Project project, @NotNull PsiElement element, @NotNull ModPsiUpdater updater) {
if (!(element instanceof PyStarExpression starExpression)) return;
PyStarExpression starExpression = (PyStarExpression)element;
PyExpression expression = starExpression.getExpression();
if (expression == null) return;
PsiFile file = starExpression.getContainingFile();