PY-73366 Jupyter: Add ability to use own language of psi elements

GitOrigin-RevId: 57a4df0c69209f58f99762f293b47ce2d535bf3f
This commit is contained in:
Nikita.Ashihmin
2024-06-22 11:01:41 +04:00
committed by intellij-monorepo-bot
parent 1d190186ab
commit 4d5bff8b61

View File

@@ -3,6 +3,7 @@ package com.jetbrains.python.psi.impl;
import com.intellij.extapi.psi.StubBasedPsiElementBase;
import com.intellij.lang.ASTNode;
import com.intellij.lang.Language;
import com.intellij.openapi.util.TextRange;
import com.intellij.openapi.util.text.StringUtil;
import com.intellij.psi.*;
@@ -11,7 +12,6 @@ import com.intellij.psi.stubs.IStubElementType;
import com.intellij.psi.stubs.StubElement;
import com.intellij.psi.templateLanguages.OuterLanguageElement;
import com.jetbrains.python.PythonFileType;
import com.jetbrains.python.PythonLanguage;
import com.jetbrains.python.ast.PyAstElementVisitor;
import com.jetbrains.python.codeInsight.typing.PyTypingTypeProvider;
import com.jetbrains.python.psi.*;
@@ -36,8 +36,13 @@ public class PyBaseElementImpl<T extends StubElement> extends StubBasedPsiElemen
@NotNull
@Override
public PythonLanguage getLanguage() {
return (PythonLanguage)PythonFileType.INSTANCE.getLanguage();
public Language getLanguage() {
if (this instanceof PyFile) {
return PythonFileType.INSTANCE.getLanguage();
}
PsiFile file = getContainingFile();
return file.getLanguage();
}
@Override