PY-13668 Fix anchor element for local import insertion

python/src/com/jetbrains/python/codeInsight/imports/AddImportHelper.java

@@ -74,8 +74,7 @@ public class AddImportHelper {
 
   @Nullable
   public static PsiElement getLocalInsertPosition(@NotNull PyElement anchor) {
-    final PyStatement enclosingStatement = PsiTreeUtil.getParentOfType(anchor, PyStatement.class, false);
-    return enclosingStatement != null ? enclosingStatement.getFirstChild() : null;
+    return PsiTreeUtil.getParentOfType(anchor, PyStatement.class, false);
   }
 
   public enum ImportPriority {

python/src/com/jetbrains/python/codeInsight/imports/ImportCandidateHolder.java

@@ -52,7 +52,7 @@ class ImportCandidateHolder implements Comparable<ImportCandidateHolder> {
    * Creates new instance.
    *
    * @param importable    an element that could be imported either from import element or from file.
-   * @param file          the file which is the source of the importable
+   * @param file          the file which is the source of the importable (module for symbols, containing directory for modules and packages)
    * @param importElement an existing import element that can be a source for the importable.
    * @param path          import path for the file, as a qualified name (a.b.c)
    *                      For top-level imported symbols it's <em>qualified name of containing module</em> (or package for __init__.py).

python/testData/addImport/localFromImport.after.py

@@ -1,5 +1,6 @@
 def func():
     for _ range(10):
         from package.module import foo
+
         foo
 #       <ref>

python/testData/addImport/localImport.after.py

@@ -1,6 +1,7 @@
 def func():
     try:
         import module
+
         module
 #       <ref>
     except:

python/testData/addImport/localImportInlineBranch.after.py

@@ -0,0 +1,6 @@
+def func():
+    if True:
+        import module
+
+        module
+#            <ref>

python/testData/addImport/localImportInlineBranch.py

@@ -0,0 +1,3 @@
+def func():
+    if True: module
+#            <ref>

python/testData/addImport/localImportInlineFunctionBody.after.py

@@ -0,0 +1,5 @@
+def func():
+    import module
+
+    module
+#           <ref>

python/testData/addImport/localImportInlineFunctionBody.py

@@ -0,0 +1,2 @@
+def func(): module
+#           <ref>

python/testSrc/com/jetbrains/python/PyAddImportTest.java

@@ -69,6 +69,16 @@ public class PyAddImportTest extends PyTestCase {
     doAddLocalImport("module", null);
   }
 
+  // PY-13668
+  public void testLocalImportInlineFunctionBody() {
+    testLocalImport();
+  }
+
+  // PY-13668
+  public void testLocalImportInlineBranch() {
+    testLocalImport();
+  }
+
   /**
    * Add local import statement
    * @param name      reference name in corresponding import element