From eb35f95a8d2795c4c394e80399a3d584b72cbdb8 Mon Sep 17 00:00:00 2001
From: Georgii Ustinov <georgii.ustinov@jetbrains.com>
Date: Thu, 14 Dec 2023 17:03:02 +0200
Subject: [PATCH] Migrate PyCreatePropertyQuickFix and
 PyDefaultArgumentQuickFix to ModCommand

PY-65297

GitOrigin-RevId: 7c5579c5f2f2a6f06ada8306eb82573ba684cbcf
---
 .../quickfix/PyCreatePropertyQuickFix.java           | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/python/python-psi-impl/src/com/jetbrains/python/inspections/quickfix/PyCreatePropertyQuickFix.java b/python/python-psi-impl/src/com/jetbrains/python/inspections/quickfix/PyCreatePropertyQuickFix.java
index 81222dd8ee40..06a58536f3d0 100644
--- a/python/python-psi-impl/src/com/jetbrains/python/inspections/quickfix/PyCreatePropertyQuickFix.java
+++ b/python/python-psi-impl/src/com/jetbrains/python/inspections/quickfix/PyCreatePropertyQuickFix.java
@@ -1,8 +1,8 @@
 // Copyright 2000-2018 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE file.
 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.jetbrains.python.PyPsiBundle;
@@ -13,7 +13,7 @@ import com.jetbrains.python.psi.types.TypeEvalContext;
 import com.jetbrains.python.refactoring.PyPsiRefactoringUtil;
 import org.jetbrains.annotations.NotNull;
 
-public class PyCreatePropertyQuickFix implements LocalQuickFix {
+public class PyCreatePropertyQuickFix extends PsiUpdateModCommandQuickFix {
   private final AccessDirection myAccessDirection;
 
   public PyCreatePropertyQuickFix(AccessDirection dir) {
@@ -27,8 +27,7 @@ public class PyCreatePropertyQuickFix implements LocalQuickFix {
   }
 
   @Override
-  public void applyFix(@NotNull final Project project, @NotNull final ProblemDescriptor descriptor) {
-    final PsiElement element = descriptor.getPsiElement();
+  public void applyFix(@NotNull final Project project, @NotNull final PsiElement element, @NotNull ModPsiUpdater updater) {
     if (element instanceof PyQualifiedExpression) {
       final PyExpression qualifier = ((PyQualifiedExpression)element).getQualifier();
       if (qualifier != null) {
@@ -40,7 +39,8 @@ public class PyCreatePropertyQuickFix implements LocalQuickFix {
           final String fieldName = "_" + propertyName;
           final PyElementGenerator generator = PyElementGenerator.getInstance(project);
           final PyFunction property = generator.createProperty(LanguageLevel.forElement(cls), propertyName, fieldName, myAccessDirection);
-          PyPsiRefactoringUtil.addElementToStatementList(property, cls.getStatementList(), myAccessDirection == AccessDirection.READ);
+          final PyStatementList statementsList = updater.getWritable(cls.getStatementList());
+          PyPsiRefactoringUtil.addElementToStatementList(property, statementsList, myAccessDirection == AccessDirection.READ);
         }
       }
     }