MDS/openide
1
0
Fork 0
mirror of https://gitflic.ru/project/openide/openide.git synced 2026-09-27 10:03:11 +07:00
Code Issues Packages Projects Releases Wiki Activity

[evaluation-plugin] code-generation: fix collecting methods bodies for python

Browse Source 
GitOrigin-RevId: 786d96900d277649ee422b890abed5b6cc84328f
This commit is contained in:
Alexey Kalina
2024-07-02 14:06:08 +00:00
committed by intellij-monorepo-bot
parent 3d85ac63df
commit 7efc1bb1f9
1 changed files with 5 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
plugins/evaluation-plugin/languages/python/src/com/intellij/cce/visitor/PythonCodeGenerationVisitor.kt
+5 -12
View File
@@ -1,12 +1,10 @@
package com.intellij.cce.visitor
import com.intellij.cce.core.*
import com.intellij.psi.PsiElement
import com.intellij.psi.PsiElementVisitor
import com.intellij.psi.util.startOffset
import com.jetbrains.python.psi.PyFunction
import com.jetbrains.python.psi.PyRecursiveElementVisitor
import com.jetbrains.python.psi.PyStatementList
class PythonCodeGenerationVisitor : CodeGenerationVisitorBase(Language.PYTHON) {
override fun createPsiVisitor(codeFragment: CodeFragment): PsiElementVisitor {
@@ -19,16 +17,11 @@ private class PythonCodeGenerationPsiVisitor(private val codeFragment: CodeFragm
codeFragment.addChild(
CodeToken(node.text, node.startOffset, SimpleTokenProperties.create(TypeProperty.METHOD, SymbolLocation.PROJECT) {})
)
val body = node.statementList
codeFragment.addChild(
CodeToken(body.text, body.startOffset, SimpleTokenProperties.create(TypeProperty.METHOD_BODY, SymbolLocation.PROJECT) {})
)
super.visitPyFunction(node)
}
override fun visitElement(element: PsiElement) {
if (element is PyStatementList) {
codeFragment.addChild(
CodeToken(element.text, element.startOffset, SimpleTokenProperties.create(TypeProperty.METHOD_BODY, SymbolLocation.PROJECT) {})
)
}
super.visitElement(element)
}
}