LanguageLevel (PsiElement psi) { final PsiFile file = psi.getContainingFile(); if (!(file instanceof PyFile)) { final PsiElement context = file.getContext(); if (context != null) return getLanguageLevel(context); return LanguageLevel.getDefault(); } return ((PyFile)file).getLanguageLevel(); }