From 2f3425afe529453182a9dd448703bf30da6068fd Mon Sep 17 00:00:00 2001 From: Vladimir Dolzhenko Date: Fri, 15 Mar 2024 15:12:12 +0100 Subject: [PATCH] Add fast applicability check GitOrigin-RevId: a7702d51d6f31eb64ccfb7bd95c4eb7d90179690 --- .../codeInsight/intentions/PyHideTypeAnnotationsIntention.kt | 2 ++ 1 file changed, 2 insertions(+) diff --git a/python/python-psi-impl/src/com/jetbrains/python/codeInsight/intentions/PyHideTypeAnnotationsIntention.kt b/python/python-psi-impl/src/com/jetbrains/python/codeInsight/intentions/PyHideTypeAnnotationsIntention.kt index 98d2a38b04ec..40f6ec9f70a5 100644 --- a/python/python-psi-impl/src/com/jetbrains/python/codeInsight/intentions/PyHideTypeAnnotationsIntention.kt +++ b/python/python-psi-impl/src/com/jetbrains/python/codeInsight/intentions/PyHideTypeAnnotationsIntention.kt @@ -8,6 +8,7 @@ import com.intellij.psi.PsiFile import com.intellij.psi.util.PsiTreeUtil import com.jetbrains.python.PyPsiBundle import com.jetbrains.python.psi.PyAnnotation +import com.jetbrains.python.psi.PyFile import com.jetbrains.python.psi.PyUtil /** @@ -19,6 +20,7 @@ class PyHideTypeAnnotationsIntention : PyBaseIntentionAction() { } override fun isAvailable(project: Project, editor: Editor?, file: PsiFile?): Boolean { + if (file !is PyFile) return false val offset = TargetElementUtilBase.adjustOffset(file, editor?.document ?: return false, editor.caretModel.offset) val element = PyUtil.findNonWhitespaceAtOffset(file, offset) val annotation = PsiTreeUtil.getParentOfType(element, PyAnnotation::class.java)