String (@NotNull PsiElement psiElement) { if (psiElement instanceof PyClass) { return "reference.dialogs.findUsages.class"; } if (psiElement instanceof PyFunction) { return "reference.dialogs.findUsages.method"; } if (psiElement instanceof PyReferenceExpression || psiElement instanceof PyTargetExpression || psiElement instanceof PyParameter) { return "reference.dialogs.findUsages.variable"; } return null; }