MDS/openide
1
0
Fork 0
mirror of https://gitflic.ru/project/openide/openide.git synced 2026-04-30 10:20:15 +07:00
Code Issues Packages Projects Releases Wiki Activity

PY-77360: Create module-level docstring if it is not there

Browse Source 
GitOrigin-RevId: 840b9d7ca76db288184d0b8e33a833d65fd97948
This commit is contained in:
Ilia Zakoulov
2025-05-09 23:06:09 +02:00
committed by intellij-monorepo-bot
parent 89ce7b0460
commit 86334df8ad
1 changed files with 17 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
python/python-syntax-core/src/com/jetbrains/python/documentation/docstrings/PyDocstringGenerator.java
View File

@@ -21,6 +21,7 @@ import com.intellij.openapi.project.Project;
import com.intellij.openapi.util.Pair;
import com.intellij.openapi.util.TextRange;
import com.intellij.openapi.util.text.StringUtil;
import com.intellij.psi.PsiComment;
import com.intellij.psi.PsiElement;
import com.intellij.psi.PsiWhiteSpace;
import com.intellij.psi.util.PsiTreeUtil;
@@ -440,6 +441,22 @@ public final class PyDocstringGenerator {
docStringExpression.replace(replacement.getExpression());
}
else {
if (myDocStringOwner instanceof PyAstFile pyAstFile) {
PsiElement lastCommentOrNull = null;
for (PsiElement child : pyAstFile.getChildren()) {
if (child instanceof PsiWhiteSpace) {
continue;
}
if (child instanceof PsiComment) {
lastCommentOrNull = child;
}
else {
break;
}
}
myDocStringOwner.addAfter(replacement, lastCommentOrNull);
return myDocStringOwner;
}
PyAstStatementListContainer container = ObjectUtils.tryCast(myDocStringOwner, PyAstStatementListContainer.class);
if (container == null) {
throw new IllegalStateException("Should be a function or class");