PyImportElement (@NotNull Editor editor, @NotNull PsiFile file) { final PsiElement elementAtCaret = file.findElementAt(editor.getCaretModel().getOffset()); final PyImportElement importElement = PsiTreeUtil.getParentOfType(elementAtCaret, PyImportElement.class); PyPsiUtils.assertValid(importElement); if (importElement != null && importElement.isValid()) { return importElement; } else { return null; } }