IDEA-CR-49176: select distinct files before optimizing imports (PY-36484)

GitOrigin-RevId: 48b187e19ee36fd274c1670ecc388c661f11f774

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

@@ -282,7 +282,11 @@ class PyInlineFunctionProcessor(project: Project,
       }
     }
 
-    imports.forEach { PyClassRefactoringUtil.optimizeImports(it.element!!.containingFile!!) }
+    imports.asSequence()
+      .map { it.element?.containingFile }
+      .filterNotNull()
+      .distinct()
+      .forEach { PyClassRefactoringUtil.optimizeImports(it) }
 
     if (myRemoveDeclaration) {
       val stubFunction = PyiUtil.getPythonStub(myFunction)

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

@@ -0,0 +1,9 @@
+z = 1 + 2
+print(z)
+res = z
+
+
+def bar():
+    z = 1 + 2
+    print(z)
+    res1 = z

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

@@ -0,0 +1,9 @@
+from src import foo
+
+
+res = foo(1, 2)
+
+
+def bar():
+    from src import foo
+    res1 = fo<caret>o(1, 2)

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

@@ -0,0 +1,4 @@
+def foo(x, y):
+    z = x + y
+    print(z)
+    return z

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

@@ -0,0 +1,3 @@
+z = 1 + 2
+print(z)
+res = z

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

@@ -0,0 +1,3 @@
+from src import fo<caret>o
+
+res = foo(1, 2)

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

@@ -0,0 +1,4 @@
+def foo(x, y):
+    z = x + y
+    print(z)
+    return z

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

@@ -83,6 +83,8 @@ class PyInlineFunctionTest : PyTestCase() {
   fun testRemovingDocstringAndTypeComment() = doTest()
   fun testRemovingDocstringOfEmptyFunction() = doTest()
   fun testKeepingComments() = doTest()
+  fun testInvocationOnImport() = doTest(inlineThis = false, remove = true)
+  fun testImportedLocally() = doTest(inlineThis = false, remove = true)
   //fun testInlineImportedAs() = doTest(inlineThis = false)
   fun testRemoveFunctionWithStub() {
     doTest(inlineThis = false, remove = true)