PyCallExpression (final PsiElement element) { final PsiElement parent = element.getParent(); if (parent instanceof PyCallExpression expression){ return expression.getCallee() == element ? expression : null; } return null; }