when completing members of imported module, look at local imports, not only file-level ones (PY-3668)

python/src/com/jetbrains/python/psi/types/PyImportedModuleType.java

@@ -4,7 +4,9 @@ import com.intellij.codeInsight.lookup.LookupElement;
 import com.intellij.codeInsight.lookup.LookupElementBuilder;
 import com.intellij.psi.PsiDirectory;
 import com.intellij.psi.PsiElement;
+import com.intellij.psi.util.PsiTreeUtil;
 import com.intellij.util.ProcessingContext;
+import com.jetbrains.python.codeInsight.controlflow.ScopeOwner;
 import com.jetbrains.python.psi.AccessDirection;
 import com.jetbrains.python.psi.PyExpression;
 import com.jetbrains.python.psi.PyFile;
@@ -51,7 +53,9 @@ public class PyImportedModuleType implements PyType {
 
   public Object[] getCompletionVariants(String completionPrefix, PyExpression location, ProcessingContext context) {
     List<LookupElement> result = new ArrayList<LookupElement>();
-    final List<PyImportElement> importTargets = myImportedModule.getContainingFile().getImportTargets();
+    ScopeOwner scopeOwner = PsiTreeUtil.getParentOfType(location, ScopeOwner.class);
+    assert scopeOwner != null;
+    final List<PyImportElement> importTargets = PyModuleType.getVisibleImports(scopeOwner);
     final int imported = myImportedModule.getImportedPrefix().getComponentCount();
     for (PyImportElement importTarget : importTargets) {
       final PyQualifiedName qName = importTarget.getImportedQName();

python/src/com/jetbrains/python/psi/types/PyModuleType.java

@@ -167,7 +167,7 @@ public class PyModuleType implements PyType { // Modules don't descend from obje
   }
 
   @NotNull
-  private static List<PyImportElement> getVisibleImports(@NotNull ScopeOwner owner) {
+  public static List<PyImportElement> getVisibleImports(@NotNull ScopeOwner owner) {
     final List<PyImportElement> visibleImports = new ArrayList<PyImportElement>();
     PyResolveUtil.scopeCrawlUp(new PsiScopeProcessor() {
       @Override

python/testData/completion/py3668/py3668.after.py

@@ -0,0 +1,3 @@
+def func():
+    import foo.bar
+    foo.bar

python/testData/completion/py3668/py3668.py

@@ -0,0 +1,3 @@
+def func():
+    import foo.bar
+    foo.<caret>

python/testSrc/com/jetbrains/python/PythonCompletionTest.java

@@ -613,4 +613,11 @@ public class PythonCompletionTest extends PyTestCase {
     doTestByText("def foo(**<caret>)");
     myFixture.checkResult("def foo(**kwargs)");
   }
+
+  public void testLocalImportedModule() {  // PY-3668
+    myFixture.copyDirectoryToProject("completion/py3668", "");
+    myFixture.configureByFile("py3668.py");
+    myFixture.completeBasic();
+    myFixture.checkResultByFile("completion/py3668/py3668.after.py");
+  }
 }