PyDictLiteralExpression (@NotNull final PyExpression arg) { final PyReferenceExpression referenceExpression = PsiTreeUtil.getChildOfType(arg, PyReferenceExpression.class); if (referenceExpression != null) { final PsiElement resolveResult = referenceExpression.getReference().resolve(); if (resolveResult instanceof PyTargetExpression) { final PyAssignmentStatement assignmentStatement = PsiTreeUtil.getParentOfType(resolveResult, PyAssignmentStatement.class); return PsiTreeUtil.getChildOfType(assignmentStatement, PyDictLiteralExpression.class); } } return null; }