Another case of import action inside an import statement quashed.

This commit is contained in:
Dmitry Cheryasov
2008-05-27 04:19:53 +04:00
parent 5072d6753a
commit e765989a3d
@@ -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;