From 26130cb2e9c2dd1529dcc7d42659147a88965d12 Mon Sep 17 00:00:00 2001 From: Vyacheslav Tamarin Date: Mon, 5 Dec 2022 16:11:59 +0300 Subject: [PATCH] Add sdk checking --- .../plugin/language/python/PythonLanguageAssistant.kt | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/utbot-intellij-python/src/main/kotlin/org/utbot/intellij/plugin/language/python/PythonLanguageAssistant.kt b/utbot-intellij-python/src/main/kotlin/org/utbot/intellij/plugin/language/python/PythonLanguageAssistant.kt index 00f1bf7889..8339d80f2d 100644 --- a/utbot-intellij-python/src/main/kotlin/org/utbot/intellij/plugin/language/python/PythonLanguageAssistant.kt +++ b/utbot-intellij-python/src/main/kotlin/org/utbot/intellij/plugin/language/python/PythonLanguageAssistant.kt @@ -9,6 +9,9 @@ import com.intellij.psi.PsiFile import com.jetbrains.python.psi.PyClass import com.jetbrains.python.psi.PyFile import com.jetbrains.python.psi.PyFunction +import com.jetbrains.python.sdk.PythonSdkType +import org.jetbrains.kotlin.idea.util.projectStructure.module +import org.jetbrains.kotlin.idea.util.projectStructure.sdk import org.utbot.intellij.plugin.language.agnostic.LanguageAssistant object PythonLanguageAssistant : LanguageAssistant() { @@ -45,6 +48,9 @@ object PythonLanguageAssistant : LanguageAssistant() { val file = e.getData(CommonDataKeys.PSI_FILE) as? PyFile ?: return null val element = findPsiElement(file, editor) ?: return null + if (file.module?.sdk?.sdkType !is PythonSdkType) + return null + val containingFunction = getContainingElement(element) val containingClass = getContainingElement(element)