PyAugAssignmentStatement (final PsiElement element) { final PyAugAssignmentStatement statement = PsiTreeUtil.getParentOfType(element, PyAugAssignmentStatement.class); if (statement == null){ return null; } return statement.getTarget() == element ? statement : null; }