Provide completion and resolve for from tensorflow.<...> statements (PY-33034)

GitOrigin-RevId: 5886d44f185f918cf0089249209495de3145b6d1
This commit is contained in:
Semyon Proshev
2019-07-09 12:52:33 +03:00
committed by intellij-monorepo-bot
parent c47d883ba3
commit 16a77f4042
5 changed files with 109 additions and 2 deletions

View File

@@ -8,6 +8,7 @@ import com.intellij.psi.PsiElement;
import com.intellij.psi.PsiPackage;
import com.intellij.psi.util.QualifiedName;
import com.jetbrains.python.psi.resolve.PyQualifiedNameResolveContext;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
/**
@@ -16,7 +17,7 @@ import org.jetbrains.annotations.Nullable;
public class PyJavaImportResolver implements PyImportResolver {
@Override
@Nullable
public PsiElement resolveImportReference(QualifiedName name, PyQualifiedNameResolveContext context, boolean withRoots) {
public PsiElement resolveImportReference(@NotNull QualifiedName name, @NotNull PyQualifiedNameResolveContext context, boolean withRoots) {
String fqn = name.toString();
final JavaPsiFacade psiFacade = JavaPsiFacade.getInstance(context.getProject());
final PsiPackage aPackage = psiFacade.findPackage(fqn);