fixed PY-9690 Show 'Insert documentation string stub' only on function name

This commit is contained in:
Ekaterina Tuzova
2013-05-09 21:42:01 +04:00
parent 76e34ee7c6
commit 465a0b05d5
@@ -15,6 +15,7 @@ import com.jetbrains.python.documentation.PyDocstringGenerator;
import com.jetbrains.python.documentation.doctest.PyDocstringFile;
import com.jetbrains.python.psi.PyFile;
import com.jetbrains.python.psi.PyFunction;
import com.jetbrains.python.psi.PyStatementList;
import com.jetbrains.python.psi.PyUtil;
import org.jetbrains.annotations.NotNull;
@@ -44,7 +45,8 @@ public class PyGenerateDocstringIntention extends BaseIntentionAction {
return false;
}
PyFunction function = PsiTreeUtil.getParentOfType(elementAt, PyFunction.class);
if (function == null) {
final PyStatementList statementList = PsiTreeUtil.getParentOfType(elementAt, PyStatementList.class, false, PyFunction.class);
if (function == null || statementList != null) {
return false;
}
return isAvailableForFunction(project, function);