PyForStatement (@NotNull Editor editor, @NotNull PsiFile file) { final PsiElement elementAtCaret = file.findElementAt(editor.getCaretModel().getOffset()); return PsiTreeUtil.getParentOfType(elementAtCaret, PyForStatement.class); }