IDEA-CR-49176: inline function: optimize imports at declaration site after removing function definition (PY-36721)

GitOrigin-RevId: 2bf0ffdeff5a2b5cd084b53b5e1ed76d547d07f2

python/src/com/jetbrains/python/refactoring/inline/PyInlineFunctionProcessor.kt

@@ -295,16 +295,17 @@ class PyInlineFunctionProcessor(project: Project,
       .forEach { PyClassRefactoringUtil.optimizeImports(it) }
 
     if (myRemoveDeclaration) {
+      val file = myFunction.containingFile
       val stubFunction = PyiUtil.getPythonStub(myFunction)
       if (stubFunction != null && stubFunction.isWritable) {
         stubFunction.delete()
       }
-      val typingOverloads = PyiUtil.getOverloads(myFunction, TypeEvalContext.userInitiated(myProject, myFunction.containingFile))
+      val typingOverloads = PyiUtil.getOverloads(myFunction, TypeEvalContext.userInitiated(myProject, file))
       if (typingOverloads.isNotEmpty()) {
         typingOverloads.forEach { it.delete() }
-        PyClassRefactoringUtil.optimizeImports(myFunction.containingFile)
       }
       myFunction.delete()
+      PyClassRefactoringUtil.optimizeImports(file)
       dunderAll.forEach { it.element?.delete() }
     }
   }

python/testData/refactoring/inlineFunction/optimizeImportsAtDeclarationSite/main.after.py

@@ -0,0 +1,3 @@
+from datetime import datetime
+
+print(datetime.now())

python/testData/refactoring/inlineFunction/optimizeImportsAtDeclarationSite/main.py

@@ -0,0 +1,3 @@
+from src import foo
+
+f<caret>oo()

python/testData/refactoring/inlineFunction/optimizeImportsAtDeclarationSite/src.after.py

@@ -0,0 +1,2 @@
+
+

python/testData/refactoring/inlineFunction/optimizeImportsAtDeclarationSite/src.py

@@ -0,0 +1,4 @@
+from datetime import datetime
+
+def foo():
+    print(datetime.now())

python/testSrc/com/jetbrains/python/refactoring/PyInlineFunctionTest.kt

@@ -87,6 +87,11 @@ class PyInlineFunctionTest : PyTestCase() {
   fun testImportedLocally() = doTest(inlineThis = false, remove = true)
   //fun testInlineImportedAs() = doTest(inlineThis = false)
   fun testSelfUsageDetection() = doTest(inlineThis = false, remove = true)
+  fun testOptimizeImportsAtDeclarationSite() {
+    doTest(inlineThis = false, remove = true)
+    val testName = getTestName(true)
+    myFixture.checkResultByFile("src.py", "$testName/src.after.py",true)
+  }
   fun testRemoveFunctionWithStub() {
     doTest(inlineThis = false, remove = true)
     val testName = getTestName(true)