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