diff --git a/python/src/com/jetbrains/python/validation/AddImportAction.java b/python/src/com/jetbrains/python/validation/AddImportAction.java index 10f219c313df..5ab098ed3003 100644 --- a/python/src/com/jetbrains/python/validation/AddImportAction.java +++ b/python/src/com/jetbrains/python/validation/AddImportAction.java @@ -16,11 +16,12 @@ import com.intellij.psi.PsiComment; import com.intellij.psi.PsiElement; import com.intellij.psi.PsiFile; import com.intellij.psi.PsiReference; -import com.intellij.psi.util.PsiTreeUtil; import com.intellij.psi.search.FilenameIndex; import com.intellij.psi.search.GlobalSearchScope; +import com.intellij.psi.util.PsiTreeUtil; import com.intellij.util.IncorrectOperationException; import com.jetbrains.python.PythonLanguage; +import com.jetbrains.python.psi.PyFromImportStatement; import com.jetbrains.python.psi.PyImportStatement; import com.jetbrains.python.psi.PyReferenceExpression; import org.jetbrains.annotations.NotNull; @@ -59,6 +60,7 @@ public class AddImportAction implements HintAction, QuestionAction { public boolean isAvailable(@NotNull final Project project, final Editor editor, final PsiFile file) { if (PsiTreeUtil.getParentOfType(myReference.getElement(), PyImportStatement.class) != null) return false; + if (PsiTreeUtil.getParentOfType(myReference.getElement(), PyFromImportStatement.class) != null) return false; final String referenceName = getRefName(); final PsiFile[] files = getRefFiles(referenceName); return files != null && files.length > 0;